// ------------------------------- //
// -------- Start of File -------- //
// ------------------------------- //
// ----------------------------------------------------------- // 
// C++ Source Code File Name: server.cpp
// Compiler Used: MSVC, GCC
// Produced By: DataReel Software Development Team
// File Creation Date: 10/17/2001
// Date Last Modified: 06/17/2016
// Copyright (c) 2001-2016 DataReel Software Development
// ----------------------------------------------------------- // 
// ------------- Program Description and Details ------------- // 
// ----------------------------------------------------------- // 
/*
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
 
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Lesser General Public License for more details.
 
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  
USA

This is a test program used demonstrate the use of the gxSSL
class in a multi-threaded HTTPS server application.
*/
// ----------------------------------------------------------- //   
#include "gxdlcode.h"

#if defined (__USE_ANSI_CPP__) // Use the ANSI Standard C++ library
#include <iostream>
using namespace std; // Use unqualified names for Standard C++ library
#else // Use the old iostream library by default
#include <iostream.h>
#endif // __USE_ANSI_CPP__

#include <string.h>
#include <stdlib.h>
#include "m_thread.h"

class ConsoleThread : public gxThread
{
public:
  ConsoleThread() { }
  ~ConsoleThread() { }

private:
  void *ThreadEntryRoutine(gxThread_t *thread);
};

void *ConsoleThread::ThreadEntryRoutine(gxThread_t *thread)
{
  const int cmd_len = 255;
  char sbuf[cmd_len];
  
  while(1) {
    for(int i = 0; i < cmd_len; i++) sbuf[i] = 0;
    cin >> sbuf;
    if(strcmp(sbuf, "quit") == 0) break;
    if(strcmp(sbuf, "exit") == 0) break;
    cout << "Invalid command" << "\n" << flush;
    cout << "Enter quit to exit" << "\n" << flush;
  }
  return (void *)0;
}

void KeyGenCallback(int p, int n, void *arg)
{
  char c = 'B';
  if (p == 0) c = '.';
  if (p == 1) c = '+';
  if (p == 2) c = '*';
  if (p == 3) c = '\n';
  cout << c << flush;
}

void nt_SSL_error_print(gxSSL *ssl) 
// Non-thread safe error message print
{
  if(!ssl) return;
  gxString err;
  cout << ssl->SSLExceptionMessage() << "\n" << flush;
  gxSSL::GlobalSSLError(err);
  cout << err.c_str() << "\n" << flush;
}

void InputData(const char *prompt, gxString &data)
// Function used to read a string from the input stream.
{
  data.Clear();
  if(!prompt) return;
  char buf[1024];
  memset(buf, 0, 1024);
  cout << prompt << ": " << flush;
  cin.getline(buf, sizeof(buf));
  data << buf;
}

int main(int argc, char **argv)
{
  // Check arguments. Should be only one: the port number to bind to.
  if(argc != 2) {
    cerr << "Usage: " << argv[0] << " port" << "\n" << flush;
    return 1;
  }

  ServerThread server;
  gxSSL *ssl = &server.openssl;
  gxRandom rand_num;

  servercfg->port = (gxsPort_t) atoi(argv[1]);
  const char *private_key_fname = "server_private_key.pem";
  const char *cert_req_fname = "server_cert.pem";
  const char *dh_fname = "dhparms.pem";

  if(!futils_exists(dh_fname)) {
    cout << "Generating Diffie-Hellman parameters file" << "\n" << flush;
    if(ssl->InitPRNG() != gxSSL_NO_ERROR) {
      nt_SSL_error_print(ssl);
      return 1;
    }
    if(ssl->MakeDHParms(dh_fname) != gxSSL_NO_ERROR) {
      nt_SSL_error_print(ssl);
      return 1;
    }
  }

  if(!futils_exists(private_key_fname)) {
    cout << "Generating private RSA key" << "\n" << flush;

    if(ssl->InitPRNG() != gxSSL_NO_ERROR) {
      nt_SSL_error_print(ssl);
      return 1;
    }
    if(ssl->MakeRSAPrivateKey(private_key_fname, 2048,
			      KeyGenCallback, 0) != gxSSL_NO_ERROR) {
      nt_SSL_error_print(ssl);
      return 1;
    }

    cout << "\n" << flush;
    cout << "Generating and self-signing certificate request" << "\n" << flush;
    gxCertIssuerInfo ci;
    ci.sn = rand_num.RandomNumber();
    ci.days = 365;
    InputData("Country Name XX", ci.C);
    if(ci.C.is_null()) ci.C = "US";
    InputData("State or Province", ci.ST);
    if(ci.ST.is_null()) ci.ST = "MyState";
    InputData("City or Locality", ci.L);
    if(ci.L.is_null()) ci.L = "MyCity";
    InputData("Organization Name", ci.O);
    if(ci.O.is_null()) ci.O = "My Org. Name";
    InputData("Organizational Unit", ci.OU); 
    if(ci.OU.is_null()) ci.OU = "My Org. Unit";
    InputData("Common Name (or site name)", ci.CN);
    if(ci.CN.is_null()) ci.CN = "my.server.com";

    // Add standard extensions if needed by application
    // ci.basic_constraints = "critical,CA:TRUE";
    // ci.key_usage = "critical,keyCertSign,cRLSign";
    // ci.subject_key_identifier = "hash";
  
    // Add Netscape specific extensions if needed by application
    // ci.netscape_cert_type = "sslCA";
    // ci.netscape_comment = "my comment extension";

    if(ssl->MakeCertReq(private_key_fname, ci, cert_req_fname)!= 
       gxSSL_NO_ERROR) {
      nt_SSL_error_print(ssl);
      return 1;
    }
  }

  cout << "Initializing HTTPS server..." << "\n" << flush;
  int rv = server.InitServer(servercfg->port);
  if(rv != 0) {
    cout << server.SocketExceptionMessage() << "\n" << flush;
    return 1;
  }
  
  if(!gxSSL::InitSSLibrary()) {
    cout << "Error initializing SSL library" << "\n" << flush;  
    nt_SSL_error_print(ssl);
    return 1;
  }
  cout << "Initializing SSL context object..." << "\n" << flush;
  if(!ssl->InitCTX(cert_req_fname, private_key_fname)) {
    nt_SSL_error_print(ssl);
    return 1;
  }

  cout << "Loading Diffie-Hellman parameters..." << "\n" << flush;
  if(ssl->LoadDHParms(dh_fname) != gxSSL_NO_ERROR) {
    nt_SSL_error_print(ssl);
    return 1;
  }

  // Get the host name assigned to this machine
  char hostname[gxsMAX_NAME_LEN];
  rv = server.GetHostName(hostname);
  if(rv != 0) {
    cout << server.SocketExceptionMessage() << "\n" << flush;
    return 1;
  }

  cout << "Opening HTTPS server on host " << hostname << "\n" << flush;
  cout << "Enter quit to exit" << "\n" << flush;
  cout << "\n" << flush;
  
  cout << "Listening on port " << servercfg->port << "\n" << flush;
  cout.flush();

 gxThread_t *server_thread = server.CreateThread();
  if(server_thread->GetThreadError() != gxTHREAD_NO_ERROR) {
    cout << server_thread->ThreadExceptionMessage() << "\n" << flush;
    return 1;
  }
  
  ConsoleThread console;
  gxThread_t *console_thread = console.CreateThread();
  if(console_thread->GetThreadError() != gxTHREAD_NO_ERROR) {
    cout << console_thread->ThreadExceptionMessage() << "\n" << flush;
    return 1;
  }
    
  if(console.JoinThread(console_thread) != 0) {
    cout << "Could not join the console thread" << "\n" << flush;
  }

  // Destroy the serial port receiver thread
  cout << "Stopping the server thread..." << "\n" << flush;
  servercfg->accept_clients = 0;

  // Cannot the server thread due to the blocking accept() function, so
  // cancel the thread before it is destroyed
  server.CancelThread(server_thread);
  if(server_thread->GetThreadState() == gxTHREAD_STATE_CANCELED) {
    cout << "The server thread was stopped" << "\n" << flush;
  }
  else {
    cout << "Could not cancel the server thread" << "\n" << flush;
  }

  if(server.JoinThread(server_thread) != 0) {
    cout << "Could not join the server thread" << "\n" << flush;
  }
    
  delete server_thread;
  delete console_thread;

  cout << "Exiting..." << "\n" << flush;
  return 0;
}
// ----------------------------------------------------------- // 
// ------------------------------- //
// --------- End of File --------- //
// ------------------------------- //