chenke
2017-08-09 2a12817d4ae31a591f788f3fbb82916e385e9e2d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
/************************************
 * 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;
}