int wscCPPutMessage(char *msg, int msgLen)
{
unsigned char *encodeStr = NULL;
RTMP_WSC_MSG_HDR *rtmpHdr = NULL;
int len;
char *wscCPStateVarParam="NewInMessage";
DBGPRINTF(RT_DBG_INFO, "(%s):Ready to parse the K2U msg(TotalLen=%d, headerLen=%d)!\n", __FUNCTION__, msgLen, sizeof(RTMP_WSC_MSG_HDR));
rtmpHdr = (RTMP_WSC_MSG_HDR *)msg;
DBGPRINTF(RT_DBG_INFO, "\tMsgType=%d!\n", rtmpHdr->msgType);
DBGPRINTF(RT_DBG_INFO, "\tMsgSubType=%d!\n", rtmpHdr->msgSubType);
DBGPRINTF(RT_DBG_INFO, "\tipAddress=0x%x!\n", rtmpHdr->ipAddr);
DBGPRINTF(RT_DBG_INFO, "\tMsgLen=%d!\n", rtmpHdr->msgLen);
if(rtmpHdr->msgType == WSC_OPCODE_UPNP_CTRL || rtmpHdr->msgType == WSC_OPCODE_UPNP_MGMT)
{
DBGPRINTF(RT_DBG_INFO, "Receive a UPnP Ctrl/Mgmt Msg!\n");
return 0;
}
//wsc_hexdump("wscCPPutMessage-K2UMsg", msg, msgLen);
len = ILibBase64Encode((unsigned char *)(msg + sizeof(RTMP_WSC_MSG_HDR)), rtmpHdr->msgLen, &encodeStr);
if (len >0)
{
WscUPnPCPSendAction(rtmpHdr->ipAddr, "PutMessage", &wscCPStateVarParam, &encodeStr, 1);
free(encodeStr);
}
return 0;
}
void PassiveCandidateSink(ILibWrapper_WebRTC_Connection connection, struct sockaddr_in6* candidate)
{
printf("While generating an offer?\n"rst);
SimpleRendezvousServer sender;
SimpleRendezvousServerToken token;
char *offer;
char *encodedOffer;
int encodedOfferLen;
char *h1;
ILibWrapper_WebRTC_Connection_GetUserData(connection, &sender, &token, (void**)&h1);
offer = ILibWrapper_WebRTC_Connection_AddServerReflexiveCandidateToLocalSDP(connection, candidate);
printf(green"ofer: \n %s\n"rst,offer);
encodedOfferLen = ILibBase64Encode((unsigned char*)offer, strlen(offer),(unsigned char**)&encodedOffer);
h1 = ILibString_Replace(passiveHtmlBody, passiveHtmlBodyLength, "/*{{{SDP}}}*/", 13, encodedOffer, encodedOfferLen);
free(offer);
free(encodedOffer);
SimpleRendezvousServer_Respond(sender, token, 1, h1, strlen(h1), ILibAsyncSocket_MemoryOwnership_CHAIN); // Send the SDP to the remote side
}
/* Main entry point to the sample application */
int _tmain(int argc, _TCHAR* argv[])
{
char *targetFile;
char *tempBuffer;
int tempBufferLength;
char *sourceBuffer;
char *buffer;
char *delimiter;
int delimiterLength;
int x;
FILE* TargetFile;
int TargetFileLength;
int SourceFileLength;
struct parser_result *pr;
HANDLE target,src;
FILETIME targetFT,srcFT;
DWORD ptid=0;
_CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );
targetFile = (char*)malloc(2+strlen(argv[1]));
strcpy(targetFile,argv[1]);
targetFile[strlen(argv[1])] = '~';
targetFile[strlen(argv[1])+1] = 0;
if(argc==2)
{
TargetFileLength = ILibReadFileFromDiskEx(&buffer,targetFile);
if(TargetFileLength!=0)
{
ILibWriteStringToDiskEx(argv[1],buffer,TargetFileLength);
free(buffer);
ILibDeleteFileFromDisk(targetFile);
}
return;
}
target = CreateFile(argv[1],GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
src = CreateFile(argv[2],GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
GetFileTime(target,NULL,NULL,&targetFT);
GetFileTime(src,NULL,NULL,&srcFT);
CloseHandle(target);
CloseHandle(src);
if(srcFT.dwHighDateTime < targetFT.dwHighDateTime ||
(srcFT.dwHighDateTime == targetFT.dwHighDateTime && srcFT.dwLowDateTime < targetFT.dwLowDateTime))
{
//return;
}
printf("SourceCodeInjector:\r\n");
printf(" TargetFile: %s\r\n",argv[1]);
printf(" SourceFile: %s\r\n",argv[2]);
printf(" VariableName: %s\r\n",argv[3]);
if(argc==5)
{
printf(" Delimiter: %s\r\n",argv[4]);
}
// argv[1] = TargetFile
// argv[2] = SourceFile
// argv[3] = VariableName
if(argc==5)
{
delimiter = (char*)malloc((int)strlen(argv[4])+1);
delimiterLength=sprintf(delimiter,"%s",argv[4]);
}
else
{
delimiter = (char*)malloc((int)strlen(argv[3])+20);
delimiterLength=sprintf(delimiter,"// -=S3P4R470R=- {%s}",argv[3]);
}
TargetFileLength = ILibReadFileFromDiskEx(&buffer,targetFile);
SourceFileLength = ILibReadFileFromDiskEx(&sourceBuffer,argv[2]);
if(TargetFileLength == 0)
{
TargetFileLength = ILibReadFileFromDiskEx(&buffer,argv[1]);
}
if(TargetFileLength == NULL || SourceFileLength == 0)
{
exit(0);
}
tempBufferLength = ILibBase64Encode(sourceBuffer,SourceFileLength,&tempBuffer);
free(sourceBuffer);
sourceBuffer = tempBuffer;
SourceFileLength = tempBufferLength;
pr = ILibParseString(buffer,0,TargetFileLength,delimiter,delimiterLength);
TargetFile = fopen(targetFile,"wb");
fwrite(pr->FirstResult->data,sizeof(char),pr->FirstResult->datalength,TargetFile);
fwrite(delimiter,sizeof(char),delimiterLength,TargetFile);
fwrite("\n",sizeof(char),1,TargetFile);
//fwrite("private static byte[] ",sizeof(char),22,TargetFile);
//fwrite(argv[3],sizeof(char),strlen(argv[3]),TargetFile);
//fwrite(" = {",sizeof(char),4,TargetFile);
fwrite("private static string ",sizeof(char),22,TargetFile);
fwrite(argv[3],sizeof(char),strlen(argv[3]),TargetFile);
fwrite(" = \"",sizeof(char),4,TargetFile);
for(x=0;x<SourceFileLength;++x)
{
if(argv[3] != '-')
{
if(sourceBuffer[x] == '"')
{
fwrite("\\\"",sizeof(char),2,TargetFile);
}
else if(sourceBuffer[x] == '\\')
{
fwrite("\\\\",sizeof(char),2,TargetFile);
}
else if(sourceBuffer[x] == '\r')
{
fwrite("\\r",sizeof(char),2,TargetFile);
}
else if(sourceBuffer[x] == '\n')
{
fwrite("\\n",sizeof(char),2,TargetFile);
}
else
{
fwrite(&sourceBuffer[x],sizeof(char),1,TargetFile);
}
if(x!=0 && x%100==0)
{
fwrite("\"\n",sizeof(char),2,TargetFile);
//if(x!=SourceFileLength-1)
//{
fwrite("+\"",sizeof(char),2,TargetFile);
//}
}
}
else
{
fwrite(&sourceBuffer[x],sizeof(char),1,TargetFile);
}
}
if(argv[3]!='-')
{
fwrite("\";\n",sizeof(char),3,TargetFile);
}
fwrite(delimiter,sizeof(char),delimiterLength,TargetFile);
fwrite(pr->LastResult->data,sizeof(char),pr->LastResult->datalength,TargetFile);
fflush(TargetFile);
fclose(TargetFile);
ILibDestructParserResults(pr);
free(sourceBuffer);
free(buffer);
free(delimiter);
free(targetFile);
return 0;
}
// This gets called When the browser hits one of the two URLs
void Rendezvous_OnGet(SimpleRendezvousServer sender, SimpleRendezvousServerToken token, char* path, char* receiver)
{
printf(green"on Get\n"rst);
if(strcmp(path, "/active")==0)
{
if(mConnection != NULL) { ILibWrapper_WebRTC_Connection_Disconnect(mConnection); mConnection = NULL; }
#ifdef MICROSTACK_TLS_DETECT
if(SimpleRendezvousServer_IsTLS(token)!=0)
{
printf(" Received Client Request: [TLS]\r\n");
}
else
{
printf(" Received Client Request: [No-TLS]\r\n");
}
#endif
SimpleRendezvousServer_Respond(sender, token, 1, htmlBody, htmlBodyLength, ILibAsyncSocket_MemoryOwnership_USER); // Send the HTML to the Browser
}
else if(strcmp(path, "/websocket")==0)
{
printf(red"/websocket\n"rst);
int v = SimpleRendezvousServer_WebSocket_IsRequest(token);
if(SimpleRendezvousServer_IsAuthenticated(token, "www.meshcentral.com", 19)!=0)
{
char* name = SimpleRendezvousServer_GetUsername(token);
if(SimpleRendezvousServer_ValidatePassword(token, "bryan", 5)!=0)
{
printf(red"rendevous respond\n"rst);
SimpleRendezvousServer_Respond(sender, token, 1, wshtmlbody, wshtmlBodyLength, ILibAsyncSocket_MemoryOwnership_USER); // Send the HTML to the Browser
}
}
}
else if(strcmp(path, "/websocketInit")==0)
{
printf(green"/websocketInit\n"rst);
int v = SimpleRendezvousServer_WebSocket_IsRequest(token);
SimpleRendezvousServer_UpgradeWebSocket(token, 65535, &OnWebSocket, &OnWebSocketClosed);
}
else if(strcmp(path, "/passive")==0)
{
printf("/passive\n");
char* offer;
char* encodedOffer = NULL;
int encodedOfferLen;
char *h1;
if(mConnection != NULL){ ILibWrapper_WebRTC_Connection_Disconnect(mConnection); mConnection = NULL; }
mConnection = ILibWrapper_WebRTC_ConnectionFactory_CreateConnection(mConnectionFactory, &WebRTCConnectionSink, &WebRTCDataChannelSink, &WebRTCConnectionSendOkSink);
ILibWrapper_WebRTC_Connection_SetStunServers(mConnection, stunServerList, 9);
if(useStun==0)
{
//
// If STUN was not enabled, then we just generate an offer, and encode it as Base64 into the HTML file to send to the browser
//
printf("Generating Offer...\r\n");
offer = ILibWrapper_WebRTC_Connection_GenerateOffer(mConnection, NULL);
printf("Encoding Offer...\r\n");
encodedOfferLen = ILibBase64Encode((unsigned char*)offer, strlen(offer),(unsigned char**)&encodedOffer);
h1 = ILibString_Replace(passiveHtmlBody, passiveHtmlBodyLength, "/*{{{SDP}}}*/", 13, encodedOffer, encodedOfferLen);
printf("Sending Offer...\r\n");
free(offer);
free(encodedOffer);
SimpleRendezvousServer_Respond(sender, token, 1, h1, strlen(h1), ILibAsyncSocket_MemoryOwnership_CHAIN); // Send the HTML/OFFER to the browser
}
else
{
//
// If STUN was enabled, to simplify this sample app, we'll generate the offer, but we won't do anything with it, as we'll wait until
printf(red"we get called back saying that candidate gathering was done\n"rst);
//
ILibWrapper_WebRTC_Connection_SetUserData(mConnection, sender, token, NULL);
free(ILibWrapper_WebRTC_Connection_GenerateOffer(mConnection, &PassiveCandidateSink));
// We're immediately freeing, becuase we're going to re-generate the offer when we get the candidate callback
// It would be better if we sent this offer, and just passed along the candidates separately, but that is for another sample, since that requires more complex logic.
}
}
else
{
printf(red"If we got a GET request for something other than ACTIVE or PASSIVE, we'll just return an error\n"rst);
SimpleRendezvousServer_Respond(sender, token, 0, NULL, 0, ILibAsyncSocket_MemoryOwnership_CHAIN);
}
}
