532 lines
13 KiB
C++
532 lines
13 KiB
C++
/* -*- 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/. */
|
|
|
|
#include "nsIServiceManager.h"
|
|
#include "nsICharsetConverterManager.h"
|
|
#include "nsUCSupport.h"
|
|
#include "nsString.h"
|
|
#include "nsIStringEnumerator.h"
|
|
#include "nsTArray.h"
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Global functions and data [declaration]
|
|
|
|
#define ARRAY_SIZE(_array) (sizeof(_array) / sizeof(_array[0]))
|
|
#define SMALL_BUFFER_SIZE 512
|
|
#define MED_BUFFER_SIZE 1024
|
|
#define BIG_BUFFER_SIZE 2048
|
|
|
|
static NS_DEFINE_CID(kCharsetConverterManagerCID, NS_ICHARSETCONVERTERMANAGER_CID);
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Class nsTestLog [declaration]
|
|
|
|
/**
|
|
* A Logging class for test programs.
|
|
*
|
|
* This simple test program will not trigger a component registration. So
|
|
* Mozilla has to be run once before running this, so that the necessary
|
|
* components will be registered. Also, please observe that the ContractID's are
|
|
* case sensitive now!
|
|
*
|
|
* @created 28/Mar/2000
|
|
* @author Catalin Rotaru [CATA]
|
|
*/
|
|
class nsTestLog
|
|
{
|
|
private:
|
|
|
|
static const char * kTraceDelimiter;
|
|
|
|
nsAutoCString mTrace;
|
|
|
|
public:
|
|
|
|
void AddTrace(const char * aTrace);
|
|
void DelTrace(const char * aTrace);
|
|
void PrintError(const char * aCall, const int aError);
|
|
void PrintError(const char * aCall, const char * aMessage);
|
|
};
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Class nsTestUConv [declaration]
|
|
|
|
/**
|
|
* The main class of the program.
|
|
*
|
|
* XXX Create a very general set of "bug and regression" test cases and the
|
|
* one in TestTempBug()
|
|
* XXX Apply the new argument style (pointers) to the converters interfaces
|
|
*
|
|
* @created 28/Mar/2000
|
|
* @author Catalin Rotaru [CATA]
|
|
*/
|
|
class nsTestUConv
|
|
{
|
|
private:
|
|
|
|
nsTestLog mLog;
|
|
|
|
/**
|
|
* Run the built-in set of self tests for encoders.
|
|
*/
|
|
nsresult TestEncoders();
|
|
|
|
/**
|
|
* Run the built-in set of self tests for decoders.
|
|
*/
|
|
nsresult TestDecoders();
|
|
|
|
/**
|
|
* Run the built-in set of self tests for the CharsetManager.
|
|
*/
|
|
nsresult TestCharsetManager();
|
|
|
|
/**
|
|
* Display charset detectors and their attributes.
|
|
*/
|
|
nsresult DisplayDetectors();
|
|
|
|
/**
|
|
* Display charsets and their attributes.
|
|
*/
|
|
nsresult DisplayCharsets();
|
|
|
|
/**
|
|
* Run a temporary debug test. This method is ment as a placeholder when some
|
|
* quick debugging is needed.
|
|
*/
|
|
nsresult TestTempBug();
|
|
|
|
nsresult Encode(char16_t ** aSrc, char16_t * aSrcEnd, char ** aDest,
|
|
char * aDestEnd, const nsAFlatCString& aCharset);
|
|
|
|
/**
|
|
* Bridge methods between the new argument style (poiters) and the old one
|
|
* (lengths). To be removed when the converter interfaces will switch to the
|
|
* new style.
|
|
*
|
|
* This wraps an encoder Convert() call.
|
|
*/
|
|
nsresult ConvertEncode(char16_t ** aSrc, char16_t * aSrcEnd, char ** aDest,
|
|
char * aDestEnd, nsIUnicodeEncoder * aEncoder);
|
|
|
|
/**
|
|
* This wraps an encoder Finish() call.
|
|
*/
|
|
nsresult FinishEncode(char ** aDest, char * aDestEnd,
|
|
nsIUnicodeEncoder * aEncoder);
|
|
|
|
void PrintSpaces(int aCount);
|
|
|
|
public:
|
|
|
|
/**
|
|
* Main method of the program.
|
|
*/
|
|
nsresult Main(int aArgC, char ** aArgV);
|
|
};
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Global functions and data [implementation]
|
|
|
|
int main(int argc, char ** argv)
|
|
{
|
|
nsTestUConv testObj;
|
|
nsresult res;
|
|
|
|
res = testObj.Main(argc, argv);
|
|
return (NS_FAILED(res));
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Class nsTestLog [implementation]
|
|
|
|
const char * nsTestLog::kTraceDelimiter = ".";
|
|
|
|
void nsTestLog::AddTrace(const char * aTrace)
|
|
{
|
|
mTrace.Append(aTrace);
|
|
mTrace.Append(kTraceDelimiter);
|
|
}
|
|
|
|
void nsTestLog::DelTrace(const char * aTrace)
|
|
{
|
|
mTrace.Truncate(mTrace.Length() - strlen(aTrace) - strlen(kTraceDelimiter));
|
|
}
|
|
|
|
void nsTestLog::PrintError(const char * aCall, const int aError)
|
|
{
|
|
const char * trace = mTrace.get();
|
|
printf("ERROR at %s%s code=0x%x.\n", trace, aCall, aError);
|
|
}
|
|
|
|
void nsTestLog::PrintError(const char * aCall, const char * aMessage)
|
|
{
|
|
const char * trace = mTrace.get();
|
|
printf("ERROR at %s%s reason: %s.\n", trace, aCall, aMessage);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Class nsTestUConv [implementation]
|
|
|
|
nsresult nsTestUConv::TestEncoders()
|
|
{
|
|
const char * trace = "TestEncoders";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
nsCOMPtr<nsICharsetConverterManager> ccMan =
|
|
do_GetService(kCharsetConverterManagerCID, &res);
|
|
if (NS_FAILED(res)) return res;
|
|
|
|
nsCOMPtr<nsIUTF8StringEnumerator> encoders;
|
|
res = ccMan->GetEncoderList(getter_AddRefs(encoders));
|
|
if (NS_FAILED(res)) return res;
|
|
|
|
bool hasMore;
|
|
encoders->HasMore(&hasMore);
|
|
|
|
nsAutoCString charset;
|
|
while (hasMore) {
|
|
encoders->GetNext(charset);
|
|
|
|
encoders->HasMore(&hasMore);
|
|
}
|
|
|
|
mLog.DelTrace(trace);
|
|
return res;
|
|
}
|
|
|
|
nsresult nsTestUConv::TestDecoders()
|
|
{
|
|
const char * trace = "TestDecoders";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
// XXX write me
|
|
|
|
mLog.DelTrace(trace);
|
|
return res;
|
|
}
|
|
|
|
nsresult nsTestUConv::TestCharsetManager()
|
|
{
|
|
const char * trace = "TestCharsetManager";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
nsAutoString name;
|
|
nsCOMPtr<nsIAtom> csAtom;
|
|
|
|
nsCOMPtr<nsICharsetConverterManager> ccMan =
|
|
do_GetService(kCharsetConverterManagerCID, &res);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("NS_WITH_SERVICE", res);
|
|
return res;
|
|
}
|
|
|
|
mLog.DelTrace(trace);
|
|
return res;
|
|
}
|
|
|
|
nsresult nsTestUConv::DisplayDetectors()
|
|
{
|
|
const char * trace = "DisplayDetectors";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
nsCOMPtr<nsICharsetConverterManager> ccMan =
|
|
do_GetService(kCharsetConverterManagerCID, &res);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("NS_WITH_SERVICE", res);
|
|
return res;
|
|
}
|
|
|
|
// charset detectors
|
|
nsCOMPtr<nsIUTF8StringEnumerator> detectors;
|
|
|
|
res = ccMan->GetCharsetDetectorList(getter_AddRefs(detectors));
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("GetCharsetDetectorList()", res);
|
|
return res;
|
|
}
|
|
|
|
printf("***** Character Set Detectors *****\n");
|
|
|
|
bool hasMore;
|
|
detectors->HasMore(&hasMore);
|
|
while (hasMore) {
|
|
nsAutoCString detectorName;
|
|
res = detectors->GetNext(detectorName);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("GetNext()", res);
|
|
return res;
|
|
}
|
|
|
|
printf("%s", detectorName.get());
|
|
PrintSpaces(36 - detectorName.Length()); // align to hard coded column number
|
|
|
|
nsAutoString title;
|
|
res = ccMan->GetCharsetTitle(detectorName.get(), title);
|
|
if (NS_FAILED(res)) title.SetLength(0);
|
|
printf("\"%s\"\n", NS_LossyConvertUTF16toASCII(title).get());
|
|
|
|
detectors->HasMore(&hasMore);
|
|
}
|
|
|
|
mLog.DelTrace(trace);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult nsTestUConv::DisplayCharsets()
|
|
{
|
|
const char * trace = "DisplayCharsets";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
nsCOMPtr<nsICharsetConverterManager> ccMan =
|
|
do_GetService(kCharsetConverterManagerCID, &res);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("NS_WITH_SERVICE", res);
|
|
return res;
|
|
}
|
|
|
|
nsCOMPtr<nsIUTF8StringEnumerator> decoders;
|
|
nsCOMPtr<nsIUTF8StringEnumerator> encoders;
|
|
|
|
res = ccMan->GetDecoderList(getter_AddRefs(decoders));
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("GetDecoderList()", res);
|
|
return res;
|
|
}
|
|
|
|
res = ccMan->GetEncoderList(getter_AddRefs(encoders));
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("GetEncoderList()", res);
|
|
return res;
|
|
}
|
|
|
|
|
|
printf("***** Character Sets *****\n");
|
|
|
|
uint32_t encCount = 0, decCount = 0;
|
|
uint32_t basicEncCount = 0, basicDecCount = 0;
|
|
|
|
nsTArray<nsCString> allCharsets;
|
|
|
|
nsAutoCString charset;
|
|
bool hasMore;
|
|
encoders->HasMore(&hasMore);
|
|
while (hasMore) {
|
|
res = encoders->GetNext(charset);
|
|
if (NS_SUCCEEDED(res))
|
|
allCharsets.AppendElement(charset);
|
|
|
|
encoders->HasMore(&hasMore);
|
|
}
|
|
|
|
nsAutoString prop, str;
|
|
uint32_t count = allCharsets.Length();
|
|
for (uint32_t i = 0; i < count; i++) {
|
|
|
|
const nsCString& charset = allCharsets[i];
|
|
printf("%s", charset.get());
|
|
PrintSpaces(24 - charset.Length()); // align to hard coded column number
|
|
|
|
|
|
nsCOMPtr<nsIUnicodeDecoder> dec;
|
|
res = ccMan->GetUnicodeDecoder(charset.get(), getter_AddRefs(dec));
|
|
if (NS_FAILED(res)) printf (" ");
|
|
else {
|
|
printf("D");
|
|
decCount++;
|
|
}
|
|
#ifdef DEBUG
|
|
// show the "basic" decoder classes
|
|
if (dec) {
|
|
nsCOMPtr<nsIBasicDecoder> isBasic = do_QueryInterface(dec);
|
|
if (isBasic) {
|
|
basicDecCount++;
|
|
printf("b");
|
|
}
|
|
else printf(" ");
|
|
}
|
|
else printf(" ");
|
|
#endif
|
|
|
|
nsCOMPtr<nsIUnicodeEncoder> enc;
|
|
res = ccMan->GetUnicodeEncoder(charset.get(), getter_AddRefs(enc));
|
|
if (NS_FAILED(res)) printf (" ");
|
|
else {
|
|
printf("E");
|
|
encCount++;
|
|
}
|
|
|
|
#ifdef DEBUG
|
|
if (enc) {
|
|
nsCOMPtr<nsIBasicEncoder> isBasic = do_QueryInterface(enc);
|
|
if (isBasic) {
|
|
basicEncCount++;
|
|
printf("b");
|
|
}
|
|
else printf(" ");
|
|
}
|
|
else printf(" ");
|
|
#endif
|
|
|
|
printf(" ");
|
|
|
|
prop.AssignLiteral(".notForBrowser");
|
|
res = ccMan->GetCharsetData(charset.get(), prop.get(), str);
|
|
if (dec && (NS_FAILED(res))) printf ("B");
|
|
else printf("X");
|
|
|
|
prop.AssignLiteral(".notForComposer");
|
|
res = ccMan->GetCharsetData(charset.get(), prop.get(), str);
|
|
if (enc && (NS_FAILED(res))) printf ("C");
|
|
else printf("X");
|
|
|
|
prop.AssignLiteral(".notForMailView");
|
|
res = ccMan->GetCharsetData(charset.get(), prop.get(), str);
|
|
if (dec && (NS_FAILED(res))) printf ("V");
|
|
else printf("X");
|
|
|
|
prop.AssignLiteral(".notForMailEdit");
|
|
res = ccMan->GetCharsetData(charset.get(), prop.get(), str);
|
|
if (enc && (NS_FAILED(res))) printf ("E");
|
|
else printf("X");
|
|
|
|
printf("(%3d, %3d) ", encCount, decCount);
|
|
res = ccMan->GetCharsetTitle(charset.get(), str);
|
|
if (NS_FAILED(res)) str.SetLength(0);
|
|
NS_LossyConvertUTF16toASCII buff2(str);
|
|
printf(" \"%s\"\n", buff2.get());
|
|
}
|
|
|
|
printf("%u of %u decoders are basic (%d%%)\n",
|
|
basicDecCount, decCount, (basicDecCount * 100) / decCount);
|
|
|
|
printf("%u of %u encoders are basic (%d%%)\n",
|
|
basicEncCount, encCount, (basicEncCount * 100) / encCount);
|
|
mLog.DelTrace(trace);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult nsTestUConv::TestTempBug()
|
|
{
|
|
const char * trace = "TestTempBug";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
NS_NAMED_LITERAL_CSTRING(charset, "ISO-2022-JP");
|
|
char16_t src[] = {0x0043, 0x004e, 0x0045, 0x0054, 0x0020, 0x004A, 0x0061,
|
|
0x0070, 0x0061, 0x006E, 0x0020, 0x7DE8, 0x96C6, 0x5C40};
|
|
char16_t * srcEnd = src + ARRAY_SIZE(src);
|
|
char dest[BIG_BUFFER_SIZE];
|
|
char * destEnd = dest + BIG_BUFFER_SIZE;
|
|
|
|
char16_t * p = src;
|
|
char * q = dest;
|
|
res = Encode(&p, srcEnd, &q, destEnd, charset);
|
|
|
|
mLog.DelTrace(trace);
|
|
return res;
|
|
}
|
|
|
|
nsresult nsTestUConv::Encode(char16_t ** aSrc, char16_t * aSrcEnd,
|
|
char ** aDest, char * aDestEnd,
|
|
const nsAFlatCString& aCharset)
|
|
{
|
|
const char * trace = "Encode";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
nsCOMPtr<nsICharsetConverterManager> ccMan =
|
|
do_GetService(kCharsetConverterManagerCID, &res);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("NS_WITH_SERVICE", res);
|
|
return res;
|
|
}
|
|
|
|
nsCOMPtr<nsIUnicodeEncoder> enc;
|
|
res = ccMan->GetUnicodeEncoder(aCharset.get(), getter_AddRefs(enc));
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("GetUnicodeEncoder()", res);
|
|
return res;
|
|
}
|
|
|
|
res = ConvertEncode(aSrc, aSrcEnd, aDest, aDestEnd, enc);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("Convert()", res);
|
|
return res;
|
|
}
|
|
|
|
res = FinishEncode(aDest, aDestEnd, enc);
|
|
if (NS_FAILED(res)) {
|
|
mLog.PrintError("Finish()", res);
|
|
return res;
|
|
}
|
|
|
|
mLog.DelTrace(trace);
|
|
return res;
|
|
}
|
|
|
|
nsresult nsTestUConv::ConvertEncode(char16_t ** aSrc, char16_t * aSrcEnd,
|
|
char ** aDest, char * aDestEnd,
|
|
nsIUnicodeEncoder * aEncoder)
|
|
{
|
|
char16_t * src = (*aSrc);
|
|
char * dest = (*aDest);
|
|
int32_t srcLen = aSrcEnd - src;
|
|
int32_t destLen = aDestEnd - dest;
|
|
|
|
nsresult res = aEncoder->Convert(src, &srcLen, dest, &destLen);
|
|
|
|
(*aSrc) = src + srcLen;
|
|
(*aDest) = dest + destLen;
|
|
return res;
|
|
}
|
|
|
|
nsresult nsTestUConv::FinishEncode(char ** aDest, char * aDestEnd,
|
|
nsIUnicodeEncoder * aEncoder)
|
|
{
|
|
char * dest = (*aDest);
|
|
int32_t destLen = aDestEnd - dest;
|
|
|
|
nsresult res = aEncoder->Finish(dest, &destLen);
|
|
|
|
(*aDest) = dest + destLen;
|
|
return res;
|
|
}
|
|
|
|
void nsTestUConv::PrintSpaces(int aCount)
|
|
{
|
|
for (int i = 0; i < aCount; i++) printf(" ");
|
|
}
|
|
|
|
nsresult nsTestUConv::Main(int aArgC, char ** aArgV)
|
|
{
|
|
const char * trace = "Main";
|
|
mLog.AddTrace(trace);
|
|
nsresult res = NS_OK;
|
|
|
|
if (aArgC < 2) {
|
|
// no arguments were passed to the program, so we just run the self tests
|
|
res = TestCharsetManager();
|
|
if (NS_SUCCEEDED(res)) res = TestEncoders();
|
|
if (NS_SUCCEEDED(res)) res = TestDecoders();
|
|
} else if (!strcmp(aArgV[1], "-tempbug")) {
|
|
// we are testing a temporary bug
|
|
res = TestTempBug();
|
|
} else if (!strcmp(aArgV[1], "-display")) {
|
|
// display all the available data
|
|
res = DisplayDetectors();
|
|
if (NS_SUCCEEDED(res)) res = DisplayCharsets();
|
|
}
|
|
|
|
mLog.DelTrace(trace);
|
|
return res;
|
|
}
|