server.cpp

#include <Poco/Net/ServerSocket.h>
#include <Poco/Net/HTTPServer.h>
#include <Poco/Net/HTTPRequestHandler.h>
#include <Poco/Net/HTTPRequestHandlerFactory.h>
#include <Poco/Net/HTTPResponse.h>
#include <Poco/Net/HTTPServerRequest.h>
#include <Poco/Net/HTTPServerResponse.h>
#include <Poco/Util/ServerApplication.h>
#include <iostream>
#include <string>
#include <vector>
#include <stdio.h>

using namespace Poco::Net;
using namespace Poco::Util;
using namespace std;



bool is_numeric( char *number_string){
  string temp_number_string(number_string);
  return (temp_number_string.find_first_not_of( "0123456789" ) == string::npos);
}



class MyGetRequestHandler : public HTTPRequestHandler {
public:
  virtual void handleRequest(HTTPServerRequest &req, HTTPServerResponse &resp){
    resp.setStatus(HTTPResponse::HTTP_OK);
    resp.setContentType("text/html");
    
    // Generate requested HTML file
    ostream& out = resp.send();
    out << "<h1>Hello world! This files name is " << req.getURI() << "</h1>"
        << "<p>Host: "           << req.getHost()   << "</p>"
        << "<p>Method: "         << req.getMethod() << "</p>";
    out.flush();
    
    cout << endl << "Client requested: =" << req.getURI() << endl;  
  }
};

class MyPutRequestHandler : public HTTPRequestHandler {
public:
  virtual void handleRequest(HTTPServerRequest &req, HTTPServerResponse &resp){
    resp.setStatus(HTTPResponse::HTTP_OK);
    resp.setContentType("text/html");
    
    string fileName = req.getURI();
    
    ostream& out = resp.send();
    out << "<h1>Hello friend!</h1>"
        << "<p>Host: "     << req.getHost()   << "</p>"
        << "<p>Method: "   << req.getMethod() << "</p>"
        << "<p>Filename: " << req.getURI()    << "</p>";
    
    out.flush();
   
    /** Does not work
      StreamSocket strs = req.acceptConnection();
      SocketStream ostr(strs);
      std::string file("test.in");
      Poco::FileInputStream istr(file, std::ios::binary);
      StreamCopier::copyStream(istr, ostr);
    */
    
    cout << endl << "URI=" << req.getURI() << endl;  
  }
};

class MyRequestHandlerFactory : public HTTPRequestHandlerFactory 
{
public:
  virtual HTTPRequestHandler* createRequestHandler(const HTTPServerRequest &req) {
    
    string httpMethod = req.getMethod();
    
    if( httpMethod.compare( HTTPRequest::HTTP_GET ) == 0 ){
      return new MyGetRequestHandler;

    } else if( httpMethod.compare( HTTPRequest::HTTP_PUT ) == 0 ) {     
      return  new MyPutRequestHandler;

    } else {
      // Other request type

    }
  }
};

class MyServerApp : public ServerApplication {
protected:
  int main(const vector<string> &args){
    
    HTTPServer s(new MyRequestHandlerFactory, ServerSocket( atoi( args[1].c_str() ) ), new HTTPServerParams );
    
    s.start();
    cout << endl << "Server started" << endl;
    cout << "Listening on port: " << args[1] << endl ; 
    
    waitForTerminationRequest();  
    
    cout << endl << "Shutting down server..." << endl;
    s.stop();
    
    return Application::EXIT_OK;
  }
};


void usage(){
  cout << "Usage: ./server <PORT_NUMBER>" << endl;
  cout << "Please provide a port in range of 1024 - 49151 for the server to run on. Recommended: 300" << endl;
  exit(-1);
}

int main(int argc, char* argv[]){
  
  
  if( argc <= 1 ) {  // Check that at least one argument was given. 
    cout << "Missing arguments" << endl << endl; 
    usage();
    
  } else if( !is_numeric( argv[1] )){ // Check that only strings containing numbers were passed
    cout << '"' << argv[1] << "\" is not a valid port number." << endl << endl;
    usage();
    
  } else if( not (1023 < atoi( argv[1] ) and atoi( argv[1] ) < 49152) ){ // Check that port was in specified range
    cout << argv[1] << " is out of the specified range. Please enter a valid port number." << endl << endl;
    usage();
    
  } else {
    // Nothing to do here, all good 
  }
  
  
  MyServerApp app;
  return app.run(argc, argv);
}