| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/media/webrtc/window_icon_util.h" |
| |
| #include "ui/base/x/x11_util.h" |
| #include "ui/gfx/x/x11.h" |
| #include "ui/gfx/x/x11_atom_cache.h" |
| #include "ui/gfx/x/x11_error_tracker.h" |
| #include "ui/gfx/x/x11_types.h" |
| |
| gfx::ImageSkia GetWindowIcon(content::DesktopMediaID id) { |
| DCHECK(id.type == content::DesktopMediaID::TYPE_WINDOW); |
| |
| Display* display = gfx::GetXDisplay(); |
| Atom property = gfx::GetAtom("_NET_WM_ICON"); |
| Atom actual_type; |
| int actual_format; |
| unsigned long bytes_after; // NOLINT: type required by XGetWindowProperty |
| unsigned long size; |
| long* data; |
| |
| // The |error_tracker| essentially provides an empty X error handler for |
| // the call of XGetWindowProperty. The motivation is to guard against crash |
| // for any reason that XGetWindowProperty fails. For example, at the time that |
| // XGetWindowProperty is called, the window handler (a.k.a |id.id|) may |
| // already be invalid due to the fact that the end user has closed the |
| // corresponding window, etc. |
| std::unique_ptr<gfx::X11ErrorTracker> error_tracker( |
| new gfx::X11ErrorTracker()); |
| int status = XGetWindowProperty(display, id.id, property, 0L, ~0L, x11::False, |
| AnyPropertyType, &actual_type, &actual_format, |
| &size, &bytes_after, |
| reinterpret_cast<unsigned char**>(&data)); |
| error_tracker.reset(); |
| |
| if (status != x11::Success) { |
| return gfx::ImageSkia(); |
| } |
| |
| // The format of |data| is concatenation of sections like |
| // [width, height, pixel data of size width * height], and the total bytes |
| // number of |data| is |size|. And here we are picking the largest icon. |
| int width = 0; |
| int height = 0; |
| int start = 0; |
| int i = 0; |
| while (i + 1 < static_cast<int>(size)) { |
| if ((i == 0 || static_cast<int>(data[i] * data[i + 1]) > width * height) && |
| (i + 1 + data[i] * data[i + 1] < static_cast<int>(size))) { |
| width = static_cast<int>(data[i]); |
| height = static_cast<int>(data[i + 1]); |
| start = i + 2; |
| } |
| i = i + 2 + static_cast<int>(data[i] * data[i + 1]); |
| } |
| |
| SkBitmap result; |
| SkImageInfo info = SkImageInfo::MakeN32(width, height, kUnpremul_SkAlphaType); |
| result.allocPixels(info); |
| |
| uint32_t* pixels_data = reinterpret_cast<uint32_t*>(result.getPixels()); |
| |
| for (long y = 0; y < height; ++y) { |
| for (long x = 0; x < width; ++x) { |
| pixels_data[result.rowBytesAsPixels() * y + x] = |
| static_cast<uint32_t>(data[start + width * y + x]); |
| } |
| } |
| |
| XFree(data); |
| return gfx::ImageSkia::CreateFrom1xBitmap(result); |
| } |