The Tor Browser / file comparison
comparison: intl/uconv/tests/nsconv.cpp
intl/uconv/tests/nsconv.cpp
- branch
- TOR_BUG_3246
- changeset 7
- 129ffea94266
equal
deleted
inserted
replaced
| -1:000000000000 | 0:6045ed811eee |
|---|---|
| 1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ | |
| 2 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 3 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 5 | |
| 6 // Utility that converts file encoded in one charset codepage to | |
| 7 // another encoding | |
| 8 | |
| 9 #include "nscore.h" | |
| 10 #include "nsString.h" | |
| 11 #include "nsIServiceManager.h" | |
| 12 #include "nsICharsetConverterManager.h" | |
| 13 #include "nsIUnicodeEncoder.h" | |
| 14 #include "nsIUnicodeDecoder.h" | |
| 15 | |
| 16 static NS_DEFINE_CID(kCharsetConverterManagerCID, NS_ICHARSETCONVERTERMANAGER_CID); | |
| 17 | |
| 18 #include <stdio.h> | |
| 19 #include <string.h> | |
| 20 #include <stdlib.h> | |
| 21 void usage() | |
| 22 { | |
| 23 printf( | |
| 24 "nsconv -f fromcode -t tocode infile outfile\n" | |
| 25 "nsconv -f fromcode -t tocode infile > outfile\n" | |
| 26 "nsconv -f fromcode -t tocode < infile > outfile\n" | |
| 27 ); | |
| 28 } | |
| 29 | |
| 30 #define INBUFSIZE (1024*16) | |
| 31 #define MEDBUFSIZE (1024*16*2) | |
| 32 #define OUTBUFSIZE (1024*16*8) | |
| 33 char inbuffer[INBUFSIZE]; | |
| 34 char outbuffer[OUTBUFSIZE]; | |
| 35 char16_t medbuffer[MEDBUFSIZE]; | |
| 36 | |
| 37 int main(int argc, const char** argv) | |
| 38 { | |
| 39 nsIUnicodeEncoder* encoder = nullptr; | |
| 40 nsIUnicodeDecoder* decoder = nullptr; | |
| 41 FILE* fin = 0; | |
| 42 FILE* fout = 0; | |
| 43 FILE* infile = 0; | |
| 44 FILE* outfile = 0; | |
| 45 nsresult res= NS_OK; | |
| 46 | |
| 47 NS_InitXPCOM2(nullptr, nullptr, nullptr); | |
| 48 | |
| 49 // get ccMain; | |
| 50 nsCOMPtr<nsICharsetConverterManager> ccMain = | |
| 51 do_GetService(kCharsetConverterManagerCID, &res); | |
| 52 if(NS_FAILED(res)) | |
| 53 { | |
| 54 fprintf(stderr, "Cannot get Character Converter Manager %x\n", res); | |
| 55 return -1; | |
| 56 } | |
| 57 | |
| 58 int i; | |
| 59 if(argc > 4) | |
| 60 { | |
| 61 for(i =0; i < argc; i++) | |
| 62 { | |
| 63 if(strcmp(argv[i], "-f") == 0) | |
| 64 { | |
| 65 // User has specified the charset to convert from | |
| 66 nsAutoCString str; | |
| 67 | |
| 68 // First check if a charset alias was given, | |
| 69 // and convert to the canonical name | |
| 70 res = ccMain->GetCharsetAlias(argv[i+1], str); | |
| 71 if (NS_FAILED(res)) | |
| 72 { | |
| 73 fprintf(stderr, "Cannot get charset alias for %s %x\n", | |
| 74 argv[i+1], res); | |
| 75 goto error_exit; | |
| 76 } | |
| 77 | |
| 78 // Finally create the decoder | |
| 79 res = ccMain->GetUnicodeDecoder(str.get(), &decoder); | |
| 80 if(NS_FAILED(res)) { | |
| 81 fprintf(stderr, "Cannot get Unicode decoder %s %x\n", | |
| 82 argv[i+1],res); | |
| 83 goto error_exit; | |
| 84 } | |
| 85 | |
| 86 } | |
| 87 | |
| 88 if(strcmp(argv[i], "-t") == 0) | |
| 89 { | |
| 90 // User has specified which charset to convert to | |
| 91 nsAutoCString str; | |
| 92 | |
| 93 // First check if a charset alias was given, | |
| 94 // and convert to the canonical name | |
| 95 res = ccMain->GetCharsetAlias(argv[i+1], str); | |
| 96 if (NS_FAILED(res)) | |
| 97 { | |
| 98 fprintf(stderr, "Cannot get charset alias for %s %x\n", | |
| 99 argv[i+1], res); | |
| 100 goto error_exit; | |
| 101 } | |
| 102 | |
| 103 // Finally create the encoder | |
| 104 res = ccMain->GetUnicodeEncoderRaw(str.get(), &encoder); | |
| 105 if(NS_FAILED(res)) { | |
| 106 fprintf(stderr, "Cannot get Unicode encoder %s %x\n", | |
| 107 argv[i+1],res); | |
| 108 goto error_exit; | |
| 109 } | |
| 110 } | |
| 111 } | |
| 112 | |
| 113 if (argc > 5) | |
| 114 { | |
| 115 // The user has specified an input file | |
| 116 // if we have more than four arguments | |
| 117 fin = infile = fopen(argv[5], "rb"); | |
| 118 if(!infile) | |
| 119 { | |
| 120 usage(); | |
| 121 fprintf(stderr,"cannot open input file %s\n", argv[5]); | |
| 122 goto error_exit; | |
| 123 } | |
| 124 | |
| 125 if (argc > 6) | |
| 126 { | |
| 127 // The user has specified an output file | |
| 128 // if we have more than four arguments | |
| 129 fout = outfile = fopen(argv[6], "ab"); | |
| 130 if(!outfile) | |
| 131 { | |
| 132 usage(); | |
| 133 fprintf(stderr,"cannot open output file %s\n", argv[6]); | |
| 134 goto error_exit; | |
| 135 } | |
| 136 } | |
| 137 else | |
| 138 fout = stdout; | |
| 139 } | |
| 140 else | |
| 141 { | |
| 142 // No inputfiles are given. Read and write | |
| 143 // to/from standard in and standard out | |
| 144 fin = stdin; | |
| 145 fout = stdout; | |
| 146 } | |
| 147 | |
| 148 int32_t insize,medsize,outsize; | |
| 149 while((insize=fread(inbuffer, 1,INBUFSIZE, fin)) > 0) | |
| 150 { | |
| 151 medsize=MEDBUFSIZE; | |
| 152 | |
| 153 res = decoder->Convert(inbuffer,&insize, medbuffer, &medsize); | |
| 154 if(NS_FAILED(res)) { | |
| 155 fprintf(stderr, "failed in decoder->Convert %x\n",res); | |
| 156 goto error_exit; | |
| 157 } | |
| 158 outsize = OUTBUFSIZE; | |
| 159 res = encoder->Convert(medbuffer, &medsize, outbuffer,&outsize); | |
| 160 if(NS_FAILED(res)) { | |
| 161 fprintf(stderr, "failed in encoder->Convert %x\n",res); | |
| 162 goto error_exit; | |
| 163 } | |
| 164 fwrite(outbuffer, 1, outsize, fout); | |
| 165 | |
| 166 } | |
| 167 | |
| 168 // Clean up | |
| 169 if (infile != 0) | |
| 170 fclose(infile); | |
| 171 if (outfile != 0) | |
| 172 fclose(outfile); | |
| 173 fprintf(stderr, "Done!\n"); | |
| 174 NS_IF_RELEASE(encoder); | |
| 175 NS_IF_RELEASE(decoder); | |
| 176 return 0; | |
| 177 } | |
| 178 usage(); | |
| 179 error_exit: | |
| 180 // Clean up after | |
| 181 if (infile != 0) | |
| 182 fclose(infile); | |
| 183 if (outfile != 0) | |
| 184 fclose(outfile); | |
| 185 NS_IF_RELEASE(encoder); | |
| 186 NS_IF_RELEASE(decoder); | |
| 187 return -1; | |
| 188 } |
