#include <iostream> #include <string> #include <sys/types.h> #include <sys/socket.h> #include <sys/event.h> #include <sys/time.h> #include <netinet/in.h> #include <arpa/inet.h> #include <errno.h> const std::string IP = "192.168.79.18"; const int PORT = 4312; const int MAX_EVENT_COUNT = 5000; const int MAX_RECV_BUFF = 65535; int listener_; char buf_[MAX_RECV_BUFF]; int CreateListener(); bool Register(int kq, int fd); void WaitEvent(int kq); void HandleEvent(int kq, struct kevent* events, int nevents); void Accept(int kq, int connSize); void Receive(int sock, int availBytes); void Enqueue(const char* buf, int bytes); int main(int argc, char* argv[]) { listener_ = CreateListener(); if (listener_ == -1) return -1; int kq = kqueue(); if (!Register(kq, listener_)) { std::cerr << "Register listener to kq failed./n"; return -1; } WaitEvent(kq); return 0; } int CreateListener() { int sock = socket(PF_INET, SOCK_STREAM, 0); if (sock == -1) { std::cerr << "socket() failed:" << errno << std::endl; return -1; } struct sockaddr_in addr; addr.sin_family = AF_INET; addr.sin_port = htons(PORT); addr.sin_addr.s_addr = inet_addr(IP.c_str()); if (bind(sock, (struct sockaddr*)&addr, sizeof(struct sockaddr)) == -1) { std::cerr << "bind() failed:" << errno << std::endl; return -1; } if (listen(sock, 5) == -1) { std::cerr << "listen() failed:" << errno << std::endl; return -1; } return sock; } bool Register(int kq, int fd) { struct kevent changes[1]; EV_SET(&changes[0], fd, EVFILT_READ, EV_ADD, 0, 0, NULL); int ret = kevent(kq, changes, 1, NULL, 0, NULL); if (ret == -1) return false; return true; } void WaitEvent(int kq) { struct kevent events[MAX_EVENT_COUNT]; while (true) { int ret = kevent(kq, NULL, 0, events, MAX_EVENT_COUNT, NULL); if (ret == -1) { std::cerr << "kevent failed!/n"; continue; } HandleEvent(kq, events, ret); } } void HandleEvent(int kq, struct kevent* events, int nevents) { for (int i = 0; i < nevents; i++) { int sock = events[i].ident; int data = events[i].data; if (sock == listener_) Accept(kq, data); else Receive(sock, data); } } void Accept(int kq, int connSize) { for (int i = 0; i < connSize; i++) { int client = accept(listener_, NULL, NULL); if (client == -1) { std::cerr << "Accept failed./n"; continue; } if (!Register(kq, client)) { std::cerr << "Register client failed./n"; return; } } } void Receive(int sock, int availBytes) { int bytes = recv(sock, buf_, availBytes, 0); if (bytes == 0 || bytes == -1) { close(sock); std::cerr << "client close or recv failed./n"; return; } // Write buf to the receive queue. Enqueue(buf_, bytes); } void Enqueue(const char* buf, int bytes) { }
