diff options
Diffstat (limited to 'intl/uconv/tests/nsconv.cpp')
-rw-r--r-- | intl/uconv/tests/nsconv.cpp | 188 |
1 files changed, 188 insertions, 0 deletions
diff --git a/intl/uconv/tests/nsconv.cpp b/intl/uconv/tests/nsconv.cpp new file mode 100644 index 0000000000..116f6fc1f3 --- /dev/null +++ b/intl/uconv/tests/nsconv.cpp @@ -0,0 +1,188 @@ +/* -*- 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; +} |