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