File indexing completed on 2024-05-19 16:34:58
0001 /* 0002 KWin - the KDE window manager 0003 This file is part of the KDE project. 0004 0005 SPDX-FileCopyrightText: 2014 Martin Gräßlin <mgraesslin@kde.org> 0006 0007 SPDX-License-Identifier: GPL-2.0-or-later 0008 */ 0009 #include "udev.h" 0010 #include "utils/common.h" 0011 // Qt 0012 #include <QByteArray> 0013 #include <QDebug> 0014 // system 0015 #include <cerrno> 0016 #include <functional> 0017 #include <libudev.h> 0018 0019 namespace KWin 0020 { 0021 0022 Udev::Udev() 0023 : m_udev(udev_new()) 0024 { 0025 } 0026 0027 Udev::~Udev() 0028 { 0029 if (m_udev) { 0030 udev_unref(m_udev); 0031 } 0032 } 0033 0034 class UdevEnumerate 0035 { 0036 public: 0037 UdevEnumerate(Udev *udev); 0038 ~UdevEnumerate(); 0039 0040 enum class Match { 0041 SubSystem, 0042 SysName 0043 }; 0044 void addMatch(Match match, const char *name); 0045 void scan(); 0046 std::vector<UdevDevice::Ptr> find(); 0047 0048 private: 0049 Udev *m_udev; 0050 0051 struct EnumerateDeleter 0052 { 0053 void operator()(udev_enumerate *e) 0054 { 0055 udev_enumerate_unref(e); 0056 } 0057 }; 0058 std::unique_ptr<udev_enumerate, EnumerateDeleter> m_enumerate; 0059 }; 0060 0061 UdevEnumerate::UdevEnumerate(Udev *udev) 0062 : m_udev(udev) 0063 , m_enumerate(udev_enumerate_new(*m_udev)) 0064 { 0065 } 0066 0067 UdevEnumerate::~UdevEnumerate() = default; 0068 0069 void UdevEnumerate::addMatch(UdevEnumerate::Match match, const char *name) 0070 { 0071 if (!m_enumerate) { 0072 return; 0073 } 0074 switch (match) { 0075 case Match::SubSystem: 0076 udev_enumerate_add_match_subsystem(m_enumerate.get(), name); 0077 break; 0078 case Match::SysName: 0079 udev_enumerate_add_match_sysname(m_enumerate.get(), name); 0080 break; 0081 default: 0082 Q_UNREACHABLE(); 0083 break; 0084 } 0085 } 0086 0087 void UdevEnumerate::scan() 0088 { 0089 if (!m_enumerate) { 0090 return; 0091 } 0092 udev_enumerate_scan_devices(m_enumerate.get()); 0093 } 0094 0095 std::vector<UdevDevice::Ptr> UdevEnumerate::find() 0096 { 0097 if (!m_enumerate) { 0098 return {}; 0099 } 0100 std::vector<UdevDevice::Ptr> vect; 0101 udev_list_entry *it = udev_enumerate_get_list_entry(m_enumerate.get()); 0102 while (it) { 0103 auto current = it; 0104 it = udev_list_entry_get_next(it); 0105 auto device = m_udev->deviceFromSyspath(udev_list_entry_get_name(current)); 0106 if (device) { 0107 vect.push_back(std::move(device)); 0108 } 0109 } 0110 return vect; 0111 } 0112 0113 std::vector<UdevDevice::Ptr> Udev::listGPUs() 0114 { 0115 if (!m_udev) { 0116 return {}; 0117 } 0118 #if defined(Q_OS_FREEBSD) 0119 std::vector<UdevDevice::Ptr> r; 0120 r.push_back(deviceFromSyspath("/dev/dri/card0")); 0121 return r; 0122 #else 0123 UdevEnumerate enumerate(this); 0124 enumerate.addMatch(UdevEnumerate::Match::SubSystem, "drm"); 0125 enumerate.addMatch(UdevEnumerate::Match::SysName, "card[0-9]"); 0126 enumerate.scan(); 0127 auto vect = enumerate.find(); 0128 std::sort(vect.begin(), vect.end(), [](const UdevDevice::Ptr &device1, const UdevDevice::Ptr &device2) { 0129 // if set as boot GPU, prefer 1 0130 if (device1->isBootVga()) { 0131 return true; 0132 } 0133 // if set as boot GPU, prefer 2 0134 if (device2->isBootVga()) { 0135 return false; 0136 } 0137 return true; 0138 }); 0139 return vect; 0140 #endif 0141 } 0142 0143 UdevDevice::Ptr Udev::deviceFromSyspath(const char *syspath) 0144 { 0145 auto dev = udev_device_new_from_syspath(m_udev, syspath); 0146 if (!dev) { 0147 qCWarning(KWIN_CORE) << "failed to retrieve device for" << syspath << strerror(errno); 0148 return {}; 0149 } 0150 return UdevDevice::Ptr(new UdevDevice(dev)); 0151 } 0152 0153 std::unique_ptr<UdevMonitor> Udev::monitor() 0154 { 0155 auto m = std::make_unique<UdevMonitor>(this); 0156 if (m->isValid()) { 0157 return m; 0158 } else { 0159 return nullptr; 0160 } 0161 } 0162 0163 UdevDevice::UdevDevice(udev_device *device) 0164 : m_device(device) 0165 { 0166 Q_ASSERT(device); 0167 } 0168 0169 UdevDevice::~UdevDevice() 0170 { 0171 udev_device_unref(m_device); 0172 } 0173 0174 QString UdevDevice::devNode() const 0175 { 0176 return QString::fromUtf8(udev_device_get_devnode(m_device)); 0177 } 0178 0179 dev_t UdevDevice::devNum() const 0180 { 0181 return udev_device_get_devnum(m_device); 0182 } 0183 0184 const char *UdevDevice::property(const char *key) 0185 { 0186 return udev_device_get_property_value(m_device, key); 0187 } 0188 0189 QMap<QByteArray, QByteArray> UdevDevice::properties() const 0190 { 0191 QMap<QByteArray, QByteArray> r; 0192 auto it = udev_device_get_properties_list_entry(m_device); 0193 auto current = it; 0194 udev_list_entry_foreach(current, it) 0195 { 0196 r.insert(udev_list_entry_get_name(current), udev_list_entry_get_value(current)); 0197 } 0198 return r; 0199 } 0200 0201 bool UdevDevice::hasProperty(const char *key, const char *value) 0202 { 0203 const char *p = property(key); 0204 if (!p) { 0205 return false; 0206 } 0207 return qstrcmp(p, value) == 0; 0208 } 0209 0210 bool UdevDevice::isBootVga() const 0211 { 0212 auto pci = udev_device_get_parent_with_subsystem_devtype(m_device, "pci", nullptr); 0213 if (!pci) { 0214 return false; 0215 } 0216 const char *systAttrValue = udev_device_get_sysattr_value(pci, "boot_vga"); 0217 return systAttrValue && qstrcmp(systAttrValue, "1") == 0; 0218 } 0219 0220 QString UdevDevice::seat() const 0221 { 0222 QString deviceSeat = udev_device_get_property_value(m_device, "ID_SEAT"); 0223 if (deviceSeat.isEmpty()) { 0224 deviceSeat = QStringLiteral("seat0"); 0225 } 0226 return deviceSeat; 0227 } 0228 0229 QString UdevDevice::action() const 0230 { 0231 return QString::fromLocal8Bit(udev_device_get_action(m_device)); 0232 } 0233 0234 UdevMonitor::UdevMonitor(Udev *udev) 0235 : m_monitor(udev_monitor_new_from_netlink(*udev, "udev")) 0236 { 0237 } 0238 0239 UdevMonitor::~UdevMonitor() 0240 { 0241 if (m_monitor) { 0242 udev_monitor_unref(m_monitor); 0243 } 0244 } 0245 0246 int UdevMonitor::fd() const 0247 { 0248 if (m_monitor) { 0249 return udev_monitor_get_fd(m_monitor); 0250 } 0251 return -1; 0252 } 0253 0254 void UdevMonitor::filterSubsystemDevType(const char *subSystem, const char *devType) 0255 { 0256 if (!m_monitor) { 0257 return; 0258 } 0259 udev_monitor_filter_add_match_subsystem_devtype(m_monitor, subSystem, devType); 0260 } 0261 0262 void UdevMonitor::enable() 0263 { 0264 if (!m_monitor) { 0265 return; 0266 } 0267 udev_monitor_enable_receiving(m_monitor); 0268 } 0269 0270 UdevDevice::Ptr UdevMonitor::getDevice() 0271 { 0272 if (!m_monitor) { 0273 return UdevDevice::Ptr(); 0274 } 0275 auto dev = udev_monitor_receive_device(m_monitor); 0276 if (!dev) { 0277 return {}; 0278 } 0279 0280 return UdevDevice::Ptr(new UdevDevice(dev)); 0281 } 0282 0283 }