OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "ui/surface/transport_dib.h" | |
6 | |
7 #include <errno.h> | |
8 #include <stdlib.h> | |
9 #include <sys/ipc.h> | |
10 #include <sys/shm.h> | |
11 | |
12 #include "base/logging.h" | |
13 #include "base/memory/scoped_ptr.h" | |
14 #include "skia/ext/platform_canvas.h" | |
15 #include "ui/base/x/x11_util.h" | |
16 #include "ui/gfx/size.h" | |
17 | |
18 // The shmat system call uses this as it's invalid return address | |
19 static void *const kInvalidAddress = (void*) -1; | |
20 | |
21 TransportDIB::TransportDIB() | |
22 : address_(kInvalidAddress), | |
23 x_shm_(0), | |
24 display_(NULL), | |
25 inflight_counter_(0), | |
26 detached_(false), | |
27 size_(0) { | |
28 } | |
29 | |
30 TransportDIB::~TransportDIB() { | |
31 if (address_ != kInvalidAddress) { | |
32 shmdt(address_); | |
33 address_ = kInvalidAddress; | |
34 } | |
35 | |
36 if (x_shm_) { | |
37 DCHECK(display_); | |
38 ui::DetachSharedMemory(display_, x_shm_); | |
39 } | |
40 } | |
41 | |
42 // static | |
43 TransportDIB* TransportDIB::Create(size_t size, uint32 sequence_num) { | |
44 const int shmkey = shmget(IPC_PRIVATE, size, 0600); | |
45 if (shmkey == -1) { | |
46 DLOG(ERROR) << "Failed to create SysV shared memory region" | |
47 << " errno:" << errno; | |
48 return NULL; | |
49 } else { | |
50 VLOG(1) << "Created SysV shared memory region " << shmkey; | |
51 } | |
52 | |
53 void* address = shmat(shmkey, NULL /* desired address */, 0 /* flags */); | |
54 // Here we mark the shared memory for deletion. Since we attached it in the | |
55 // line above, it doesn't actually get deleted but, if we crash, this means | |
56 // that the kernel will automatically clean it up for us. | |
57 shmctl(shmkey, IPC_RMID, 0); | |
58 if (address == kInvalidAddress) | |
59 return NULL; | |
60 | |
61 TransportDIB* dib = new TransportDIB; | |
62 | |
63 dib->key_.shmkey = shmkey; | |
64 dib->address_ = address; | |
65 dib->size_ = size; | |
66 return dib; | |
67 } | |
68 | |
69 // static | |
70 TransportDIB* TransportDIB::Map(Handle handle) { | |
71 scoped_ptr<TransportDIB> dib(CreateWithHandle(handle)); | |
72 if (!dib->Map()) | |
73 return NULL; | |
74 return dib.release(); | |
75 } | |
76 | |
77 // static | |
78 TransportDIB* TransportDIB::CreateWithHandle(Handle shmkey) { | |
79 TransportDIB* dib = new TransportDIB; | |
80 dib->key_.shmkey = shmkey; | |
81 return dib; | |
82 } | |
83 | |
84 // static | |
85 bool TransportDIB::is_valid_handle(Handle dib) { | |
86 return dib >= 0; | |
87 } | |
88 | |
89 // static | |
90 bool TransportDIB::is_valid_id(Id id) { | |
91 return id.shmkey != -1; | |
92 } | |
93 | |
94 skia::PlatformCanvas* TransportDIB::GetPlatformCanvas(int w, int h) { | |
95 if ((address_ == kInvalidAddress && !Map()) || !VerifyCanvasSize(w, h)) | |
96 return NULL; | |
97 return skia::CreatePlatformCanvas(w, h, true, | |
98 reinterpret_cast<uint8_t*>(memory()), | |
99 skia::RETURN_NULL_ON_FAILURE); | |
100 } | |
101 | |
102 bool TransportDIB::Map() { | |
103 if (!is_valid_id(key_)) | |
104 return false; | |
105 if (address_ != kInvalidAddress) | |
106 return true; | |
107 | |
108 struct shmid_ds shmst; | |
109 if (shmctl(key_.shmkey, IPC_STAT, &shmst) == -1) | |
110 return false; | |
111 | |
112 void* address = shmat(key_.shmkey, NULL /* desired address */, 0 /* flags */); | |
113 if (address == kInvalidAddress) | |
114 return false; | |
115 | |
116 address_ = address; | |
117 size_ = shmst.shm_segsz; | |
118 return true; | |
119 } | |
120 | |
121 void* TransportDIB::memory() const { | |
122 DCHECK_NE(address_, kInvalidAddress); | |
123 return address_; | |
124 } | |
125 | |
126 TransportDIB::Id TransportDIB::id() const { | |
127 return key_; | |
128 } | |
129 | |
130 TransportDIB::Handle TransportDIB::handle() const { | |
131 return key_.shmkey; | |
132 } | |
133 | |
134 XID TransportDIB::MapToX(Display* display) { | |
135 if (!x_shm_) { | |
136 x_shm_ = ui::AttachSharedMemory(display, key_.shmkey); | |
137 display_ = display; | |
138 } | |
139 | |
140 return x_shm_; | |
141 } | |
142 | |
143 void TransportDIB::DecreaseInFlightCounter() { | |
144 CHECK(inflight_counter_); | |
145 inflight_counter_--; | |
146 if (!inflight_counter_ && detached_) | |
147 delete this; | |
148 } | |
149 | |
150 void TransportDIB::Detach() { | |
151 CHECK(!detached_); | |
152 detached_ = true; | |
153 if (!inflight_counter_) | |
154 delete this; | |
155 } | |
OLD | NEW |