joachim99@77: // uninstallHelper.cpp : Defines the entry point for the console application.
joachim99@77: //
joachim99@77: #define _CRT_SECURE_NO_DEPRECATE
joachim99@77: #include "stdafx.h"
joachim99@77: #include <iostream>
joachim99@77: #include <string>
joachim99@77: #include <vector>
joachim99@77: #include <list>
joachim99@77: #include <windows.h>
joachim99@77: #include <string.h>
joachim99@77: #include <io.h>
joachim99@77: 
joachim99@77: //#define __stdcall
joachim99@77: 
joachim99@77: // For compilation download the NSIS source package and modify the following
joachim99@77: // line to point to the exdll.h-file
joachim99@77: #include "C:/Programme/NSIS/Contrib/ExDll/exdll.h"
joachim99@77: 
joachim99@77: struct ReplacementItem
joachim99@77: { char* fileType; char* operationType; };
joachim99@77: 
joachim99@77: ReplacementItem g_replacementTable[] = {
joachim99@77:    "text_file_delta",   "xcompare",
joachim99@77:    "text_file_delta",   "xmerge",
joachim99@77:    "whole_copy",        "xcompare",
joachim99@77:    "whole_copy",        "xmerge",
joachim99@77:    "z_text_file_delta", "xcompare",
joachim99@77:    "z_text_file_delta", "xmerge",
joachim99@77:    "z_whole_copy",      "xcompare",
joachim99@77:    "z_whole_copy",      "xmerge",
joachim99@77:    "_xml",              "xcompare",
joachim99@77:    "_xml",              "xmerge",
joachim99@77:    "_xml2",             "xcompare",
joachim99@77:    "_xml2",             "xmerge",
joachim99@77:    "_rftdef",           "xcompare",
joachim99@77:    "_rftmap",           "xcompare",
joachim99@77:    "_rftvp",            "xcompare",
joachim99@77:    "_xtools",           "xcompare",
joachim99@77:    0,0
joachim99@77: };
joachim99@77: 
joachim99@77: struct LineItem
joachim99@77: {
joachim99@77:    std::string fileType;
joachim99@77:    std::string opType;
joachim99@77:    std::string command;
joachim99@77:    std::string fileOpPart;
joachim99@77: };
joachim99@77: 
joachim99@77: // Return true if successful, else false
joachim99@77: bool readAndParseMapFile( const std::string& filename, std::list<LineItem>& lineItemList )
joachim99@77: {
joachim99@77:    // Read file
joachim99@77:    FILE* pFile = fopen( filename.c_str(), "r" );
joachim99@77:    if (pFile)
joachim99@77:    {
joachim99@77:       fseek(pFile,0,SEEK_END);
joachim99@77:       int size = ftell(pFile);
joachim99@77:       fseek(pFile,0,SEEK_SET);
joachim99@77:       std::vector<char> buf( size );
joachim99@77:       fread( &buf[0], 1, size, pFile );
joachim99@77:       fclose( pFile );
joachim99@77: 
joachim99@77:       // Replace strings
joachim99@77:       int lineStartPos=0;
joachim99@77:       int wordInLine = 0;
joachim99@77:       LineItem lineItem;
joachim99@77:       for( int i=0; i<size; )
joachim99@77:       {
joachim99@77:          if( buf[i] == '\n' || buf[i] == '\r' )
joachim99@77:          {
joachim99@77:             ++i;
joachim99@77:             wordInLine = 0;
joachim99@77:             lineStartPos = i;
joachim99@77:             continue;
joachim99@77:          }
joachim99@77:          if( buf[i] == ' ' || buf[i] == '\t' )
joachim99@77:          {
joachim99@77:             ++i;
joachim99@77:             continue;
joachim99@77:          }
joachim99@77:          else
joachim99@77:          {
joachim99@77:             int wordStartPos = i;
joachim99@77:             if (wordInLine<2)
joachim99@77:             {
joachim99@77:                while ( i<size && !( buf[i] == ' ' || buf[i] == '\t' ) )
joachim99@77:                   ++i;
joachim99@77: 
joachim99@77:                std::string word( &buf[wordStartPos], i-wordStartPos );
joachim99@77:                if (wordInLine==0)
joachim99@77:                   lineItem.fileType = word;
joachim99@77:                else
joachim99@77:                   lineItem.opType = word;
joachim99@77:                ++wordInLine;
joachim99@77:             }
joachim99@77:             else
joachim99@77:             {
joachim99@77:                lineItem.fileOpPart = std::string( &buf[lineStartPos], i-lineStartPos );
joachim99@77:                while ( i<size && !( buf[i] == '\n' || buf[i] == '\r' ) )
joachim99@77:                   ++i;
joachim99@77: 
joachim99@77:                std::string word( &buf[wordStartPos], i-wordStartPos );
joachim99@77:                lineItem.command = word;
joachim99@77:                lineItemList.push_back( lineItem );
joachim99@77:             }
joachim99@77:          }
joachim99@77:       }
joachim99@77:    }
joachim99@77:    else
joachim99@77:    {
joachim99@77:       return false;
joachim99@77:    }
joachim99@77:    return true;
joachim99@77: }
joachim99@77: 
joachim99@77: bool writeMapFile( const std::string& filename, const std::list<LineItem>& lineItemList )
joachim99@77: {
joachim99@77:    FILE* pFile = fopen( filename.c_str(), "w" );
joachim99@77:    if (pFile)
joachim99@77:    {
joachim99@77:       std::list<LineItem>::const_iterator i = lineItemList.begin();
joachim99@77:       for( ; i!=lineItemList.end(); ++i )
joachim99@77:       {
joachim99@77:          const LineItem& li = *i;
joachim99@77:          fprintf( pFile, "%s%s\n", li.fileOpPart.c_str(), li.command.c_str() );
joachim99@77:       }
joachim99@77:       fclose( pFile );
joachim99@77:    }
joachim99@77:    else
joachim99@77:    {
joachim99@77:       return false;
joachim99@77:    }
joachim99@77:    return true;
joachim99@77: }
joachim99@77: 
joachim99@77: std::string toUpper( const std::string& s )
joachim99@77: {
joachim99@77:    std::string s2 = s;
joachim99@77: 
joachim99@77:    for( unsigned int i=0; i<s.length(); ++i )
joachim99@77:    {
joachim99@77:       s2[i] = toupper( s2[i] );
joachim99@77:    }
joachim99@77:    return s2;
joachim99@77: }
joachim99@77: 
joachim99@77: int install( const _TCHAR* subCommand, const _TCHAR* kdiff3CommandPath )
joachim99@77: {
joachim99@77:    std::string installCommand = subCommand; // "install" or "uninstall"
joachim99@77:    std::string kdiff3Command  = kdiff3CommandPath;
joachim99@77: 
joachim99@77:    /*
joachim99@77:    std::wstring installCommand = subCommand; // "install" or "uninstall"
joachim99@77:    std::wstring wKDiff3Command  = kdiff3CommandPath;
joachim99@77:    std::string kdiff3Command;
joachim99@77:    kdiff3Command.reserve( wKDiff3Command.length()+1 );
joachim99@77:    kdiff3Command.resize( wKDiff3Command.length() );
joachim99@77:    BOOL bUsedDefaultChar = FALSE;
joachim99@77:    int successLen = WideCharToMultiByte(  CP_ACP, 0, 
joachim99@77:       wKDiff3Command.c_str(), int(wKDiff3Command.length()),
joachim99@77:       &kdiff3Command[0], int(kdiff3Command.length()), 0, &bUsedDefaultChar );
joachim99@77: 
joachim99@77:    if ( successLen != kdiff3Command.length() || bUsedDefaultChar )
joachim99@77:    {
joachim99@77:       std::cerr << "KDiff3 command contains characters that don't map to ansi code page.\n"
joachim99@77:           "Aborting clearcase installation.\n"
joachim99@77:           "Try to install KDiff3 in another path that doesn't require special characters.\n";
joachim99@77:       return -1;
joachim99@77:    }
joachim99@77:    */
joachim99@77: 
joachim99@77:    // Try to locate cleartool, the clearcase tool in the path
joachim99@77:    char buffer[1000];
joachim99@77:    char* pLastPart = 0;
joachim99@77:    int len = SearchPathA(0, "cleartool.exe", 0, sizeof(buffer)/sizeof(buffer[0]), 
joachim99@77:                          buffer, &pLastPart );
joachim99@77:    if ( len+1<int(sizeof(buffer)/sizeof(buffer[0])) && pLastPart )
joachim99@77:    {
joachim99@77:       pLastPart[-1] = 0;
joachim99@77:       pLastPart = strchr( buffer, '\\' ); // cd up (because cleartool.exe is in bin subdir)
joachim99@77:       if ( pLastPart )
joachim99@77:          pLastPart[1]=0;
joachim99@77: 
joachim99@77:       std::string path( buffer );
joachim99@77:       path += _T("lib\\mgrs\\map");
joachim99@77:       std::string bakName = path + _T(".preKDiff3Install");
joachim99@77:       
joachim99@77:       if ( installCommand == _T("install") )
joachim99@77:       {
joachim99@77:          std::list<LineItem> lineItemList;
joachim99@77:          bool bSuccess = readAndParseMapFile( path, lineItemList );
joachim99@77:          if ( !bSuccess )
joachim99@77:          {
joachim99@77:             std::cerr << "Error reading original map file.\n";
joachim99@77:             return -1;
joachim99@77:          }
joachim99@77: 
joachim99@77:          // Create backup
joachim99@77:          if ( access( bakName.c_str(), 0 )!=0 ) // Create backup only if not exists yet
joachim99@77:          {
joachim99@77:             if ( rename( path.c_str(), bakName.c_str() ) )
joachim99@77:             {
joachim99@77:                std::cerr << "Error renaming original map file.\n";
joachim99@77:                return -1;
joachim99@77:             }
joachim99@77:          }
joachim99@77: 
joachim99@77:          std::list<LineItem>::iterator i = lineItemList.begin();
joachim99@77:          for( ; i!=lineItemList.end(); ++i )
joachim99@77:          {
joachim99@77:             LineItem& li = *i;
joachim99@77:             for (int j=0;;++j)
joachim99@77:             {
joachim99@77:                ReplacementItem& ri = g_replacementTable[j];
joachim99@77:                if ( ri.fileType==0 || ri.operationType==0 )
joachim99@77:                   break;
joachim99@77:                if ( li.fileType == ri.fileType && li.opType == ri.operationType )
joachim99@77:                {
joachim99@77:                   li.command = kdiff3Command.c_str();
joachim99@77:                   break;
joachim99@77:                }
joachim99@77:             }
joachim99@77:          }
joachim99@77: 
joachim99@77:          bSuccess = writeMapFile( path, lineItemList );
joachim99@77:          if ( !bSuccess )
joachim99@77:          {
joachim99@77:             if ( rename( bakName.c_str(), path.c_str() ) )
joachim99@77:                std::cerr << "Error writing new map file, restoring old file also failed.\n";
joachim99@77:             else
joachim99@77:                std::cerr << "Error writing new map file, old file restored.\n";
joachim99@77: 
joachim99@77:             return -1;
joachim99@77:          }
joachim99@77:       }
joachim99@77:       else if ( installCommand == _T("uninstall") )
joachim99@77:       {
joachim99@77:          std::list<LineItem> lineItemList;
joachim99@77:          bool bSuccess = readAndParseMapFile( path, lineItemList );
joachim99@77:          if ( !bSuccess )
joachim99@77:          {
joachim99@77:             std::cerr << "Error reading original map file\n.";
joachim99@77:             return -1;
joachim99@77:          }
joachim99@77: 
joachim99@77:          std::list<LineItem> lineItemListBak;
joachim99@77:          bSuccess = readAndParseMapFile( bakName, lineItemListBak );
joachim99@77:          if ( !bSuccess )
joachim99@77:          {
joachim99@77:             std::cerr << "Error reading backup map file.\n";
joachim99@77:             return -1;
joachim99@77:          }
joachim99@77: 
joachim99@77:          std::list<LineItem>::iterator i = lineItemList.begin();
joachim99@77:          for( ; i!=lineItemList.end(); ++i )
joachim99@77:          {
joachim99@77:             LineItem& li = *i;
joachim99@77:             if ((int)toUpper(li.command).find("KDIFF3")>=0)
joachim99@77:             {
joachim99@77:                std::list<LineItem>::const_iterator j = lineItemListBak.begin();
joachim99@77:                for (;j!=lineItemListBak.end();++j)
joachim99@77:                {
joachim99@77:                   const LineItem& bi = *j;  // backup iterator
joachim99@77:                   if ( li.fileType == bi.fileType && li.opType == bi.opType )
joachim99@77:                   {
joachim99@77:                      li.command = bi.command;
joachim99@77:                      break;
joachim99@77:                   }
joachim99@77:                }
joachim99@77:             }
joachim99@77:          }
joachim99@77: 
joachim99@77:          bSuccess = writeMapFile( path, lineItemList );
joachim99@77:          if ( !bSuccess )
joachim99@77:          {
joachim99@77:             std::cerr << "Error writing map file.";
joachim99@77: 
joachim99@77:             return -1;
joachim99@77:          }
joachim99@77:       }
joachim99@77:    }
joachim99@77:    return 0;
joachim99@77: }
joachim99@77: 
joachim99@77: extern "C"
joachim99@77: void __declspec(dllexport) nsisPlugin(HWND hwndParent, int string_size, 
joachim99@77:                                       char *variables, stack_t **stacktop,
joachim99@77:                                       extra_parameters *extra)
joachim99@77: {
joachim99@77:    //g_hwndParent=hwndParent;
joachim99@77: 
joachim99@77:    EXDLL_INIT();
joachim99@77:    {
joachim99@77:       std::string param1( g_stringsize, ' ' );
joachim99@77:       int retVal = popstring( &param1[0] );
joachim99@77:       if ( retVal == 0 )
joachim99@77:       {
joachim99@77:          std::string param2( g_stringsize, ' ' );
joachim99@77:          retVal = popstring( &param2[0] );
joachim99@77:          if ( retVal == 0 )
joachim99@77:             install( param1.c_str(), param2.c_str() );
joachim99@77:          return;
joachim99@77:       }
joachim99@77:       std::cerr << "Not enough parameters." << std::endl;
joachim99@77:    }
joachim99@77: }
joachim99@77: 
joachim99@77: /*
joachim99@77: int _tmain(int argc, _TCHAR* argv[])
joachim99@77: {
joachim99@77:    if ( argc<3 ) 
joachim99@77:    {
joachim99@77:       std::cout << "This program is needed to install/uninstall KDiff3 for clearcase.\n"
joachim99@77:                    "It tries to patch the map file (clearcase-subdir\\lib\\mgrs\\map)\n"
joachim99@77:                    "Usage 1: ccInstHelper install pathToKdiff3.exe\n"
joachim99@77:                    "Usage 2: ccInstHelper uninstall pathToKdiff3.exe\n"
joachim99@77:                    "Backups of the original map files are created in the dir of the map file.\n";
joachim99@77:    }
joachim99@77:    else
joachim99@77:    {
joachim99@77:       return install( argv[1], argv[2] );
joachim99@77:    }
joachim99@77:    
joachim99@77:    return 0;
joachim99@77: }
joachim99@77: */