The Tor Browser / file revision

 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

 /* This Source Code Form is subject to the terms of the Mozilla Public

  * License, v. 2.0. If a copy of the MPL was not distributed with this

  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

 // Utility that converts file encoded in one charset codepage to

 // another encoding

 #include "nscore.h"

 #include "nsString.h"

 #include "nsIServiceManager.h"

 #include "nsICharsetConverterManager.h"

 #include "nsIUnicodeEncoder.h"

 #include "nsIUnicodeDecoder.h"

 static NS_DEFINE_CID(kCharsetConverterManagerCID, NS_ICHARSETCONVERTERMANAGER_CID);

 #include <stdio.h>

 #include <string.h>

 #include <stdlib.h>

 void usage()

 {

   printf(

     "nsconv -f fromcode -t tocode infile outfile\n"

     "nsconv -f fromcode -t tocode infile > outfile\n"

     "nsconv -f fromcode -t tocode < infile > outfile\n"

     );

 }

 #define INBUFSIZE (1024*16)

 #define MEDBUFSIZE (1024*16*2)

 #define OUTBUFSIZE (1024*16*8)

 char inbuffer[INBUFSIZE];

 char outbuffer[OUTBUFSIZE];

 char16_t  medbuffer[MEDBUFSIZE];

 int main(int argc, const char** argv)

 {

   nsIUnicodeEncoder* encoder = nullptr;

   nsIUnicodeDecoder* decoder = nullptr;

   FILE* fin = 0;

   FILE* fout = 0;

   FILE* infile = 0;

   FILE* outfile = 0;

   nsresult res= NS_OK;

   NS_InitXPCOM2(nullptr, nullptr, nullptr);

   // get ccMain;

   nsCOMPtr<nsICharsetConverterManager> ccMain =

       do_GetService(kCharsetConverterManagerCID, &res);

   if(NS_FAILED(res))

   {

     fprintf(stderr, "Cannot get Character Converter Manager %x\n", res);

     return -1;

   }

   int i;

   if(argc > 4)

   {

     for(i =0; i < argc; i++)

     {

       if(strcmp(argv[i], "-f") == 0)

       {

         // User has specified the charset to convert from

         nsAutoCString str;

         // First check if a charset alias was given, 

         // and convert to the canonical name

         res = ccMain->GetCharsetAlias(argv[i+1], str);

         if (NS_FAILED(res))

         {

           fprintf(stderr, "Cannot get charset alias for %s %x\n",

                   argv[i+1], res);

           goto error_exit;

         }

         // Finally create the decoder

         res = ccMain->GetUnicodeDecoder(str.get(), &decoder);

         if(NS_FAILED(res)) {

           fprintf(stderr, "Cannot get Unicode decoder %s %x\n", 

                   argv[i+1],res);

           goto error_exit;

         }

       }

       if(strcmp(argv[i], "-t") == 0)

       {

         // User has specified which charset to convert to

         nsAutoCString str;

         // First check if a charset alias was given, 

         // and convert to the canonical name

         res = ccMain->GetCharsetAlias(argv[i+1], str);

         if (NS_FAILED(res))

         {

           fprintf(stderr, "Cannot get charset alias for %s %x\n",

                   argv[i+1], res);

           goto error_exit;

         }

         // Finally create the encoder 

         res = ccMain->GetUnicodeEncoderRaw(str.get(), &encoder);

         if(NS_FAILED(res)) {

           fprintf(stderr, "Cannot get Unicode encoder %s %x\n", 

                   argv[i+1],res);

           goto error_exit;

         }

       }

     }

     if (argc > 5)

     {

       // The user has specified an input file 

       // if we have more than four arguments

       fin = infile = fopen(argv[5], "rb");

       if(!infile) 

       {  

         usage();

         fprintf(stderr,"cannot open input file %s\n", argv[5]);

         goto error_exit; 

       }

       if (argc > 6)

       {

         // The user has specified an output file

         // if we have more than four arguments

         fout = outfile = fopen(argv[6], "ab");

         if(!outfile) 

         {  

           usage();

           fprintf(stderr,"cannot open output file %s\n", argv[6]);

           goto error_exit; 

         }

       }

       else

         fout = stdout;

     }

     else

     {

       // No inputfiles are given. Read and write

       // to/from standard in and standard out

       fin = stdin;

       fout = stdout;

     }

     int32_t insize,medsize,outsize;

     while((insize=fread(inbuffer, 1,INBUFSIZE, fin)) > 0)

     {

       medsize=MEDBUFSIZE;

       res = decoder->Convert(inbuffer,&insize, medbuffer, &medsize);

       if(NS_FAILED(res)) {

         fprintf(stderr, "failed in decoder->Convert %x\n",res);

         goto error_exit;

       }

       outsize = OUTBUFSIZE;

       res = encoder->Convert(medbuffer, &medsize, outbuffer,&outsize);

       if(NS_FAILED(res)) {

         fprintf(stderr, "failed in encoder->Convert %x\n",res);

         goto error_exit;

       }

       fwrite(outbuffer, 1, outsize, fout);

     }

     // Clean up 

     if (infile != 0)

       fclose(infile);

     if (outfile != 0)

       fclose(outfile);

     fprintf(stderr, "Done!\n");

     NS_IF_RELEASE(encoder);

     NS_IF_RELEASE(decoder);

     return 0;

   }

   usage();

   error_exit:

   // Clean up after

   if (infile != 0)

     fclose(infile);

   if (outfile != 0)

     fclose(outfile);

   NS_IF_RELEASE(encoder);

   NS_IF_RELEASE(decoder);

   return -1;

 }