From 8126a6be82f04a1972806ef235e581dec0bac650 Mon Sep 17 00:00:00 2001 From: Dominik Wenger Date: Fri, 27 Jun 2008 22:48:25 +0000 Subject: rbutil: forgot the new files git-svn-id: svn://svn.rockbox.org/rockbox/trunk@17815 a1c6a512-1295-4272-9138-f99709370657 --- rbutil/rbutilqt/detect.cpp | 409 +++++++++++++++++++++++++++++++++++++++++++++ rbutil/rbutilqt/detect.h | 51 ++++++ 2 files changed, 460 insertions(+) create mode 100644 rbutil/rbutilqt/detect.cpp create mode 100644 rbutil/rbutilqt/detect.h diff --git a/rbutil/rbutilqt/detect.cpp b/rbutil/rbutilqt/detect.cpp new file mode 100644 index 0000000000..95dee307b2 --- /dev/null +++ b/rbutil/rbutilqt/detect.cpp @@ -0,0 +1,409 @@ +/*************************************************************************** + * __________ __ ___. + * Open \______ \ ____ ____ | | _\_ |__ _______ ___ + * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / + * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < + * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ + * \/ \/ \/ \/ \/ + * + * Copyright (C) 2007 by Dominik Wenger + * $Id: detect.cpp 17769 2008-06-23 20:31:44Z Domonoky $ + * + * All files in this archive are subject to the GNU General Public License. + * See the file COPYING in the source tree root for full license agreement. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ****************************************************************************/ + + +#include "detect.h" + +#include +#include +#include +#include +#include + + +// Windows Includes +#if defined(Q_OS_WIN32) +#if defined(UNICODE) +#define _UNICODE +#endif +#include +#include +#include +#include +#include +#endif + +// Linux and Mac includes +#if defined(Q_OS_LINUX) || defined(Q_OS_MACX) +#include +#include +#include +#include +#endif + +// Linux includes +#if defined(Q_OS_LINUX) +#include +#endif + +// Mac includes +#if defined(Q_OS_MACX) +#include +#include +#include +#endif + + +/** @brief detect permission of user (only Windows at moment). + * @return enum userlevel. + */ +#if defined(Q_OS_WIN32) +enum userlevel Detect::userPermissions(void) +{ + LPUSER_INFO_1 buf; + NET_API_STATUS napistatus; + wchar_t userbuf[UNLEN]; + DWORD usersize = UNLEN; + BOOL status; + enum userlevel result; + + status = GetUserNameW(userbuf, &usersize); + if(!status) + return ERR; + + napistatus = NetUserGetInfo(NULL, userbuf, (DWORD)1, (LPBYTE*)&buf); + + switch(buf->usri1_priv) { + case USER_PRIV_GUEST: + result = GUEST; + break; + case USER_PRIV_USER: + result = USER; + break; + case USER_PRIV_ADMIN: + result = ADMIN; + break; + default: + result = ERR; + break; + } + NetApiBufferFree(buf); + + return result; +} + +/** @brief detects user permissions (only Windows at moment). + * @return a user readable string with the permission. + */ +QString Detect::userPermissionsString(void) +{ + QString result; + int perm = getUserPermissions(); + switch(perm) { + case GUEST: + result = QObject::tr("Guest"); + break; + case ADMIN: + result = QObject::tr("Admin"); + break; + case USER: + result = QObject::tr("User"); + break; + default: + result = QObject::tr("Error"); + break; + } + return result; +} +#endif + + +/** @brief detects current Username. + * @return string with Username. + */ +QString Detect::userName(void) +{ +#if defined(Q_OS_WIN32) + wchar_t userbuf[UNLEN]; + DWORD usersize = UNLEN; + BOOL status; + + status = GetUserNameW(userbuf, &usersize); + + return QString::fromWCharArray(userbuf); +#endif +#if defined(Q_OS_LINUX) || defined(Q_OS_MACX) + struct passwd *user; + user = getpwuid(geteuid()); + return QString(user->pw_name); +#endif +} + + +/** @brief detects the OS Version + * @return String with OS Version. + */ +QString Detect::osVersionString(void) +{ + QString result; +#if defined(Q_OS_WIN32) + OSVERSIONINFO osvi; + ZeroMemory(&osvi, sizeof(OSVERSIONINFO)); + osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO); + GetVersionEx(&osvi); + + result = QString("Windows version %1.%2, ").arg(osvi.dwMajorVersion).arg(osvi.dwMinorVersion); + result += QString("build %1 (%2)").arg(osvi.dwBuildNumber).arg(QString::fromWCharArray(osvi.szCSDVersion)); +#endif +#if defined(Q_OS_LINUX) || defined(Q_OS_MACX) + struct utsname u; + int ret; + ret = uname(&u); + + result = QString("CPU: %1
System: %2
Release: %3
Version: %4") + .arg(u.machine).arg(u.sysname).arg(u.release).arg(u.version); +#endif + return result; +} + +/** @brief detect devices based on usb pid / vid. + * @return list with usb VID / PID values. + */ +QList Detect::listUsbIds(void) +{ + QList usbids; + // usb pid detection +#if defined(Q_OS_LINUX) | defined(Q_OS_MACX) + usb_init(); + usb_find_busses(); + usb_find_devices(); + struct usb_bus *b; + b = usb_busses; + + while(b) { + qDebug() << "bus:" << b->dirname << b->devices; + if(b->devices) { + qDebug() << "devices present."; + struct usb_device *u; + u = b->devices; + while(u) { + uint32_t id; + id = u->descriptor.idVendor << 16 | u->descriptor.idProduct; + if(id) usbids.append(id); + u = u->next; + } + } + b = b->next; + } +#endif + +#if defined(Q_OS_WIN32) + HDEVINFO deviceInfo; + SP_DEVINFO_DATA infoData; + DWORD i; + + // Iterate over all devices + // by doing it this way it's unneccessary to use GUIDs which might be not + // present in current MinGW. It also seemed to be more reliably than using + // a GUID. + // See KB259695 for an example. + deviceInfo = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_ALLCLASSES | DIGCF_PRESENT); + + infoData.cbSize = sizeof(SP_DEVINFO_DATA); + + for(i = 0; SetupDiEnumDeviceInfo(deviceInfo, i, &infoData); i++) { + DWORD data; + LPTSTR buffer = NULL; + DWORD buffersize = 0; + + // get device desriptor first + // for some reason not doing so results in bad things (tm) + while(!SetupDiGetDeviceRegistryProperty(deviceInfo, &infoData, + SPDRP_DEVICEDESC,&data, (PBYTE)buffer, buffersize, &buffersize)) { + if(GetLastError() == ERROR_INSUFFICIENT_BUFFER) { + if(buffer) free(buffer); + // double buffer size to avoid problems as per KB888609 + buffer = (LPTSTR)malloc(buffersize * 2); + } + else { + break; + } + } + + // now get the hardware id, which contains PID and VID. + while(!SetupDiGetDeviceRegistryProperty(deviceInfo, &infoData, + SPDRP_HARDWAREID,&data, (PBYTE)buffer, buffersize, &buffersize)) { + if(GetLastError() == ERROR_INSUFFICIENT_BUFFER) { + if(buffer) free(buffer); + // double buffer size to avoid problems as per KB888609 + buffer = (LPTSTR)malloc(buffersize * 2); + } + else { + break; + } + } + + unsigned int vid, pid, rev; + if(_stscanf(buffer, _TEXT("USB\\Vid_%x&Pid_%x&Rev_%x"), &vid, &pid, &rev) != 3) { + qDebug() << "Error getting USB ID -- possibly no USB device"; + } + else { + uint32_t id; + id = vid << 16 | pid; + usbids.append(id); + qDebug("VID: %04x PID: %04x", vid, pid); + } + if(buffer) free(buffer); + } + SetupDiDestroyDeviceInfoList(deviceInfo); + +#endif + return usbids; +} + + +/** @brief detects current system proxy + * @return QUrl with proxy or empty + */ +QUrl Detect::systemProxy(void) +{ +#if defined(Q_OS_LINUX) + return QUrl(getenv("http_proxy")); +#elif defined(Q_OS_WIN32) + HKEY hk; + wchar_t proxyval[80]; + DWORD buflen = 80; + long ret; + DWORD enable; + DWORD enalen = sizeof(DWORD); + + ret = RegOpenKeyEx(HKEY_CURRENT_USER, + _TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings"), + 0, KEY_QUERY_VALUE, &hk); + if(ret != ERROR_SUCCESS) return QUrl(""); + + ret = RegQueryValueEx(hk, _TEXT("ProxyServer"), NULL, NULL, (LPBYTE)proxyval, &buflen); + if(ret != ERROR_SUCCESS) return QUrl(""); + + ret = RegQueryValueEx(hk, _TEXT("ProxyEnable"), NULL, NULL, (LPBYTE)&enable, &enalen); + if(ret != ERROR_SUCCESS) return QUrl(""); + + RegCloseKey(hk); + + //qDebug() << QString::fromWCharArray(proxyval) << QString("%1").arg(enable); + if(enable != 0) + return QUrl("http://" + QString::fromWCharArray(proxyval)); + else + return QUrl(""); +#else + return QUrl(""); +#endif +} + + +/** @brief detects the installed Rockbox version + * @return QString with version. Empty if not aviable + */ +QString Detect::installedVersion(QString mountpoint) +{ + // read rockbox-info.txt + QFile info(mountpoint +"/.rockbox/rockbox-info.txt"); + if(!info.open(QIODevice::ReadOnly)) + { + return ""; + } + + while (!info.atEnd()) { + QString line = info.readLine(); + + if(line.contains("Version:")) + { + return line.remove("Version:").trimmed(); + } + } + info.close(); + return ""; +} + + +/** @brief detects installed rockbox target id + * @return TargetId of installed rockbox, or -1 if not available + */ +int Detect::installedTargetId(QString mountpoint) +{ + // read rockbox-info.txt + QFile info(mountpoint +"/.rockbox/rockbox-info.txt"); + if(!info.open(QIODevice::ReadOnly)) + { + return -1; + } + + while (!info.atEnd()) + { + QString line = info.readLine(); + if(line.contains("Target id:")) + { + qDebug() << line; + return line.remove("Target id:").trimmed().toInt(); + } + } + info.close(); + return -1; +} + + +/** @brief checks different Enviroment things. Ask if user wants to continue. + * @param settings A pointer to rbutils settings class + * @param permission if it should check for permission + * @param targetId the targetID to check for. if it is -1 no check is done. + * @return true if everything is ok, or user wants to continue + */ +bool Detect::check(RbSettings* settings,bool permission,int targetId) +{ + QString text = ""; + + // check permission + if(permission) + { +#if defined(Q_OS_WIN32) + if(Detect::userPermissions() != Detect::ADMIN) + { + text += QObject::tr("Permissions are not sufficient! \n Run with admin rights. \n\n"); + } +#endif + } + + // Check TargetId + if(targetId > 0) + { + int installedID = Detect::installedTargetId(settings->mountpoint()); + if( installedID != -1 && installedID != targetId) + { + text += QObject::tr("Target mismatch detected. \n\n" + "Installed target: %1.\n" + "New Target: %2.\n\n").arg(settings->nameOfTargetId(installedID),settings->curName()); + + } + } + + // show message Box + if(text != "") + { + text += QObject::tr("\n Do you want to continue ?"); + if(QMessageBox::warning(NULL, QObject::tr("Problems detected"),text, + QMessageBox::Yes | QMessageBox::No) == QMessageBox::No) + { + return false; + } + } + + return true; +} + + diff --git a/rbutil/rbutilqt/detect.h b/rbutil/rbutilqt/detect.h new file mode 100644 index 0000000000..98d6040ac0 --- /dev/null +++ b/rbutil/rbutilqt/detect.h @@ -0,0 +1,51 @@ +/*************************************************************************** + * __________ __ ___. + * Open \______ \ ____ ____ | | _\_ |__ _______ ___ + * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / + * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < + * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ + * \/ \/ \/ \/ \/ + * + * Copyright (C) 2007 by Dominik Wenger + * $Id: detect.h 17769 2008-06-23 20:31:44Z Domonoky $ + * + * All files in this archive are subject to the GNU General Public License. + * See the file COPYING in the source tree root for full license agreement. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ****************************************************************************/ + + +#ifndef DETECT_H +#define DETECT_H + +#include +#include +#include "rbsettings.h" + +class Detect +{ +public: + Detect() {} + +#if defined(Q_OS_WIN32) + enum userlevel { ERR, GUEST, USER, ADMIN }; + static enum userlevel userPermissions(void); + static QString userPermissionsString(void); +#endif + + static QString userName(void); + static QString osVersionString(void); + static QList listUsbIds(void); + + static QUrl systemProxy(void); + static QString installedVersion(QString mountpoint); + static int installedTargetId(QString mountpoint); + + static bool check(RbSettings* settings,bool permission,int targetId); + +#endif +}; + -- cgit v1.2.3