/************************************
|
* 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"
|
|
#include <iostream>
|
#include <logger.h>
|
#include <MaterialBuffer.h>
|
|
Logger g_logger(std::cout);
|
|
/*
|
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)
|
{
|
EVPHeader* evpHeader = new (mesg) EVPHeader;
|
evpHeader->proto = EVPProto::EVPP_RAW_BIN; // 1
|
evpHeader->cmd = FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_COMPARE; // 133
|
evpHeader->size = sizeof(EVPHeader);
|
|
FDP_Image* fdpImage1 = nullptr;
|
{
|
fdpImage1 = new (mesg + evpHeader->size) FDP_Image;
|
|
fdpImage1->db_id = 0; // -1
|
fdpImage1->mb_type = MB_Frame::MBFT_JPEG; // 7
|
fdpImage1->width = 291;
|
fdpImage1->height = 194;
|
|
FILE* pFile = fopen("compare1.jpg", "rb");
|
fdpImage1->size = fread(fdpImage1->buff, 1, length, pFile);
|
fclose(pFile);
|
pFile = nullptr;
|
|
evpHeader->size += sizeof(FDP_Image) + fdpImage1->size;
|
fdpImage1->hton();
|
}
|
|
FDP_Image* fdpImage2 = nullptr;
|
{
|
fdpImage2 = new (mesg + evpHeader->size) FDP_Image;
|
|
fdpImage2->db_id = 0; // -1
|
fdpImage2->mb_type = MB_Frame::MBFT_JPEG; // 14
|
fdpImage2->width = 198;
|
fdpImage2->height = 154;
|
|
FILE* pFile = fopen("compare2.jpg", "rb");
|
fdpImage2->size = fread(fdpImage2->buff, 1, length, pFile);
|
fclose(pFile);
|
pFile = nullptr;
|
|
evpHeader->size += sizeof(FDP_Image) + fdpImage2->size;
|
fdpImage2->hton();
|
}
|
|
//FDP_Image* fdpImage2 = nullptr;
|
//{
|
// fdpImage2 = new (mesg + evpHeader->size) FDP_Image;
|
//
|
// fdpImage2->db_id = 0; // -1
|
// fdpImage2->mb_type = MB_Frame::MBFT_RGB565; // 14
|
// fdpImage2->width = 52;
|
// fdpImage2->height = 52;
|
//
|
// FILE* pFile = fopen("face-13-w52-h52.rgb565", "rb");
|
// fdpImage2->size = fread(fdpImage2->buff, 1, length, pFile);
|
// fclose(pFile);
|
// pFile = nullptr;
|
//
|
// evpHeader->size += sizeof(FDP_Image) + fdpImage2->size;
|
// fdpImage2->hton();
|
//}
|
|
length = evpHeader->size;
|
evpHeader->hton();
|
}
|
|
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", length);
|
|
// build socket
|
int port = 15436;
|
|
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("127.0.0.1");
|
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;
|
}
|