main.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <unistd.h>

#include <parseUtils.h>

/*#define BUFFER_SIZE 4096

static const char* szret[] = { "I get a correct result\n", "Something wrong\n" };*/
/* master program to parse http header*/
int main( int argc, char* argv[] )
{
  if( argc <= 2 )
  {
    printf( "usage: %s ip_address port_number\n", basename( argv[0] ) );
    return 1;
  }
  
  const char* ip = argv[1];
  int port = atoi( argv[2] );
  
  struct scokaddr_in address;
  bzero( &address, sizeof( address ) );
  address.sin_family = AF_INET;
  inet_pton( AF_INET, ip, &address.sin_addr );
  address.sin_port = htons(port);
  
  int listenfd = socket( PF_INET, SOCK_STREAM, 0 );
  assert( listenfd >= 0 );
  
  int ret = bind( listenfd, ( struct sockaddr* ) &address, sizeof( address ) );
  //assert( -1 != ret );
  if ( -1 != ret )
  {
    swhitch (errno):
      case EACCES：
        printf( "the address is protected, it doesn't have access.\n" );
        return 1;
        //break;
      case EADDRINUSE:
        printf( "the address has been used.\n" )
        return 1;
        //break;
      default:
        printf( "there is a error in bind(...)." );
        return 1;
        //assert(false);
        //break;
  }
  
  ret = listen( listenfd, 5 );
  assert( -1 != ret );
  
  struct sockaddr_in client_address;
  socklent_t client_addrlength = sizeof( client_address );
  int fd = accept( listenfd, ( struct sockaddr* )&client_address, &client_addrlength );
  if( fd < 0 )
  {
    printf("errno is : %d\n", errno);
  }
  else
  {
    char buffer[BUFFER_SIZE];
    memset( buffer, '\0', BUFFER_SIZE );
    int data_read = 0;  
    int read_index = 0;  //当前已经读取了多少客户端数据
    int checked_index = 0;  //当前已经分析完的客户数据。
    int start_line = 0;   //行在buffer中的起始位置
    /*设置主状态机的初始状态*/
    CHECK_STATE checkstate = CHECK_STATE_REQUESTING;
    
    while ( 1 )
    {
      data_read = recv( fd, buffer + read_index, BUFFER_SIZE - read_index, 0);
      if ( -1 == data_read )
      {
        printf("reading failing\n");
        break;
      }
      else if ( 0 == data_read )
      {
        printf("remote client has closed the connection\n");
        break;
      }
      
      read_index += data_read;
      /*分析目前应读取到的客户数据*/
      HTTP_CODE result = parse_content( buffer, checked_index, checkstate );
      int res = 0;
      if ( NO_REQUEST == result )
      {
        continue;
      }
      else if ( GET_REQUEST == result )
      {
        res = send( fd, szret[0], strlen( szret[0] ), 0 );
        if ( -1 == res )
        {
          printf("sending fail");
        }
        break;
      }
      else 
      {
        res = send( fd, szret[1], strlen( szret[1] ), 0 );
         if ( -1 == res )
        {
          printf("sending fail");
        }
        break;
      }
    }
    close( fd );
  }
  
  close( listenfd );
  
  return 0;
}