/*
* Class: javm_nativeimp_NsDatagramSocketImpl
* Method: peek
* Signature: (Ljavm/net/InetAddress;)I
*/
JNIEXPORT jint JNICALL Java_agentj_nativeimp_NsDatagramSocketImpl_peek
(JNIEnv *env, jobject javaobj, jobject longAddress) {
PLOG(PL_DEBUG, "Java_agentj_nativeimp_NsDatagramSocketImpl_peek: Entering\n");
ProtoSocket *socket = (ProtoSocket *)getSocket(env, javaobj); // gets the socket...
int len=1;
char *bdata = new char[len];
unsigned int actualRead=len;
socket->Recv(bdata, actualRead);
delete []bdata;
char sendersAddress[5];
int sendersPort;
ProtoAddress sendersProtoAddress = socket->GetDestination();
sendersPort = sendersProtoAddress.GetPort();
sendersProtoAddress.GetHostString(sendersAddress, strlen(sendersAddress));
jclass longClass = env->FindClass("java/lang/Long");
jmethodID setAddressID = env->GetMethodID(longClass, "<init>", "(J)V");
longAddress = env->NewObject(longClass, setAddressID, (jlong)sendersProtoAddress.SimGetAddress());
return 0;
}
void Pipe2Sock::SendMessage()
{
if(socket.IsOpen())
{
TRACE("Sending message %s to %s/%d\n",msg_buffer,dst_addr.GetHostString(),dst_addr.GetPort());
unsigned int len = strlen(msg_buffer);
socket.SendTo(msg_buffer,len,dst_addr);
memset(msg_buffer,0,msg_len); //clear out the message
}
}
int main(int argc, char* argv[])
{
UINT16 serverPort = 0;
ProtoAddress serverAddr;
if (3 == argc) // "simple listen <portNumber>"
{
if (!strncmp("listen", argv[1], strlen(argv[1])))
{
if (1 != (sscanf(argv[2], "%hu", &serverPort)))
{
fprintf(stderr, "simple: bad <port>\n");
serverPort = 0;
}
}
}
else if (4 == argc) // "simple connect <serverAddr> <serverPort>
{
if (!strncmp("connect", argv[1], strlen(argv[1])))
{
if (1 != (sscanf(argv[3], "%hu", &serverPort)))
{
fprintf(stderr, "simple: bad <serverPort>\n");
serverPort = 0;
}
else if (!serverAddr.ResolveFromString(argv[2]))
{
fprintf(stderr, "simple: bad <serverAddr>\n");
serverPort = 0;
}
else
{
serverAddr.SetPort(serverPort);
}
}
}
if (!serverAddr.IsValid() && (0 == serverPort))
{
usage();
return -1;
}
if (serverAddr.IsValid()) // connect to server address as a "client" and make a request
{
ProtoSocket clientSocket(ProtoSocket::TCP);
fprintf(stderr, "simple: client connecting to server: %s/%hu ...\n",
serverAddr.GetHostString(), serverAddr.GetPort());
if (!clientSocket.Connect(serverAddr))
{
fprintf(stderr, "simple: error connecting to server: %s/%hu\n",
serverAddr.GetHostString(),
serverAddr.GetPort());
return -1;
}
fprintf(stderr, "simple: client sending request to server ...\n");
char* clientRequest = "Hello Server, this is a simple protolib client!";
unsigned int length = strlen(clientRequest) + 1;
unsigned int sent = 0;
while (sent < length)
{
unsigned int numBytes = length - sent;
if (!clientSocket.Send(clientRequest+sent, numBytes))
{
fprintf(stderr, "simple: error sending to server\n");
clientSocket.Close();
return -1;
}
else
{
sent += numBytes;
}
}
fprintf(stderr, "simple: client awaiting response from server ...\n");
bool receiving = true;
while (receiving)
{
char buffer[256];
buffer[256] = '\0';
unsigned int numBytes = 255;
if (!clientSocket.Recv(buffer, numBytes))
{
fprintf(stderr, "simple: error receiving from server\n");
clientSocket.Close();
return -1;
}
else if (numBytes > 0)
{
fprintf(stdout, "simple: client recvd \"%s\" from server: %s/%hu\n",
buffer,
serverAddr.GetHostString(),
serverAddr.GetPort());
}
else
{
fprintf(stderr, "simple: server shutdown connection.\n");
receiving = false;
}
} // end while(receiving)
clientSocket.Close();
}
else // act as a "server" listening to the indicated port, responding to requests
{
for (;;)
{
ProtoSocket serverSocket(ProtoSocket::TCP);
if (!serverSocket.Listen(serverPort))
{
fprintf(stderr, "simple: server error listening\n");
serverSocket.Close();
return -1;
}
else
{
fprintf(stderr, "simple: server listening on port:%hu ... (use <CTRL-C> to exit)\n", serverPort);
if (!serverSocket.Accept())
{
fprintf(stderr, "simple: server error accepting connection\n");
serverSocket.Close();
return -1;
}
fprintf(stderr, "simple: server accepted connection from client: %s/%hu ...\n",
serverSocket.GetDestination().GetHostString(),
serverSocket.GetDestination().GetPort());
bool receiving = true;
while (receiving)
{
char buffer[256];
buffer[256] = '\0';
unsigned int numBytes = 255;
if (!serverSocket.Recv(buffer, numBytes))
{
fprintf(stderr, "simple: error receiving from client\n");
serverSocket.Close();
return -1;
}
else if (numBytes > 0)
{
fprintf(stdout, "simple: server recvd \"%s\" from client: %s/%hu\n",
buffer,
serverSocket.GetDestination().GetHostString(),
serverSocket.GetDestination().GetPort());
if (NULL != strchr(buffer, '!'))
{
fprintf(stderr, "simple: server recvd EOT character '!' from client ...\n");
receiving = false;
}
break;
}
else
{
fprintf(stderr, "simple: client closed connection\n");
receiving = false;
}
}
fprintf(stderr, "simple: server sending response to client ...\n");
char* serverResponse = "Hi there Client, this is a simple protolib server";
unsigned int length = strlen(serverResponse) + 1;
unsigned int sent = 0;
while (sent < length)
{
unsigned int numBytes = length - sent;
if (!serverSocket.Send(serverResponse+sent, numBytes))
{
fprintf(stderr, "simple: error sending to client\n");
serverSocket.Close();
return -1;
}
else
{
sent += numBytes;
}
}
serverSocket.Shutdown();
// After "Shutdown" on a blocking socket, ProtoSocket::Recv() will unblock upon
// receiving "FIN" from client TCP
char buffer[8];
unsigned int numBytes = 8;
serverSocket.Recv(buffer, numBytes);
if (0!= numBytes)
fprintf(stderr, "simple: server received extra data from client?\n\n");
else
fprintf(stderr, "simple: server transmission complete.\n\n");
}
serverSocket.Close();
} // end for (;;)
}
} // end main();
