cpputools/source/regcomplazy/regcomplazy.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: regcomplazy.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 #include <stdio.h>
  32
  33 #include "sal/main.h"
  34 #include <osl/diagnose.h>
  35 #include <osl/thread.h>
  36 #include <osl/file.hxx>
  37 #include <rtl/strbuf.hxx>
  38 #include <rtl/ustrbuf.hxx>
  39
  40 #include <vector>
  41
  42 #include <registry/registry.hxx>
  43
  44
  45 #define OUSTR(x) ::rtl::OUString::createFromAscii( x )
  46 #define OSToOUS(x) ::rtl::OStringToOUString(x, osl_getThreadTextEncoding())
  47 #define OUSToOS(x) ::rtl::OUStringToOString(x, osl_getThreadTextEncoding())
  48 using namespace ::rtl;
  49
  50 typedef ::std::vector< ::rtl::OString > OSVector;
  51
  52 typedef ::std::pair< ::rtl::OString, OSVector > DataPair;
  53
  54 typedef ::std::vector< DataPair > DataVector;
  55
  56 struct CompDescriptor {
  57     OString sImplementationName;
  58     OString sComponentName;
  59     OString sLoaderName;
  60     OSVector vSupportedServices;
  61     DataVector vData;
  62 };
  63
  64 typedef ::std::vector< CompDescriptor > CDescrVector;
  65
  66 static void print_options() SAL_THROW( () )
  67 {
  68     printf(
  69         "\nusage: regcomplazy [-v]registry_file cmp_descr_file ...\n\n"
  70         "Register a cmponent using a comp description file.\n"
  71         "Option -v prints verbose output on stdout.\n" );
  72 }
  73
  74 static bool checkImplValue(RegistryValueList<sal_Char*>* pValueList, OString sImplName) {
  75     for (sal_uInt32 i=0; i < pValueList->getLength(); i++) {
  76         if (sImplName.equals(pValueList->getElement(i)))
  77             return true;
  78     }
  79
  80     return false;
  81 }
  82
  83 //==================================================================================================
  84 SAL_IMPLEMENT_MAIN_WITH_ARGS(argc, argv)
  85 {
  86     if (argc < 3)
  87     {
  88         print_options();
  89         return 1;
  90     }
  91
  92     bool bVerbose = false;
  93     int nPos = 1;
  94     if ('-' == argv[ nPos ][ 0 ] && 'v' == argv[ nPos ][ 1 ])
  95     {
  96         if ('\0' != argv[ nPos ][ 2 ])
  97         {
  98             print_options();
  99             return 1;
 100         }
 101         bVerbose = true;
 102         ++nPos;
 103     }
 104
 105     OUString sys_path( OUSTR( argv[ nPos ] ) );
 106     OUString reg_url;
 107     oslFileError rc = osl_getFileURLFromSystemPath( sys_path.pData, &reg_url.pData );
 108     if (osl_File_E_None != rc)
 109     {
 110         if (bVerbose)
 111             fprintf( stderr, "\nERROR: cannot make file url out of %s\n", argv[nPos]);
 112         return 1;
 113     }
 114
 115     FILE* fDescr = fopen(argv[ ++nPos ], "r");
 116     OStringBuffer sBuffer(512);
 117
 118     if ( fDescr) {
 119         size_t totalSize = 0;
 120         size_t readSize  = 0;
 121         char   pBuffer[513];
 122
 123         while ( !feof(fDescr) )
 124         {
 125             if ( (readSize = fread(pBuffer, 1, 512, fDescr)) > 0
 126                  && !ferror(fDescr) ) {
 127                 totalSize += readSize;
 128                 if (totalSize >= 512)
 129                     sBuffer.ensureCapacity(totalSize * 2);
 130
 131                 sBuffer.append(pBuffer, readSize);
 132             }
 133         }
 134     }
 135
 136     OString sDescr = sBuffer.makeStringAndClear();
 137     sal_Int32 nTokenIndex = 0;
 138
 139     CDescrVector vDescr;
 140     CompDescriptor descr;
 141     bool bFirst = true;
 142
 143     do {
 144         OString sTmp = sDescr.getToken(0, '\x0A', nTokenIndex);
 145         OString sToken(sTmp);
 146         if (sTmp.pData->buffer[sTmp.getLength()-1] == '\x0D')
 147             sToken = sTmp.copy(0, sTmp.getLength()-1);
 148
 149         if (sToken.indexOf("[Data]") >= 0) {
 150             do {
 151                 OString sTmp2 = sDescr.getToken(0, '\x0A', nTokenIndex);
 152                 OString sToken2(sTmp2);
 153                 if (sTmp2.pData->buffer[sTmp2.getLength()-1] == '\x0D')
 154                     sToken2 = sTmp2.copy(0, sTmp2.getLength()-1);
 155
 156                 if ((sToken2.getLength() > 0) && (sToken2.pData->buffer[0] != '[')) {
 157                     OString dataKey(sToken2.copy(0, sToken2.indexOf('=')));
 158                     OString sValues(sToken2.copy(sToken2.indexOf('=')+1));
 159                     sal_Int32 nVIndex = 0;
 160                     OSVector vValues;
 161                     do {
 162                         OString sValue = sValues.getToken(0, ';', nVIndex);
 163                         vValues.push_back(sValue);
 164                     } while (nVIndex >= 0 );
 165                     descr.vData.push_back(DataPair(dataKey, vValues));
 166                 } else
 167                     break;
 168             } while (nTokenIndex >= 0 );
 169         }
 170         if ( sToken.indexOf("[ComponentDescriptor]") >= 0) {
 171             if (bFirst)
 172                 bFirst = false;
 173             else
 174                 vDescr.push_back(descr);
 175
 176             descr = CompDescriptor();
 177         }
 178         else if ( sToken.indexOf("ImplementationName=") >= 0) {
 179             descr.sImplementationName = sToken.copy(19);
 180         }
 181         else if ( sToken.indexOf("ComponentName=") >= 0) {
 182             descr.sComponentName = sToken.copy(14);
 183         }
 184         else if ( sToken.indexOf("LoaderName=") >= 0) {
 185             descr.sLoaderName = sToken.copy(11);
 186         }
 187         else if ( (sToken.indexOf("[SupportedServices]") < 0) &&
 188                   (sToken.getLength() > 0) &&
 189                   (sToken.pData->buffer[0] != '[') ) {
 190             descr.vSupportedServices.push_back(sToken);
 191         }
 192     } while (nTokenIndex >= 0 );
 193     // insert the last descriptor
 194     vDescr.push_back(descr);
 195
 196     Registry *pReg = new Registry;
 197
 198     RegistryKey rootKey, key, subKey, serviceKey;
 199
 200     if (pReg->open(reg_url, REG_READWRITE))
 201     {
 202         if (pReg->create(reg_url))
 203         {
 204             if (bVerbose)
 205                 fprintf(stderr, "ERROR: open registry \"%s\" failed\n", argv[1]);
 206             return 1;
 207         }
 208     }
 209     if (pReg->openRootKey(rootKey)) {
 210         if (bVerbose)
 211             fprintf(stderr, "ERROR: open root key failed\n");
 212         return 1;
 213     }
 214
 215     CDescrVector::const_iterator comp_iter = vDescr.begin();
 216     do {
 217         OString sImplName = (*comp_iter).sImplementationName;
 218         OUStringBuffer sbImpl;
 219         sbImpl.appendAscii("/IMPLEMENTATIONS/");
 220         sbImpl.append(OSToOUS(sImplName));
 221         OUString sImplKeyName = sbImpl.makeStringAndClear();
 222
 223         if (rootKey.openKey(sImplKeyName, key) == REG_NO_ERROR) {
 224             if (bVerbose) {
 225                fprintf(stderr, "WARNING: implementation entry for \"%s\" already exists, existing entries are overwritten\n", sImplName.getStr());
 226             }
 227         } else {
 228             if (rootKey.createKey(sImplKeyName, key)) {
 229                 if (bVerbose) {
 230                     fprintf(stderr, "ERROR: can't create new implementation entry \"%s\".\n", sImplName.getStr());
 231                 }
 232                 return 1;
 233             }
 234         }
 235
 236         OString sLoaderName = (*comp_iter).sLoaderName;
 237         OUString usKeyName(OUSTR("UNO/ACTIVATOR"));
 238         key.createKey(usKeyName, subKey);
 239         subKey.setValue(OUString(), RG_VALUETYPE_STRING,
 240                         (sal_Char*)sLoaderName.getStr(), sLoaderName.getLength()+1);
 241
 242         OString sCompName = (*comp_iter).sComponentName;
 243         usKeyName = OUSTR("UNO/LOCATION");
 244         key.createKey(usKeyName, subKey);
 245         subKey.setValue(OUString(), RG_VALUETYPE_STRING,
 246                         (sal_Char*)sCompName.getStr(), sCompName.getLength()+1);
 247
 248         if ((*comp_iter).vData.size() > 0) {
 249             usKeyName = OUSTR("DATA");
 250             RegistryKey dataKey, valueKey;
 251             key.createKey(usKeyName, dataKey);
 252
 253             DataVector::const_iterator data_iter = ((*comp_iter).vData).begin();
 254             do {
 255                 OUString sDataKey(OSToOUS((*data_iter).first));
 256                 dataKey.createKey(sDataKey, valueKey);
 257
 258                 OSVector::const_iterator value_iter = ((*data_iter).second).begin();
 259                 int vlen = (*data_iter).second.size();
 260                 sal_Char** pValueList = (sal_Char**)rtl_allocateZeroMemory(
 261                     vlen * sizeof(sal_Char*));
 262                 int i = 0;
 263                 do {
 264                     pValueList[i] = (sal_Char*)rtl_allocateZeroMemory(
 265                         (*value_iter).getLength()+1 * sizeof(sal_Char));
 266                     rtl_copyMemory(pValueList[i], (sal_Char*)(*value_iter).getStr(),
 267                                (*value_iter).getLength()+1);
 268                     i++;
 269                     value_iter++;
 270                 } while (value_iter != (*data_iter).second.end());
 271
 272                 valueKey.setStringListValue(OUString(), pValueList, vlen);
 273
 274                 // free memory
 275                 for (i=0; i<vlen; i++)
 276                     rtl_freeMemory(pValueList[i]);
 277                 rtl_freeMemory(pValueList);
 278
 279                 data_iter++;
 280             } while (data_iter != (*comp_iter).vData.end());
 281         }
 282
 283         usKeyName = OUSTR("UNO/SERVICES");
 284         key.createKey(usKeyName, subKey);
 285
 286         rootKey.createKey(OUSTR("/SERVICES"), serviceKey);
 287
 288         OSVector::const_iterator serv_iter = ((*comp_iter).vSupportedServices).begin();
 289         OUString usServiceKeyName;
 290         do {
 291             usServiceKeyName = OSToOUS(*serv_iter);
 292             // write service key in impl section
 293             subKey.createKey(usServiceKeyName, key);
 294
 295             if (serviceKey.openKey(usServiceKeyName, key) == REG_NO_ERROR) {
 296                 RegistryValueList<sal_Char*> valueList;
 297                 serviceKey.getStringListValue(usServiceKeyName, valueList);
 298                 if ( checkImplValue(&valueList, sImplName) ) {
 299                     serv_iter++;
 300                     continue;
 301                 }
 302
 303                 sal_uInt32 nServices = valueList.getLength()+1;
 304                 sal_Char** pImplList = (sal_Char**)rtl_allocateZeroMemory(
 305                     nServices * sizeof(sal_Char*));
 306                 pImplList[0] = (sal_Char*)rtl_allocateZeroMemory(
 307                     sImplName.getLength()+1 * sizeof(sal_Char));
 308                 rtl_copyMemory(pImplList[0], (sal_Char*)sImplName.getStr(),
 309                                sImplName.getLength()+1);
 310                 for (sal_uInt32 i=0; i < valueList.getLength(); i++) {
 311                     pImplList[i+1]=valueList.getElement(i);
 312                 }
 313                 key.setStringListValue(OUString(), pImplList, nServices);
 314
 315                 // free memory
 316                 rtl_freeMemory(pImplList[0]);
 317                 rtl_freeMemory(pImplList);
 318
 319             } else {
 320                 serviceKey.createKey(usServiceKeyName, key);
 321
 322                 sal_Char* pImplList[1];
 323                 pImplList[0] = (sal_Char*)rtl_allocateZeroMemory(
 324                     sImplName.getLength()+1 * sizeof(sal_Char));
 325                 rtl_copyMemory(pImplList[0], (sal_Char*)sImplName.getStr(),
 326                                sImplName.getLength()+1);
 327                 key.setStringListValue(OUString(), pImplList, 1);
 328
 329                 // free memory
 330                 rtl_freeMemory(pImplList[0]);
 331             }
 332             serv_iter++;
 333         } while (serv_iter != (*comp_iter).vSupportedServices.end());
 334
 335         comp_iter++;
 336     } while (comp_iter != vDescr.end());
 337
 338     key.closeKey();
 339     subKey.closeKey();
 340     serviceKey.closeKey();
 341     rootKey.closeKey();
 342     pReg->close();
 343
 344     return 0;
 345 }