/****************************************************************************+
−
**+
−
** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).+
−
** All rights reserved.+
−
** Contact: Nokia Corporation (qt-info@nokia.com)+
−
**+
−
** This file is part of the tools applications of the Qt Toolkit.+
−
**+
−
** $QT_BEGIN_LICENSE:LGPL$+
−
** No Commercial Usage+
−
** This file contains pre-release code and may not be distributed.+
−
** You may use this file in accordance with the terms and conditions+
−
** contained in the Technology Preview License Agreement accompanying+
−
** this package.+
−
**+
−
** GNU Lesser General Public License Usage+
−
** Alternatively, this file may be used under the terms of the GNU Lesser+
−
** General Public License version 2.1 as published by the Free Software+
−
** Foundation and appearing in the file LICENSE.LGPL included in the+
−
** packaging of this file.  Please review the following information to+
−
** ensure the GNU Lesser General Public License version 2.1 requirements+
−
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.+
−
**+
−
** In addition, as a special exception, Nokia gives you certain additional+
−
** rights.  These rights are described in the Nokia Qt LGPL Exception+
−
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.+
−
**+
−
** If you have questions regarding the use of this file, please contact+
−
** Nokia at qt-info@nokia.com.+
−
**+
−
**+
−
**+
−
**+
−
**+
−
**+
−
**+
−
**+
−
** $QT_END_LICENSE$+
−
**+
−
****************************************************************************/+
−
+
−
#include <QFile>+
−
#include <QProcess>+
−
#include <QLibraryInfo>+
−
#include <qt_windows.h>+
−
#include <io.h>+
−
+
−
QT_BEGIN_NAMESPACE+
−
+
−
static QString quotePath(const QString &s)+
−
{+
−
    if (!s.startsWith(QLatin1Char('\"')) && s.contains(QLatin1Char(' ')))+
−
        return QLatin1Char('\"') + s + QLatin1Char('\"');+
−
    return s;+
−
}+
−
+
−
+
−
static bool runWithQtInEnvironment(const QString &cmd)+
−
{+
−
    QProcess proc;+
−
    +
−
    // prepend the qt binary directory to the path+
−
    QStringList env = QProcess::systemEnvironment();+
−
    for (int i=0; i<env.count(); ++i) {+
−
        QString var = env.at(i);+
−
        int setidx = var.indexOf(QLatin1Char('='));+
−
        if (setidx != -1) {+
−
            QString varname = var.left(setidx).trimmed().toUpper();+
−
            if (varname == QLatin1String("PATH")) {+
−
                var = var.mid(setidx + 1);+
−
                var = QLatin1String("PATH=") + +
−
                    QLibraryInfo::location(QLibraryInfo::BinariesPath) ++
−
                    QLatin1Char(';') + var;+
−
                env[i] = var;+
−
                break;+
−
            }+
−
        }+
−
    }+
−
+
−
    proc.setEnvironment(env);+
−
    proc.start(cmd);+
−
    proc.waitForFinished(-1);+
−
    +
−
    return (proc.exitCode() == 0);+
−
}+
−
+
−
static bool attachTypeLibrary(const QString &applicationName, int resource, const QByteArray &data, QString *errorMessage)+
−
{+
−
    HANDLE hExe = BeginUpdateResource((const wchar_t *)applicationName.utf16(), false);+
−
    if (hExe == 0) {+
−
        if (errorMessage)+
−
            *errorMessage = QString::fromLatin1("Failed to attach type library to binary %1 - could not open file.").arg(applicationName);+
−
        return false;+
−
    }+
−
    if (!UpdateResource(hExe, L"TYPELIB", MAKEINTRESOURCE(resource), 0, (void*)data.data(), data.count())) {+
−
        EndUpdateResource(hExe, true);+
−
        if (errorMessage)+
−
            *errorMessage = QString::fromLatin1("Failed to attach type library to binary %1 - could not update file.").arg(applicationName);+
−
        return false;+
−
    }+
−
+
−
    if (!EndUpdateResource(hExe,false)) {+
−
        if (errorMessage)+
−
            *errorMessage = QString::fromLatin1("Failed to attach type library to binary %1 - could not write file.").arg(applicationName);+
−
        return false;+
−
    }+
−
    +
−
    if (errorMessage)+
−
        *errorMessage = QString::fromLatin1("Type library attached to %1.").arg(applicationName);+
−
    return true;+
−
}+
−
+
−
static bool registerServer(const QString &input)+
−
{+
−
    bool ok = false;    +
−
    if (input.endsWith(QLatin1String(".exe"))) {+
−
        ok = runWithQtInEnvironment(quotePath(input) + QLatin1String(" -regserver"));+
−
    } else {+
−
        HMODULE hdll = LoadLibrary((const wchar_t *)input.utf16());+
−
        if (!hdll) {+
−
            fprintf(stderr, "Couldn't load library file %s\n", (const char*)input.toLocal8Bit().data());+
−
            return false;+
−
        }+
−
        typedef HRESULT(__stdcall* RegServerProc)();+
−
        RegServerProc DllRegisterServer = (RegServerProc)GetProcAddress(hdll, "DllRegisterServer");+
−
        if (!DllRegisterServer) {+
−
            fprintf(stderr, "Library file %s doesn't appear to be a COM library\n", (const char*)input.toLocal8Bit().data());+
−
            return false;+
−
        }+
−
        ok = DllRegisterServer() == S_OK;+
−
    }+
−
    return ok;+
−
}+
−
+
−
static bool unregisterServer(const QString &input)+
−
{+
−
    bool ok = false;+
−
    if (input.endsWith(QLatin1String(".exe"))) {        +
−
        ok = runWithQtInEnvironment(quotePath(input) + QLatin1String(" -unregserver"));+
−
    } else {+
−
        HMODULE hdll = LoadLibrary((const wchar_t *)input.utf16());+
−
        if (!hdll) {+
−
            fprintf(stderr, "Couldn't load library file %s\n", (const char*)input.toLocal8Bit().data());+
−
            return false;+
−
        }+
−
        typedef HRESULT(__stdcall* RegServerProc)();+
−
        RegServerProc DllUnregisterServer = (RegServerProc)GetProcAddress(hdll, "DllUnregisterServer");+
−
        if (!DllUnregisterServer) {+
−
            fprintf(stderr, "Library file %s doesn't appear to be a COM library\n", (const char*)input.toLocal8Bit().data());+
−
            return false;+
−
        }+
−
        ok = DllUnregisterServer() == S_OK;+
−
    }+
−
    return ok;+
−
}+
−
+
−
static HRESULT dumpIdl(const QString &input, const QString &idlfile, const QString &version)+
−
{+
−
    HRESULT res = E_FAIL;+
−
    +
−
    if (input.endsWith(QLatin1String(".exe"))) {+
−
        if (runWithQtInEnvironment(quotePath(input) + QLatin1String(" -dumpidl ") + idlfile + QLatin1String(" -version ") + version))+
−
            res = S_OK;+
−
    } else {+
−
        HMODULE hdll = LoadLibrary((const wchar_t *)input.utf16());+
−
        if (!hdll) {+
−
            fprintf(stderr, "Couldn't load library file %s\n", (const char*)input.toLocal8Bit().data());+
−
            return 3;+
−
        }+
−
        typedef HRESULT(__stdcall* DumpIDLProc)(const QString&, const QString&);+
−
        DumpIDLProc DumpIDL = (DumpIDLProc)GetProcAddress(hdll, "DumpIDL");+
−
        if (!DumpIDL) {+
−
            fprintf(stderr, "Couldn't resolve 'DumpIDL' symbol in %s\n", (const char*)input.toLocal8Bit().data());+
−
            return 3;+
−
        }+
−
        res = DumpIDL(idlfile, version);+
−
        FreeLibrary(hdll);+
−
    }+
−
    +
−
    return res;+
−
}+
−
+
−
static void slashify(QString &s)+
−
{+
−
    if (!s.contains(QLatin1Char('/')))+
−
        return;+
−
    +
−
    int i = 0;+
−
    while (i < (int)s.length()) {+
−
        if (s[i] == QLatin1Char('/'))+
−
            s[i] = QLatin1Char('\\');+
−
        ++i;+
−
    }+
−
}+
−
+
−
int runIdc(int argc, char **argv)+
−
{+
−
    QString error;+
−
    QString tlbfile;+
−
    QString idlfile;+
−
    QString input;+
−
    QString version = QLatin1String("1.0");+
−
    +
−
    int i = 1;+
−
    while (i < argc) {+
−
        QString p = QString::fromLocal8Bit(argv[i]).toLower();+
−
        +
−
        if (p == QLatin1String("/idl") || p == QLatin1String("-idl")) {+
−
            ++i;+
−
            if (i > argc) {+
−
                error = QLatin1String("Missing name for interface definition file!");+
−
                break;+
−
            }+
−
            idlfile = QLatin1String(argv[i]);+
−
            idlfile = idlfile.trimmed().toLower();            +
−
        } else if (p == QLatin1String("/version") || p == QLatin1String("-version")) {+
−
            ++i;+
−
            if (i > argc)+
−
                version = QLatin1String("1.0");+
−
            else+
−
                version = QLatin1String(argv[i]);+
−
        } else if (p == QLatin1String("/tlb") || p == QLatin1String("-tlb")) {+
−
            ++i;+
−
            if (i > argc) {+
−
                error = QLatin1String("Missing name for type library file!");+
−
                break;+
−
            }+
−
            tlbfile = QLatin1String(argv[i]);+
−
            tlbfile = tlbfile.trimmed().toLower();            +
−
        } else if (p == QLatin1String("/v") || p == QLatin1String("-v")) {+
−
            fprintf(stdout, "Qt Interface Definition Compiler version 1.0\n");+
−
            return 0;+
−
        } else if (p == QLatin1String("/regserver") || p == QLatin1String("-regserver")) {+
−
            if (!registerServer(input)) {+
−
                fprintf(stderr, "Failed to register server!\n");+
−
                return 1;+
−
            }+
−
            fprintf(stderr, "Server registered successfully!\n");+
−
            return 0;+
−
        } else if (p == QLatin1String("/unregserver") || p == QLatin1String("-unregserver")) {+
−
            if (!unregisterServer(input)) {+
−
                fprintf(stderr, "Failed to unregister server!\n");+
−
                return 1;+
−
            }+
−
            fprintf(stderr, "Server unregistered successfully!\n");+
−
            return 0;+
−
        } else if (p[0] == QLatin1Char('/') || p[0] == QLatin1Char('-')) {+
−
            error = QLatin1String("Unknown option \"") + p + QLatin1Char('\"');+
−
            break;+
−
        } else {+
−
            input = QLatin1String(argv[i]);+
−
            input = input.trimmed().toLower();            +
−
        }+
−
        i++;+
−
    }+
−
    if (!error.isEmpty()) {+
−
        fprintf(stderr, "%s", error.toLatin1().data());+
−
        fprintf(stderr, "\n");+
−
        return 5;+
−
    }+
−
    if (input.isEmpty()) {+
−
        fprintf(stderr, "No input file specified!\n");+
−
        return 1;+
−
    }+
−
    if (input.endsWith(QLatin1String(".exe")) && tlbfile.isEmpty() && idlfile.isEmpty()) {+
−
        fprintf(stderr, "No type output file specified!\n");+
−
        return 2;+
−
    }+
−
    if (input.endsWith(QLatin1String(".dll")) && idlfile.isEmpty() && tlbfile.isEmpty()) {+
−
        fprintf(stderr, "No interface definition file and no type library file specified!\n");+
−
        return 3;+
−
    }+
−
    slashify(input);+
−
    if (!tlbfile.isEmpty()) {+
−
        slashify(tlbfile);+
−
        QFile file(tlbfile);+
−
        if (!file.open(QIODevice::ReadOnly)) {+
−
            fprintf(stderr, "Couldn't open %s for read\n", (const char*)tlbfile.toLocal8Bit().data());+
−
            return 4;+
−
        }+
−
        QByteArray data = file.readAll();+
−
        QString error;+
−
        bool ok = attachTypeLibrary(input, 1, data, &error);+
−
        fprintf(stderr, "%s", error.toLatin1().data());+
−
        fprintf(stderr, "\n");+
−
        return ok ? 0 : 4;+
−
    } else if (!idlfile.isEmpty()) {+
−
        slashify(idlfile);+
−
        idlfile = quotePath(idlfile);+
−
        fprintf(stderr, "\n\n%s\n\n", (const char*)idlfile.toLocal8Bit().data());+
−
        quotePath(input);+
−
        HRESULT res = dumpIdl(input, idlfile, version);+
−
        +
−
        switch(res) {+
−
        case S_OK:+
−
            break;+
−
        case E_FAIL:+
−
            fprintf(stderr, "IDL generation failed trying to run program %s!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
        case -1:+
−
            fprintf(stderr, "Couldn't open %s for writing!\n", (const char*)idlfile.toLocal8Bit().data());+
−
            return res;+
−
        case 1:+
−
            fprintf(stderr, "Malformed appID value in %s!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
        case 2:+
−
            fprintf(stderr, "Malformed typeLibID value in %s!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
        case 3:+
−
            fprintf(stderr, "Class has no metaobject information (error in %s)!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
        case 4:+
−
            fprintf(stderr, "Malformed classID value in %s!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
        case 5:+
−
            fprintf(stderr, "Malformed interfaceID value in %s!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
        case 6:+
−
            fprintf(stderr, "Malformed eventsID value in %s!\n", (const char*)input.toLocal8Bit().data());+
−
            return res;+
−
            +
−
        default:+
−
            fprintf(stderr, "Unknown error writing IDL from %s\n", (const char*)input.toLocal8Bit().data());+
−
            return 7;+
−
        }+
−
    }+
−
    return 0;+
−
}+
−
+
−
QT_END_NAMESPACE+
−
+
−
int main(int argc, char **argv)+
−
{+
−
    return QT_PREPEND_NAMESPACE(runIdc)(argc, argv);+
−
}+
−