Changeset 63 for vtcross/branches

Timestamp:

11/20/08 15:17:38 (16 years ago)

Author:

trnewman

Message:

Added better socket functionality.

Files:

• vtcross/branches/trnewman/CR_shell/src/main_cognitive_radio.cpp (modified) (3 diffs)

vtcross/branches/trnewman/CR_shell/src/main_cognitive_radio.cpp

@@ -1 +1 @@
 #include <iostream>
+#include <sys/types.h>
+#include <sys/wait.h>
 #include "tinyxml.h"
 #include "tinystr.h"
@@ -208 +210 @@
 
 
-int StartShell() {
-   // Start socket server
-
-
-   try
-   {
-      //Create the socket
-      ServerSocket server ( SERVER_PORT );
-      ServerSocket server_1 ( 30001 );
-
-      cout << "Waiting for Cognitive Engine to connect: port " << SERVER_PORT << endl;;
-      while ( true )
-      {
-         ServerSocket new_sock;
-         server.accept ( new_sock );
-
-         try
-         {
-            while ( true ) {
-               string data;
-               new_sock >> data;
-               new_sock << data;
-            }
-         }
-         catch ( SocketException& ) {}
-
-      }
-   }
-   catch ( SocketException& e )
-   {
-      cout << "Exception was caught:" << e.description() << "\nExiting.\n";
-   } 
-
-   return 0;
+void error(char *msg)
+{
+    perror(msg);
+    exit(1);
+}
+
+void StartMessageParser( int socketfd) {
+        char buffer[256];
+        int n;
+
+        while(1) {
+                // Clear incoming data buffer
+                bzero(buffer,256);
+                
+                // Wait for incoming message
+                n = recv(socketfd,buffer,255,0);
+                if (n < 0) error("ERROR reading from socket");
+                if (n == 0) error("Client has closed the connection.");
+                
+                // Print message
+                printf("Here is the message: %s\n",buffer);
+
+                // Write message back to client
+                n = write(socketfd,"I got your message",18);
+                if (n < 0) error("ERROR writing to socket");
+        }
+}
+
+int StartShell(int port) {
+        // Start socket server
+        int sockfd, newsockfd, clilen;
+        struct sockaddr_in serv_addr, cli_addr;
+
+        // Setup socket connection
+     
+        sockfd = socket(AF_INET, SOCK_STREAM, 0);
+        if (sockfd < 0) 
+                error("ERROR opening socket");
+        bzero((char *) &serv_addr, sizeof(serv_addr));
+        serv_addr.sin_family = AF_INET;
+        serv_addr.sin_addr.s_addr = INADDR_ANY;
+        serv_addr.sin_port = htons(port);
+        if (bind(sockfd, (struct sockaddr *) &serv_addr,
+                sizeof(serv_addr)) < 0) 
+                error("ERROR on binding");
+        listen(sockfd,5);
+        clilen = sizeof(cli_addr);
+        newsockfd = accept(sockfd, 
+                (struct sockaddr *) &cli_addr, 
+                (socklen_t*)&clilen);
+        if (newsockfd < 0) 
+                error("ERROR on accept");
+
+        // Begin parsing the messages
+        StartMessageParser(newsockfd);
+
+        return 0;
 }
 
@@ -275 +300 @@
         cout << "Configuration file parsing completed." << endl;
 
-        StartShell();
+        // Open cognitive engine port   
+        StartShell(30000);
+
+        // Open policy engine port
+//      StartShell(30001);
 
    return 1;