diff options
Diffstat (limited to 'util/install/keygen/main.cpp')
-rw-r--r-- | util/install/keygen/main.cpp | 250 |
1 files changed, 250 insertions, 0 deletions
diff --git a/util/install/keygen/main.cpp b/util/install/keygen/main.cpp new file mode 100644 index 0000000..b6eee41 --- /dev/null +++ b/util/install/keygen/main.cpp @@ -0,0 +1,250 @@ +/**************************************************************************** +** +** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies). +** Contact: Qt Software Information (qt-info@nokia.com) +** +** This file is part of the utils 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 either Technology Preview License Agreement or the +** Beta Release License Agreement. +** +** 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.0, included in the file LGPL_EXCEPTION.txt in this +** package. +** +** GNU General Public License Usage +** Alternatively, this file may be used under the terms of the GNU +** General Public License version 3.0 as published by the Free Software +** Foundation and appearing in the file LICENSE.GPL included in the +** packaging of this file. Please review the following information to +** ensure the GNU General Public License version 3.0 requirements will be +** met: http://www.gnu.org/copyleft/gpl.html. +** +** If you are unsure which license is appropriate for your use, please +** contact the sales department at qt-sales@nokia.com. +** $QT_END_LICENSE$ +** +****************************************************************************/ + +#include <qfile.h> + +#include <stdio.h> +#include <stdlib.h> + +#include "keyinfo.h" + +static void printNewKey( uint features, const QDate& expiryDate ) +{ + QFile in; + QFile out; + QString fn; + char block[10]; + + fn.sprintf( "next.%.2x", features ); + in.setFileName( fn ); + if ( !in.open(QFile::ReadOnly) ) + return; + int ent = QString( in.readAll() ).toInt(); + in.close(); + + fn.sprintf( "table.%.2x", features ); + in.setFileName( fn ); + if ( !in.open(QFile::ReadOnly) ) + return; + in.seek( ent * 10 ); + in.read( block, 9 ); + block[9] = '\0'; + in.close(); + + ent++; + if ( ent == (1 << NumRandomBits) ) + ent = 1; // skip first entry + + fn.sprintf( "next.%.2x", features ); + out.setFileName( fn ); + if ( !out.open(QFile::WriteOnly) ) + return; + QString s = QString::number( ent ) + QChar( '\n' ); + out.write( s.toLocal8Bit(), s.length() ); + out.close(); + + if ( strlen(block) == 9 && block[4] == '-' ) + printf( "%s%s\n", block, qPrintable(encodedExpiryDate(expiryDate)) ); +} + +static QString textForFeatures( uint features ) +{ + QString text; + + if ( (features & Feature_US) != 0 ) + text += QString( " -us" ); + if ( (features & Feature_Enterprise) != 0 ) + text += QString( " -enterprise" ); + if ( (features & Feature_Unix) != 0 ) + text += QString( " -unix" ); + if ( (features & Feature_Windows) != 0 ) + text += QString( " -windows" ); + if ( (features & Feature_Mac) != 0 ) + text += QString( " -mac" ); + if ( (features & Feature_Embedded) != 0 ) + text += QString( " -embedded" ); + if ( (features & Feature_Extra1) != 0 ) + text += QString( " -extra1" ); + if ( (features & Feature_Extra2) != 0 ) + text += QString( " -extra2" ); + + text = text.trimmed(); + if ( text.isEmpty() ) + text = QString( "invalid key" ); + return text; +} + +static QString textForDate( const QDate& date ) +{ + if ( date.isValid() ) { + return date.toString( Qt::ISODate ); + } else { + return QString( "invalid date" ); + } +} + +static void reset() +{ + for ( uint features = 0; features < (1 << NumFeatures); features++ ) { + printf( "Resetting 'table.%.2x' and 'next.%.2x' (%s)\n", features, + features, qPrintable(textForFeatures(features)) ); + + QFile out; + QString fn; + + fn.sprintf( "table.%.2x", features ); + out.setFileName( fn ); + if ( !out.open(QFile::WriteOnly) ) { + fprintf( stderr, "Cannot open '%s' for writing\n", qPrintable(fn) ); + exit( EXIT_FAILURE ); + } + + for ( uint bits = 0; bits < (1 << NumRandomBits); bits++ ) { + QString k = keyForFeatures( features, bits ) + QChar( '\n' ); + out.write( k.toLatin1(), k.length() ); + + /* + We check that the generated keys give access to the + correct feature sets. This accounts for most of the + processing time of the function. + */ +#if 1 + if ( featuresForKey(k) != features ) { + fprintf( stderr, "Internal error in featuresForKey(\"%s\")\n", + qPrintable(k.trimmed()) ); + exit( EXIT_FAILURE ); + } + if ( (features & ~(Feature_US | Feature_Enterprise | + Feature_Unix)) == 0 ) { + if ( featuresForKeyOnUnix(k) != features ) { + fprintf( stderr, + "Internal error in featuresForKeyOnUnix(\"%s\")\n", + qPrintable(k.trimmed()) ); + exit( EXIT_FAILURE ); + } + } +#endif + } + out.close(); + + fn.sprintf( "next.%.2x", features ); + out.setFileName( fn ); + if ( !out.open(QFile::WriteOnly) ) { + fprintf( stderr, "Cannot open '%s' for writing\n", qPrintable(fn) ); + exit( EXIT_FAILURE ); + } + out.write( "1\n", 2 ); // skip first key + out.close(); + } +} + +int main( int argc, char **argv ) +{ + if ( argc == 1 || (strcmp(argv[1], "check") != 0 && + strcmp(argv[1], "new") != 0 && + strcmp(argv[1], "reset") != 0) ) { + fprintf( stderr, "Usage:\n" + " keygen check <key>\n" + " keygen new YYYY-MM-DD [-us] [-enterprise] [-unix]\n" + " [-windows] [-mac] [-embedded] [-extra1] [-extra2]\n" + " keygen reset\n" ); + exit( EXIT_FAILURE ); + } + + if ( strcmp(argv[1], "check") == 0 ) { + if ( argc != 3 ) { + fprintf( stderr, "Usage:\n" + " mkcode check <key>\n" ); + exit( EXIT_FAILURE ); + } + + QString key( argv[2] ); + + printf("Unix check: %s\n", + qPrintable(textForFeatures(featuresForKeyOnUnix(QString(argv[2]))))); + printf("Full check: %s\n", + qPrintable(textForFeatures(featuresForKey(QString(argv[2]))))); + if ( featuresForKey(QString(argv[2])) != 0 ) + printf("Expiry date: %s\n", + qPrintable(textForDate(decodedExpiryDate(key.mid(9))))); + } else if ( strcmp(argv[1], "new") == 0 ) { + uint features = 0; + + if ( argc < 3 ) { + fprintf( stderr, "Usage:\n" + " mkcode new YYYY-MM-DD [features]\n" ); + exit( EXIT_FAILURE ); + } + + QDate expiryDate = QDate::fromString( QString(argv[2]), Qt::ISODate ); + if ( !expiryDate.isValid() ) { + fprintf( stderr, "Date '%s' not in YYYY-MM-DD format\n", argv[2] ); + exit( EXIT_FAILURE ); + } + + for ( int i = 3; i < argc; i++ ) { + if ( strcmp(argv[i], "-us") == 0 ) { + features |= Feature_US; + } else if ( strcmp(argv[i], "-enterprise") == 0 ) { + features |= Feature_Enterprise; + } else if ( strcmp(argv[i], "-unix") == 0 ) { + features |= Feature_Unix; + } else if ( strcmp(argv[i], "-windows") == 0 ) { + features |= Feature_Windows; + } else if ( strcmp(argv[i], "-mac") == 0 ) { + features |= Feature_Mac; + } else if ( strcmp(argv[i], "-embedded") == 0 ) { + features |= Feature_Embedded; + } else if ( strcmp(argv[i], "-extra1") == 0 ) { + features |= Feature_Extra1; + } else if ( strcmp(argv[i], "-extra2") == 0 ) { + features |= Feature_Extra2; + } else { + fprintf( stderr, "Unknown flag '%s'\n", argv[i] ); + exit( EXIT_FAILURE ); + } + } + printNewKey( features, expiryDate ); + } else { + reset(); + } + return 0; +} |