#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <stdio.h>
#include <event.h>
//#include <memory.h>
//#include <malloc.h>
#define PORT 7000
#define BACKLOG 5
#define MEM_SIZE 1024
struct event_base* base;
struct sock_ev {
struct event* read_ev;
struct event* write_ev;
char* buffer;
struct sockaddr_in* pcli_addr;
};
void release_sock_event(struct sock_ev* ev)
{
event_del(ev->read_ev);
free(ev->read_ev);
free(ev->write_ev);
free(ev->buffer);
free(ev->pcli_addr);
free(ev);
}
void on_write(int sock, short event, void* arg)
{
char* buffer = (char*)arg;
send(sock, buffer, strlen(buffer), 0);
free(buffer);
}
void on_read(int sock, short event, void* arg)
{
struct event* write_ev;
int size;
struct sock_ev* ev = (struct sock_ev*)arg;
ev->buffer = (char*)malloc(MEM_SIZE);
bzero(ev->buffer, MEM_SIZE);
size = recv(sock, ev->buffer, MEM_SIZE, 0);
//printf("ready get ip port...\n");
//printf("point 2address%d\n",ev->pcli_addr);
char ip[INET_ADDRSTRLEN];
memset(ip,0,sizeof(ip));
inet_ntop(AF_INET,&(ev->pcli_addr->sin_addr),ip,INET_ADDRSTRLEN);//sprintf(ip,inet_ntoa(ev->pcli_addr->sin_addr));
unsigned short port=ev->pcli_addr->sin_port;
port=htons(port);
printf("from %s:%d,receive data:%s, size:%d\n", ip,port,ev->buffer, size);
if (size == 0) {
release_sock_event(ev);
close(sock);
return;
}
event_set(ev->write_ev, sock, EV_WRITE, on_write, ev->buffer);
event_base_set(base, ev->write_ev);
event_add(ev->write_ev, NULL);
}
void on_accept(int sock, short event, void* arg)
{
printf("on_accept.....coming one connection...\n");
struct sockaddr_in cli_addr;
int newfd, sin_size;
struct sock_ev* ev = (struct sock_ev*)malloc(sizeof(struct sock_ev));
ev->read_ev = (struct event*)malloc(sizeof(struct event));
ev->write_ev = (struct event*)malloc(sizeof(struct event));
ev->pcli_addr=(struct sockaddr_in*)malloc(sizeof(struct sockaddr_in));
sin_size = sizeof(struct sockaddr_in);
newfd = accept(sock, (struct sockaddr*)&cli_addr, &sin_size);
printf("point address%d\n",ev->pcli_addr);
memcpy(ev->pcli_addr,&cli_addr,sizeof(struct sockaddr_in)); //*ev->pcli_addr=cli_addr; //copy mem;
// char ip[INET_ADDRSTRLEN];
// memset(ip,0,sizeof(ip));
// inet_ntop(AF_INET,&(ev->pcli_addr->sin_addr),ip,INET_ADDRSTRLEN);//sprintf(ip,inet_ntoa(ev->pcli_addr->sin_addr));
// unsigned short port=ev->pcli_addr->sin_port;
// port=htons(port);
//
// printf("from %s:%d \n", ip,port);
event_set(ev->read_ev, newfd, EV_READ|EV_PERSIST, on_read, ev);
event_base_set(base, ev->read_ev);
event_add(ev->read_ev, NULL);
printf("on_accept.....end...\n");
}
int main(int argc, char* argv[])
{
struct sockaddr_in my_addr;
int sock;
sock = socket(AF_INET, SOCK_STREAM, 0);
int yes = 1;
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
memset(&my_addr, 0, sizeof(my_addr));
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(PORT);
my_addr.sin_addr.s_addr = INADDR_ANY;
bind(sock, (struct sockaddr*)&my_addr, sizeof(struct sockaddr));
listen(sock, BACKLOG);
struct event listen_ev;
base = event_base_new();
event_set(&listen_ev, sock, EV_READ|EV_PERSIST, on_accept, NULL);
event_base_set(base, &listen_ev);
event_add(&listen_ev, NULL);
event_base_dispatch(base);
return 0;
}