nodejs/node_http.cc

#include "node_http.h"
#include "node.h"

#include <oi_socket.h>
#include <ebb_request_parser.h>

#include <string>
#include <list>

using namespace v8;
using namespace std;

static Persistent<ObjectTemplate> request_template;

// globals
static Persistent<String> path_str; 
static Persistent<String> uri_str; 
static Persistent<String> query_string_str; 
static Persistent<String> fragment_str; 
static Persistent<String> method_str; 
static Persistent<String> http_version_str; 
static Persistent<String> headers_str; 

static Persistent<String> on_request_str; 
static Persistent<String> on_body_str; 
static Persistent<String> respond_str; 

static Persistent<String> copy_str;
static Persistent<String> delete_str;
static Persistent<String> get_str;
static Persistent<String> head_str;
static Persistent<String> lock_str;
static Persistent<String> mkcol_str;
static Persistent<String> move_str;
static Persistent<String> options_str;
static Persistent<String> post_str;
static Persistent<String> propfind_str;
static Persistent<String> proppatch_str;
static Persistent<String> put_str;
static Persistent<String> trace_str;
static Persistent<String> unlock_str;

#define INVALID_STATE_ERR 1

class Server {
public:
  Server (Handle<Object> _js_server);
  ~Server ();

  int Start(struct addrinfo *servinfo);
  void Stop();

  Handle<Value> Callback()
  {
    HandleScope scope;
    Handle<Value> value = js_server->Get(on_request_str);
    return scope.Close(value);
  }

private:
  oi_server server;
  Persistent<Object> js_server;
};

class HttpRequest;

class Connection {
public:
  Connection();
  ~Connection();

  void Parse(const void *buf, size_t count);
  void Write();
  void Close();
  void AddRequest (HttpRequest *request);

  oi_socket socket;
  Persistent<Function> js_onrequest;

private:
  ebb_request_parser parser;
  list<HttpRequest*> requests;
  list<HttpRequest*> finished_requests;
  friend class Server;
};

class HttpRequest {
 public:
  HttpRequest (Connection &c);
  /* Deleted from C++ as soon as possible.
   * Javascript object might linger. This is okay
   */
  ~HttpRequest();

  void MakeBodyCallback (const char *base, size_t length);
  Local<Object> CreateJSObject ();
  void Respond (Handle<Value> data);

  string path;
  string query_string;
  string fragment;
  string uri;

  list<string> header_fields;
  list<string> header_values;

  Connection &connection;
  ebb_request parser_info;

  list<oi_buf*> output;
  bool done;
  Persistent<Object> js_object;
};

static Handle<Value>
GetMethodString (int method)
{
  switch(method) {
    case EBB_COPY:      return copy_str;
    case EBB_DELETE:    return delete_str;
    case EBB_GET:       return get_str;
    case EBB_HEAD:      return head_str;
    case EBB_LOCK:      return lock_str;
    case EBB_MKCOL:     return mkcol_str;
    case EBB_MOVE:      return move_str;
    case EBB_OPTIONS:   return options_str;
    case EBB_POST:      return post_str;
    case EBB_PROPFIND:  return propfind_str;
    case EBB_PROPPATCH: return proppatch_str;
    case EBB_PUT:       return put_str;
    case EBB_TRACE:     return trace_str;
    case EBB_UNLOCK:    return unlock_str;
  }
  return Null();
}

static Handle<Value>
RespondCallback (const Arguments& args) 
{
  HandleScope scope;

  Handle<Value> v = args.Holder()->GetInternalField(0);
  if(v->IsUndefined()) {
    // check that args.Holder()->GetInternalField(0)
    // is not NULL if so raise INVALID_STATE_ERR
    printf("null request external\n");
    ThrowException(Integer::New(INVALID_STATE_ERR)); 
    return Undefined();
  }
  Handle<External> field = Handle<External>::Cast(v);
  HttpRequest* request = static_cast<HttpRequest*>(field->Value());
  request->Respond(args[0]);
  return Undefined();
}

void
HttpRequest::Respond (Handle<Value> data)
{
  if(data == Null()) {
    done = true;
  } else {
    Handle<String> s = data->ToString();
    oi_buf *buf = oi_buf_new2(s->Length());

    uint16_t expanded[s->Length()];
    s->Write(expanded, 0, s->Length());

    for(int i = 0; i < s->Length(); i++) {
      buf->base[i] = expanded[i];
    }

    output.push_back(buf);
  }
  connection.Write();
}


static void
on_path (ebb_request *req, const char *buf, size_t len)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);
  request->path.append(buf, len);
}

static void
on_uri (ebb_request *req, const char *buf, size_t len)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);
  request->uri.append(buf, len);
}

static void
on_query_string (ebb_request *req, const char *buf, size_t len)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);
  request->query_string.append(buf, len);
}

static void
on_fragment (ebb_request *req, const char *buf, size_t len)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);
  request->fragment.append(buf, len);
}

static const char upcase[] =
  "\0______________________________"
  "_________________0123456789_____"
  "__ABCDEFGHIJKLMNOPQRSTUVWXYZ____"
  "__ABCDEFGHIJKLMNOPQRSTUVWXYZ____"
  "________________________________"
  "________________________________"
  "________________________________"
  "________________________________";

static void
on_header_field (ebb_request *req, const char *buf, size_t len, int header_index)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);

  char upbuf[len];

  for(int i = 0; i < len; i++) 
    upbuf[i] = upcase[buf[i]];

  if( request->header_fields.size() == header_index - 1) {
    request->header_fields.back().append(upbuf, len);
  } else { 
    request->header_fields.push_back( string(upbuf, len) );
  }
}

static void
on_header_value (ebb_request *req, const char *buf, size_t len, int header_index)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);

  if( request->header_values.size() == header_index - 1) {
    request->header_values.back().append(buf, len);
  } else { 
    request->header_values.push_back( string(buf, len) );
  }
}

static void
on_headers_complete (ebb_request *req)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);

  HandleScope scope;

  Handle<Object> js_request = request->CreateJSObject();

  // Set up an exception handler before calling the Process function
  TryCatch try_catch;

  // Invoke the process function, giving the global object as 'this'
  // and one argument, the request.
  const int argc = 1;
  Handle<Value> argv[argc] = { js_request };
  Handle<Value> r = request->connection.js_onrequest->Call(Context::GetCurrent()->Global(), argc, argv);

  if(try_catch.HasCaught())
    node_fatal_exception(try_catch);
}

static void
on_request_complete (ebb_request *req)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);
  request->MakeBodyCallback(NULL, 0); // EOF
}

static void
on_body (ebb_request *req, const char *base, size_t length)
{
  HttpRequest *request = static_cast<HttpRequest*> (req->data);

  if(length)
    request->MakeBodyCallback(base, length);
}

static ebb_request * on_request
  ( void *data
  ) 
{
  Connection *connection = static_cast<Connection*> (data);

  HttpRequest *request = new HttpRequest(*connection);
  connection->AddRequest(request);
  
  return &request->parser_info;
}

static void on_read 
  ( oi_socket *socket
  , const void *buf
  , size_t count
  )
{
  Connection *connection = static_cast<Connection*> (socket->data);
  if(count == 0) {
    connection->Close();
  } else {
    //write(1, buf, count);
    connection->Parse(buf, count);  
  }
}

static void on_close 
  ( oi_socket *socket
  )
{
  Connection *connection = static_cast<Connection*> (socket->data);
  delete connection;
}

HttpRequest::~HttpRequest ()
{
  HandleScope scope; // needed?
  // delete a reference c++ HttpRequest
  js_object->SetInternalField(0, Undefined());
  js_object->Delete(respond_str);
  // dispose of Persistent handle so that 
  // it can be GC'd normally.
  js_object.Dispose();
}

HttpRequest::HttpRequest (Connection &c) : connection(c)
{
  ebb_request_init(&parser_info); 
  parser_info.on_path             = on_path;
  parser_info.on_query_string     = on_query_string;
  parser_info.on_uri              = on_uri;
  parser_info.on_fragment         = on_fragment;
  parser_info.on_header_field     = on_header_field;
  parser_info.on_header_value     = on_header_value;
  parser_info.on_headers_complete = on_headers_complete;
  parser_info.on_body             = on_body;
  parser_info.on_complete         = on_request_complete;
  parser_info.data                = this;

  done = false;
}

void
HttpRequest::MakeBodyCallback (const char *base, size_t length)
{
  HandleScope handle_scope;

  Handle<Value> onbody_val = js_object->Get(on_body_str);  
  if (!onbody_val->IsFunction()) return;
  Handle<Function> onbody = Handle<Function>::Cast(onbody_val);

  TryCatch try_catch;
  const int argc = 1;
  Handle<Value> argv[argc];
  
  if(length) {
    // TODO ByteArray?
    //
    
    uint16_t expanded_base[length];
    for(int i = 0; i < length; i++) {
      expanded_base[i] = base[i];
    }

    Handle<String> chunk = String::New(expanded_base, length);
    argv[0] = chunk;
  } else {
    argv[0] = Null();
  }

  Handle<Value> result = onbody->Call(js_object, argc, argv);

  if(try_catch.HasCaught())
    node_fatal_exception(try_catch);
}

Local<Object>
HttpRequest::CreateJSObject ()
{
  HandleScope scope;

  if (request_template.IsEmpty()) {
    Handle<ObjectTemplate> raw_template = ObjectTemplate::New();
    raw_template->SetInternalFieldCount(1);
    raw_template->Set(respond_str, FunctionTemplate::New(RespondCallback));

    request_template = Persistent<ObjectTemplate>::New(raw_template);
  }

  // Create an empty http request wrapper.
  Handle<Object> result = request_template->NewInstance();

  // Wrap the raw C++ pointer in an External so it can be referenced
  // from within JavaScript.
  Handle<External> request_ptr = External::New(this);

  // Store the request pointer in the JavaScript wrapper.
  result->SetInternalField(0, request_ptr);

  result->Set ( path_str
              , String::New(path.c_str(), path.length())
              );

  result->Set ( uri_str
              , String::New(uri.c_str(), uri.length())
              );

  result->Set ( query_string_str
              , String::New(query_string.c_str(), query_string.length())
              );

  result->Set ( fragment_str
              , String::New(fragment.c_str(), fragment.length())
              );

  result->Set ( method_str
              , GetMethodString(parser_info.method)
              );

  char version[10];
  snprintf ( version
           , 10 // big enough? :)
           , "%d.%d"
           , parser_info.version_major
           , parser_info.version_minor
           ); 
  result->Set ( http_version_str
              , String::New(version)
              );

  
  Handle<Object> headers = Object::New();
  list<string>::iterator field_iterator = header_fields.begin();
  list<string>::iterator value_iterator = header_values.begin();
  while( value_iterator != header_values.end() ) {
    string &f = *field_iterator;
    string &v = *value_iterator;
    
    headers->Set( String::NewSymbol(f.c_str(), f.length())
                , String::New(v.c_str(), v.length() ) 
                );

    field_iterator++;
    value_iterator++;
  }
  result->Set(headers_str, headers);

  js_object = Persistent<Object>::New(result);
  // XXX does the request's js_object need a MakeWeak callback?
  // i dont think so because at some point the connection closes
  // and we're going to delete the request. 
  
  return scope.Close(result);
}


static oi_socket*
on_connection (oi_server *_server, struct sockaddr *addr, socklen_t len)
{
  HandleScope scope;

  Server *server = static_cast<Server*> (_server->data);

  Handle<Value> callback_v = server->Callback();

  if(callback_v == Undefined())
    return NULL;

  Connection *connection = new Connection();

  Handle<Function> f = Handle<Function>::Cast(callback_v);
  connection->js_onrequest = Persistent<Function>::New(f);

  return &connection->socket;
}

Connection::Connection ()
{
  oi_socket_init (&socket, 30.0); // TODO make timeout adjustable
  socket.on_read    = on_read;
  socket.on_error   = NULL;
  socket.on_close   = on_close;
  socket.on_timeout = NULL;
  socket.on_drain   = NULL;
  socket.data       = this;

  ebb_request_parser_init (&parser);
  parser.new_request = on_request;
  parser.data        = this;
}

Connection::~Connection ()
{
  list<HttpRequest*>::iterator it = requests.begin();

  // delete all the requests

  for(it = requests.begin(); it != requests.end(); it++)
    delete *it;

  for(it = finished_requests.begin(); it != finished_requests.end(); it++)
    delete *it;
}

void
Connection::Parse(const void *buf, size_t count)
{
  // FIXME change ebb_request_parser to have void* arg
  ebb_request_parser_execute ( &parser
                             , static_cast<const char*> (buf) 
                             , count
                             );

  if(ebb_request_parser_has_error(&parser)) {
    fprintf(stderr, "parse error closing connection\n");
    oi_socket_close(&socket);
  }
}

void 
Connection::AddRequest(HttpRequest *request)
{
  requests.push_back(request);
}

void
Connection::Write ( ) 
{
  if(requests.size() == 0)
    return;

  HttpRequest *request = requests.front(); 

  while(request->output.size() > 0) {
    oi_buf *buf = request->output.front();
    oi_socket_write(&socket, buf);
    request->output.pop_front();
  }

  if(request->done) {
    if(!ebb_request_should_keep_alive(&request->parser_info)) {
      socket.on_drain = oi_socket_close;
    } 

    requests.pop_front();
    finished_requests.push_back(request);

    Write();
  }
}

void
Connection::Close ( ) 
{
  oi_socket_close(&socket);
}

static void
server_destroy (Persistent<Value> _, void *data)
{
  Server *server = static_cast<Server *> (data);
  delete server;
}

Server::Server (Handle<Object> _js_server)
{
  oi_server_init(&server, 1024);
  server.on_connection = on_connection;
  server.data = this;
  HandleScope scope;
  js_server = Persistent<Object>::New (_js_server);
  // are we ever going to need this external?
  js_server->SetInternalField (0, External::New(this));
  js_server.MakeWeak (this, server_destroy);
}

Server::~Server ()
{
  Stop();
  js_server.Dispose();
  js_server.Clear(); // necessary? 
}

int
Server::Start(struct addrinfo *servinfo) 
{
  int r = oi_server_listen(&server, servinfo);
  if(r == 0)
    oi_server_attach(&server, node_loop());
  return r;
}

void
Server::Stop() 
{
  oi_server_close (&server);
  oi_server_detach (&server);
}

/* This constructor takes 2 arguments: host, port. */
static Handle<Value>
newHTTPServer (const Arguments& args) 
{
  if (args.Length() < 3)
    return Undefined();

  HandleScope scope;

  char *host = NULL; 
  String::AsciiValue host_v(args[0]->ToString());
  if(args[0]->IsString()) {
    host = *host_v;
  }
  String::AsciiValue port(args[1]->ToString());

  Handle<Function> onrequest = Handle<Function>::Cast(args[2]);
  args.This()->Set(on_request_str, onrequest);

  // get addrinfo for localhost, PORT
  struct addrinfo *servinfo;
  struct addrinfo hints;
  memset(&hints, 0, sizeof hints);
  hints.ai_family = AF_UNSPEC;
  hints.ai_socktype = SOCK_STREAM;
  hints.ai_flags = AI_PASSIVE;
  // FIXME BLOCKING
  int r = getaddrinfo(host, *port, &hints, &servinfo);
  if (r != 0)
    return Undefined(); // XXX raise error?

  //
  //
  //
  // TODO host is ignored for now assumed localhost
  //
  //
  //
  //

  Server *server = new Server(args.This());
  if(server == NULL)
    return Undefined(); // XXX raise error?

  r = server->Start(servinfo);
  if (r != 0)
    return Undefined(); // XXX raise error?

  return args.This();
}

void
Init_http (Handle<Object> target)
{
  HandleScope scope;

  Local<FunctionTemplate> server_t = FunctionTemplate::New(newHTTPServer);
  server_t->InstanceTemplate()->SetInternalFieldCount(1);
  
  server_t->Set("INVALID_STATE_ERR", Integer::New(INVALID_STATE_ERR));

  target->Set(String::New("HTTPServer"), server_t->GetFunction());

  path_str         = Persistent<String>::New( String::NewSymbol("path") );
  uri_str          = Persistent<String>::New( String::NewSymbol("uri") );
  query_string_str = Persistent<String>::New( String::NewSymbol("query_string") );
  fragment_str     = Persistent<String>::New( String::NewSymbol("fragment") );
  method_str       = Persistent<String>::New( String::NewSymbol("method") );
  http_version_str = Persistent<String>::New( String::NewSymbol("http_version") );
  headers_str      = Persistent<String>::New( String::NewSymbol("headers") );

  on_request_str = Persistent<String>::New( String::NewSymbol("onrequest") );
  on_body_str    = Persistent<String>::New( String::NewSymbol("onbody") );
  respond_str    = Persistent<String>::New( String::NewSymbol("respond") );

  copy_str      = Persistent<String>::New( String::New("COPY") );
  delete_str    = Persistent<String>::New( String::New("DELETE") );
  get_str       = Persistent<String>::New( String::New("GET") );
  head_str      = Persistent<String>::New( String::New("HEAD") );
  lock_str      = Persistent<String>::New( String::New("LOCK") );
  mkcol_str     = Persistent<String>::New( String::New("MKCOL") );
  move_str      = Persistent<String>::New( String::New("MOVE") );
  options_str   = Persistent<String>::New( String::New("OPTIONS") );
  post_str      = Persistent<String>::New( String::New("POST") );
  propfind_str  = Persistent<String>::New( String::New("PROPFIND") );
  proppatch_str = Persistent<String>::New( String::New("PROPPATCH") );
  put_str       = Persistent<String>::New( String::New("PUT") );
  trace_str     = Persistent<String>::New( String::New("TRACE") );
  unlock_str    = Persistent<String>::New( String::New("UNLOCK") );
}