/************************************
 * For msmr 
 * server.c
 * tesing the speed of bufferevent_write
 * 2015-02-03
 * author@tom
************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <event2/event.h>
#include <event2/bufferevent.h>
#include <event2/buffer.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>

#include <unistd.h>

#include "ev_proto.h"
#include "face_daemon_proto.h"

/*
void make_msg(char* mesg, int& length)
{
	EVPHeader* evpHeader = new (mesg) EVPHeader;
	evpHeader->proto = EVPProto::EVPP_PROTOBUF;
	evpHeader->cmd = FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_PB;
	evpHeader->size = length;
	
	FILE* pFile = fopen("facelist-3.pb", "rb");
	size_t fsize = fread(mesg + sizeof(EVPHeader), 1, length - sizeof(EVPHeader), pFile);
	fclose(pFile);
}
*/

void make_msg(char* mesg, int& length)
{
	FILE* pFile = fopen("facelist-1001-0-5.pb", "rb");
	length = fread(mesg, 1, length, pFile);
	fclose(pFile);
}

int main()
{
    // build the message to be sent
    int length = 1024 * 1024; // the size of message
    char* mesg = (char*)malloc((length+1)*sizeof(char)); // Look out the end mark '/0' of a C string
    if (mesg == NULL)
        exit(1);
    int i;
    //for (i=0; i<length; i++) 
    //    strcat(mesg, "a");

	make_msg(mesg, length);

    printf("%s\n", mesg);
    printf("%d\n", (int)strlen(mesg));

    // build socket
    int port = 5432;

    struct sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = inet_addr("192.168.1.146");
    server_addr.sin_port = htons(port);

    // build event base
    struct event_base* base = event_base_new();

    // set TCP_NODELAY to let data arrive at the server side quickly
    evutil_socket_t fd;
    fd = socket(AF_INET, SOCK_STREAM, 0);
    struct bufferevent* conn = bufferevent_socket_new(base,fd,BEV_OPT_CLOSE_ON_FREE);
    int enable = 1;
    if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (void*)&enable, sizeof(enable)) < 0)
       printf("ERROR: TCP_NODELAY SETTING ERROR!\n");
    //bufferevent_setcb(conn, NULL, NULL, NULL, NULL); // For client, we don't need callback function
    bufferevent_enable(conn, EV_WRITE);
    if(bufferevent_socket_connect(conn,(struct sockaddr*)&server_addr,sizeof(server_addr)) == 0)
            printf("connect success\n");

    // start to send data
    bufferevent_write(conn,mesg,length);
    // check the output evbuffer
    struct evbuffer* output = bufferevent_get_output(conn);

    event_base_dispatch(base);
 
	char readbuf[10];
	int readbufsize = read(fd, readbuf, sizeof(readbuf));
	while(readbufsize>0)
	{
		readbuf[readbufsize] = '\0';
		printf("%s", readbuf);
		readbufsize = read(fd, readbuf, sizeof(readbuf));
	}
	printf("\n");

    free(mesg);
    mesg = NULL;

    bufferevent_free(conn);
    event_base_free(base);

    printf("Client program is over\n");

    return 0;
}