registry/tools/regmerge.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: regmerge.cxx,v $
  10  * $Revision: 1.7 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 // MARKER(update_precomp.py): autogen include statement, do not remove
  32 #include "precompiled_registry.hxx"
  33
  34 #include <stdio.h>
  35 #include <string.h>
  36
  37 #include "registry/registry.h"
  38 #include        <rtl/ustring.hxx>
  39 #include        <rtl/alloc.h>
  40 #include <osl/process.h>
  41 #include <osl/diagnose.h>
  42 #include <osl/thread.h>
  43 #include <osl/file.hxx>
  44
  45 #ifdef SAL_UNX
  46 #define SEPARATOR '/'
  47 #else
  48 #define SEPARATOR '\\'
  49 #endif
  50
  51 using namespace ::rtl;
  52 using namespace ::osl;
  53
  54 sal_Bool isFileUrl(const OString& fileName)
  55 {
  56     if (fileName.indexOf("file://") == 0 )
  57         return sal_True;
  58     return sal_False;
  59 }
  60
  61 OUString convertToFileUrl(const OString& fileName)
  62 {
  63     if ( isFileUrl(fileName) )
  64     {
  65         return OStringToOUString(fileName, osl_getThreadTextEncoding());
  66     }
  67
  68     OUString uUrlFileName;
  69     OUString uFileName(fileName.getStr(), fileName.getLength(), osl_getThreadTextEncoding());
  70     if ( fileName.indexOf('.') == 0 || fileName.indexOf(SEPARATOR) < 0 )
  71     {
  72         OUString uWorkingDir;
  73         if (osl_getProcessWorkingDir(&uWorkingDir.pData) != osl_Process_E_None)
  74         {
  75             OSL_ASSERT(false);
  76         }
  77         if (FileBase::getAbsoluteFileURL(uWorkingDir, uFileName, uUrlFileName)
  78             != FileBase::E_None)
  79         {
  80             OSL_ASSERT(false);
  81         }
  82     } else
  83     {
  84         if (FileBase::getFileURLFromSystemPath(uFileName, uUrlFileName)
  85             != FileBase::E_None)
  86         {
  87             OSL_ASSERT(false);
  88         }
  89     }
  90
  91     return uUrlFileName;
  92 }
  93
  94 int realargc;
  95 char* realargv[2048];
  96
  97 static void dumpHelp()
  98 {
  99     fprintf(stderr, "using: regmerge [-v|--verbose] mergefile mergeKeyName regfile_1 ... regfile_n\n");
 100     fprintf(stderr, "       regmerge @regcmds\nOptions:\n");
 101     fprintf(stderr, "  -v, --verbose : verbose output on stdout.\n");
 102     fprintf(stderr, "  mergefile     : specifies the merged registry file. If this file doesn't exists,\n");
 103     fprintf(stderr, "                  it is created.\n");
 104     fprintf(stderr, "  mergeKeyName  : specifies the merge key, everything is merged under this key.\n");
 105     fprintf(stderr, "                  If this key doesn't exists, it is created.\n");
 106     fprintf(stderr, "  regfile_1..n  : specifies one or more registry files which are merged.\n");
 107 }
 108
 109 static bool checkCommandFile(char* cmdfile)
 110 {
 111     FILE        *commandfile;
 112     char        option[256];
 113     bool    bVerbose = false;
 114
 115     commandfile = fopen(cmdfile+1, "r");
 116     if( commandfile == NULL )
 117     {
 118         fprintf(stderr, "ERROR: Can't open command file \"%s\"\n", cmdfile);
 119     } else
 120     {
 121         while ( fscanf(commandfile, "%s", option) != EOF )
 122         {
 123             if (option[0]== '@')
 124             {
 125                 bool bRet = checkCommandFile(option);
 126                 // ensure that the option will not be overwritten
 127                 if ( !bRet )
 128                     bVerbose = bRet;
 129             } else {
 130                 if (option[0]== '-') {
 131                     if (strncmp(option, "-v", 2)  == 0 ||
 132                         strncmp(option, "--verbose", 9) == 0)
 133                     {
 134                         bVerbose = true;
 135                     } else {
 136                         fprintf(stderr, "ERROR: unknown option \"%s\"\n", option);
 137                         dumpHelp();
 138                         exit(-1);
 139                     }
 140                 }else
 141                 {
 142                     realargv[realargc]= strdup(option);
 143                     realargc++;
 144                 }
 145             }
 146             if (realargc == 2047)
 147             {
 148                 fprintf(stderr, "ERROR: more than 2048 arguments.\n");
 149                 break;
 150             }
 151         }
 152         fclose(commandfile);
 153     }
 154
 155     return bVerbose;
 156 }
 157
 158 static bool checkCommandArgs(int argc, char **argv)
 159 {
 160     bool bVerbose = false;
 161
 162     realargc = 0;
 163
 164     for (int i=0; i<argc; i++)
 165     {
 166         if (argv[i][0]== '@')
 167         {
 168             bool bRet = checkCommandFile(argv[i]);
 169             // ensure that the option will not be overwritten
 170             if ( !bRet )
 171                 bVerbose = bRet;
 172         } else {
 173             if (argv[i][0]== '-') {
 174                 if (strncmp(argv[i], "-v", 2)  == 0 ||
 175                     strncmp(argv[i], "--verbose", 9) == 0)
 176                 {
 177                     bVerbose = true;
 178                 } else {
 179                     fprintf(stderr, "ERROR: unknown option \"%s\"\n", argv[i]);
 180                     dumpHelp();
 181                     exit(-1);
 182                 }
 183             } else
 184             {
 185                 realargv[realargc]= strdup(argv[i]);
 186                 realargc++;
 187             }
 188         }
 189     }
 190
 191     return bVerbose;
 192 }
 193
 194 static void cleanCommandArgs()
 195 {
 196     for (int i=0; i<realargc; i++)
 197     {
 198         free(realargv[i]);
 199     }
 200 }
 201
 202
 203
 204 #if (defined UNX) || (defined OS2)
 205 int main( int argc, char * argv[] )
 206 #else
 207 int _cdecl main( int argc, char * argv[] )
 208 #endif
 209 {
 210     RegHandle           hReg;
 211     RegKeyHandle        hRootKey;
 212     bool            bVerbose = checkCommandArgs(argc, argv);
 213
 214     if (realargc < 4)
 215     {
 216         dumpHelp();
 217         cleanCommandArgs();
 218         exit(1);
 219     }
 220
 221     ::rtl::OUString regName( convertToFileUrl(realargv[1]) );
 222     if (reg_openRegistry(regName.pData, &hReg, REG_READWRITE))
 223     {
 224         if (reg_createRegistry(regName.pData, &hReg))
 225         {
 226             if (bVerbose)
 227                 fprintf(stderr, "open registry \"%s\" failed\n", realargv[1]);
 228             cleanCommandArgs();
 229             exit(-1);
 230         }
 231     }
 232
 233     if (!reg_openRootKey(hReg, &hRootKey))
 234     {
 235         ::rtl::OUString mergeKeyName( ::rtl::OUString::createFromAscii(realargv[2]) );
 236         ::rtl::OUString targetRegName;
 237         for (int i = 3; i < realargc; i++)
 238         {
 239             targetRegName = convertToFileUrl(realargv[i]);
 240             RegError _ret = reg_mergeKey(
 241                 hRootKey, mergeKeyName.pData, targetRegName.pData, sal_False,
 242                 bVerbose);
 243             if (_ret)
 244             {
 245                 if (_ret == REG_MERGE_CONFLICT)
 246                 {
 247                     if (bVerbose)
 248                         fprintf(stderr, "merging registry \"%s\" under key \"%s\" in registry \"%s\".\n",
 249                                 realargv[i], realargv[2], realargv[1]);
 250                 } else
 251                 {
 252                     if (bVerbose)
 253                         fprintf(stderr, "ERROR: merging registry \"%s\" under key \"%s\" in registry \"%s\" failed.\n",
 254                                 realargv[i], realargv[2], realargv[1]);
 255                     exit(-2);
 256                 }
 257             } else
 258             {
 259                 if (bVerbose)
 260                     fprintf(stderr, "merging registry \"%s\" under key \"%s\" in registry \"%s\".\n",
 261                             realargv[i], realargv[2], realargv[1]);
 262             }
 263         }
 264
 265         if (reg_closeKey(hRootKey))
 266         {
 267             if (bVerbose)
 268                 fprintf(stderr, "closing root key of registry \"%s\" failed\n",
 269                         realargv[1]);
 270             exit(-3);
 271         }
 272     } else
 273     {
 274         if (bVerbose)
 275             fprintf(stderr, "open root key of registry \"%s\" failed\n",
 276                     realargv[1]);
 277         exit(-4);
 278     }
 279
 280     if (reg_closeRegistry(hReg))
 281     {
 282         if (bVerbose)
 283             fprintf(stderr, "closing registry \"%s\" failed\n", realargv[1]);
 284         cleanCommandArgs();
 285         exit(-5);
 286     }
 287
 288     cleanCommandArgs();
 289     return(0);
 290 }
 291
 292