mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-11-30 02:47:56 +08:00
124 lines
2.6 KiB
C++
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;
|
|
}
|