acl/lib_fiber/samples-c++/patch.cpp

124 lines
2.6 KiB
C++

#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <assert.h>
#if defined(_WIN32) || defined(_WIN64)
#include <winsock2.h>
#else
#include <unistd.h>
#include <signal.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#endif
#include "fiber/libfiber.h"
#include "patch.h"
void socket_init(void)
{
#if defined(_WIN32) || defined(_WIN64)
WORD version = 0;
WSADATA data;
FillMemory(&data, sizeof(WSADATA), 0);
version = MAKEWORD(2, 0);
if (WSAStartup(version, &data) != 0) {
abort();
}
#else
signal(SIGPIPE, SIG_IGN);
#endif
}
void socket_end(void)
{
#if defined(_WIN32) || defined(_WIN64)
WSACleanup();
#endif
}
void socket_close(SOCKET fd)
{
acl_fiber_close(fd);
}
SOCKET socket_listen(const char *ip, int port)
{
SOCKET fd;
int on;
struct sockaddr_in sa;
memset(&sa, 0, sizeof(sa));
sa.sin_family = AF_INET;
sa.sin_port = htons(port);
sa.sin_addr.s_addr = inet_addr(ip);
fd = socket(AF_INET, SOCK_STREAM, 0);
if (fd == INVALID_SOCKET) {
printf("create socket error, %s\r\n", acl_fiber_last_serror());
getchar();
exit (1);
}
#if defined(_WIN32) || defined(_WIN64)
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (const char *) &on, sizeof(on))) {
#else
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))) {
#endif
printf("setsockopt error %s\r\n", acl_fiber_last_serror());
exit (1);
}
if (bind(fd, (struct sockaddr *) &sa, sizeof(struct sockaddr)) < 0) {
printf("bind error %s\r\n", acl_fiber_last_serror());
getchar();
exit (1);
}
if (acl_fiber_listen(fd, 128) < 0) {
printf("listen error %s\r\n", acl_fiber_last_serror());
getchar();
exit (1);
}
return fd;
}
SOCKET socket_accept(SOCKET fd)
{
SOCKET cfd;
struct sockaddr_in sa;
int len = sizeof(sa);
cfd = acl_fiber_accept(fd, (struct sockaddr *)& sa, (socklen_t *)& len);
return cfd;
}
SOCKET socket_connect(const char *ip, int port)
{
SOCKET fd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in sa;
socklen_t len = (socklen_t) sizeof(sa);
if (fd == INVALID_SOCKET) {
printf("create socket error %s\r\n", acl_fiber_last_serror());
return INVALID_SOCKET;
}
memset(&sa, 0, sizeof(sa));
sa.sin_family = AF_INET;
sa.sin_port = htons(port);
sa.sin_addr.s_addr = inet_addr(ip);
if (acl_fiber_connect(fd, (const struct sockaddr *) &sa, len) < 0) {
acl_fiber_close(fd);
printf("%s: connect %s:%d erorr %s\r\n",
__FUNCTION__, ip, port, acl_fiber_last_serror());
return INVALID_SOCKET;
}
return fd;
}