structc
├── extern
├── LICENSE
├── Makefile
├── README.md
├── structc
└── structc.sln

extern
├── jemalloc
├── jemalloc-vc141-Release-static.lib
├── libuv.lib
├── pthread.h
├── pthread_lib.lib
├── sched.h
├── semaphore.h
├── strings.h
├── uv
└── uv.h

How to build jemalloc for Windows
=================================1. Install Cygwin with at least the following packages:* autoconf* autogen* gawk* grep* sed2. Install Visual Studio 2015 or 2017 with Visual C++3. Add Cygwin\bin to the PATH environment variable4. Open "x64 Native Tools Command Prompt for VS 2017"(note: x86/x64 doesn't matter at this point)5. Generate header files:sh -c "CC=cl ./autogen.sh"6. Now the project can be opened and built in Visual Studio:msvc\jemalloc_vc2017.sln

# 开发环境安装
sudo apt install gcc gdb autogen autoconf# jemalloc 安装
cd
wget https://github.com/jemalloc/jemalloc/releases/download/5.1.0/jemalloc-5.1.0.tar.bz2
tar -jxvf jemalloc-5.1.0.tar.bz2
cd jemalloc-5.1.0sh autogen.sh
make -j4
sudo make install
sudo ldconfigcd
rm -rf jemalloc-5.1.0 jemalloc-5.1.0.tar.bz2

#ifndef _H_ALLOC
#define _H_ALLOC#include <stdlib.h>
#include <string.h>// :) 高效内存分配, 莫名伤感 ~
// _MSC_VER -> Winds CL
// __GNUC__ -> Linux GCC
//
#ifdef _MSC_VER
//
// CPU 检测 x64 or x86
// ISX64 defined 表示 x64 否则 x86
//
#   if defined(_M_ARM64) || defined(_M_X64)
#       define ISX64
#   endif
//
// _M_PPC 为 PowerPC 平台定义, 现在已不支持
// so winds 可以认为都是小端平台
//
#   if defined(_M_PPC)
#       define ISBENIAN
#   endif#elif  __GNUC__#   if defined(__x86_64__)
#       define ISX64
#   endif
//
// 大小端检测 : ISBENIAN defined 表示大端
//
#   if defined(__BIG_ENDIAN__) || defined(__BIG_ENDIAN_BITFIELD)
#       define ISBENIAN
#   endif#else
#   error BUILD (￣︶￣) S
#endif// OFF_ALLOC - 关闭全局 free / malloc 配置
#ifndef OFF_ALLOC#   undef  free
#   define free    free_#   undef  strdup
#   define strdup  strdup_#   undef  malloc
#   define malloc  malloc_
#   undef  calloc
#   define calloc  calloc_
#   undef  realloc
#   define realloc realloc_#endif//OFF_ALLOC//
// free_ - free 包装函数
// ptr      : 内存首地址
// return   : void
//
extern void free_(void * ptr);//
// malloc_ - malloc 包装, 封装一些特殊业务
// size     : 分配的内存字节
// return   : 返回可使用的内存地址.
//
extern void * malloc_(size_t size);//
// strdup_ - strdup 包装函数
// s        : '\0' 结尾 C 字符串
// return   : 拷贝后新的 C 字符串
//
extern char * strdup_(const char * s);//
// calloc_ - calloc 包装, 封装一些特殊业务
// num      : 数量
// size     : 大小
// return   : 返回可用内存地址, 并且置0
//
extern void * calloc_(size_t num, size_t size);//
// realloc_ - realoc 包装函数, 封装一些特殊业务
// ptr      : 内存首地址, NULL 等同于 malloc
// size     : 重新分配的内存大小
// return   : 返回重新分配好的新地址内容
//
extern void * realloc_(void * ptr, size_t size);#endif//_H_STDEXIT

#include <stdio.h>#define OFF_ALLOC
#include "alloc.h"#define JEMALLOC_NO_DEMANGLE
#include <jemalloc/jemalloc.h>//
// free_ - free 包装函数
// ptr      : 内存首地址
// return   : void
//
inline void free_(void * ptr) {je_free(ptr);
}// 简单内存不足检测处理
static inline void * mcheck(void * ptr, size_t size) {if (NULL == ptr) {fprintf(stderr, "out of memory trying to allocate %zu\n", size);fflush(stderr);abort();}return ptr;
}//
// malloc_ - malloc 包装, 封装一些特殊业务
// size     : 分配的内存字节
// return   : 返回可使用的内存地址.
//
inline void * malloc_(size_t size) {void * ptr = je_malloc(size);return mcheck(ptr, size);
}//
// strdup_ - strdup 包装函数
// s        : '\0' 结尾 C 字符串
// return   : 拷贝后新的 C 字符串
//
inline char * strdup_(const char * s) {if (s) {size_t n = strlen(s) + 1;char * ptr = malloc_(n);return memcpy(ptr, s, n);}return NULL;
}//
// calloc_ - calloc 包装, 封装一些特殊业务
// num      : 数量
// size     : 大小
// return   : 返回可用内存地址, 并且置0
//
inline void * calloc_(size_t num, size_t size) {void * ptr = je_calloc(num, size);return mcheck(ptr, size);
}//
// realloc_ - realoc 包装函数, 封装一些特殊业务
// ptr      : 内存首地址, NULL 等同于 malloc
// size     : 重新分配的内存大小
// return   : 返回重新分配好的新地址内容
//
inline void * realloc_(void * ptr, size_t size) {void * ntr = je_realloc(ptr, size);return mcheck(ntr, size);
}

# libuv 安装
cd
wget https://github.com/libuv/libuv/archive/v1.23.1.zip
unzip v1.23.1.zip
cd libuv-1.23.1sh autogen.sh
./configure
make -j4
sudo make install
sudo ldconfigcd
#
# 注意 uv 头文件, 全部导入到系统 include 目录下面
#
rm -rf libuv-1.23.1 v1.23.1.zip

#ifndef _H_THREAD
#define _H_THREAD#include "struct.h"
#include <pthread.h>
#include <semaphore.h>//
// pthread_end - 等待启动线程结束
// tid      : 线程id
// return   : void
//
inline void pthread_end(pthread_t tid) {pthread_join(tid, NULL);
}//
// pthread_run - 异步启动线程
// id       : &tid 线程id地址
// frun     : 运行的主体
// arg      : 运行参数
// return   : 返回线程构建结果, 0 is success
//
#define pthread_run(id, frun, arg)                                  \
pthread_run_(&(id), (node_f)(frun), (void *)(intptr_t)(arg))
inline int pthread_run_(pthread_t * id, node_f frun, void * arg) {return pthread_create(id, NULL, (start_f)frun, arg);
}//
// pthread_async - 异步启动分离线程
// frun     : 运行的主体
// arg      : 运行参数
// return   : 返回 0 is success
//
#define pthread_async(frun, arg)                                    \
pthread_async_((node_f)(frun), (void *)(intptr_t)(arg))
inline int pthread_async_(node_f frun, void * arg) {int ret;pthread_t tid;pthread_attr_t attr;pthread_attr_init(&attr);pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);ret = pthread_create(&tid, &attr, (start_f)frun, arg);pthread_attr_destroy(&attr);return ret;
}#endif//_H_THREAD

#ifndef _H_STRUCT
#define _H_STRUCT#include <math.h>
#include "alloc.h"
#include <ctype.h>
#include <float.h>
#include <stdio.h>
#include <errno.h>
#include <assert.h>
#include <stdarg.h>
#include <stdint.h>
#include <stddef.h>
#include <limits.h>
#include <stdbool.h>
#include <inttypes.h>//
// enum Flag int - 函数返回值全局状态码
// >= 0 标识 Success 状态, < 0 标识 Error 状态
//
enum {SBase       =  +0, // 正确基础类型
EBase       =  -1, // 错误基础类型EParam      =  -2, // 输入参数错误EFd         =  -3, // 文件打开失败EClose      =  -4, // 文件操作关闭EAccess     =  -5, // 没有操作权限EAlloc      =  -6, // 内存操作错误EParse      =  -7, // 协议解析错误ESmall      =  -8, // 过小基础错误EBig        =  -9, // 过大基础错误ETimeout    = -10, // 操作超时错误
};//
// DCODE - DEBUG 模式下的测试宏
// DCODE({
//     puts("debug start...");
// });
//
#ifndef DCODE
#   ifdef _DEBUG
#       define DCODE(code)  do code while(0)
#   else
#       define DCODE(code)
#   endif //  ! _DEBUG
#endif  //  ! DCODE//
// icmp_f - 比较行为的类型
//  : int add_cmp(const void * now, const void * node)
//
typedef int (* icmp_f)();//
// vnew_f - 根据规则构建对象
//  : void * rtree_new(void * node)
//
typedef void * (* vnew_f)();//
// node_f - 销毁当前对象节点
//  : void list_die(void * node);
//
typedef void (* node_f)(void * node);//
// start_f - pthread create func
//  : int * run(int * arg)
//
typedef void * (* start_f)(void * arg);//
// each_f - each 循环操作, arg 外部参数, node 是内部结点
//  : int dict_echo(struct dict * node, void * arg) { return 0; }
//
typedef int (* each_f)(void * node, void * arg);//
// CERR - 打印错误信息
// EXIT - 打印错误信息, 并 exit
// IF   - 条件判断异常退出的辅助宏
//
#define CERR(fmt, ...)                                                   \
fprintf(stderr, "[%s:%s:%d][%d:%s]" fmt "\n",                            \__FILE__, __func__, __LINE__, errno, strerror(errno), ##__VA_ARGS__)#define EXIT(fmt, ...)                                                   \
do {                                                                     \CERR(fmt, ##__VA_ARGS__);                                            \exit(EXIT_FAILURE);                                                  \
} while(0)#define IF(cond)                                                         \
if ((cond)) EXIT(#cond)//
// RETURN - 打印错误信息, 并 return 返回指定结果
// val      : return的东西, 当需要 return void; 时候填 ',' 就过 or NIL
// fmt      : 双引号包裹的格式化字符串
// ...      : fmt中对应的参数
// return   : val
//
#define RETURN(val, fmt, ...)                                           \
do {                                                                    \CERR(fmt, ##__VA_ARGS__);                                           \return val;                                                         \
} while(0)#define NIL
#define RETNIL(fmt, ...)                                                \
RETURN(NIL , fmt, ##__VA_ARGS__)#define RETNUL(fmt, ...)                                                \
RETURN(NULL, fmt, ##__VA_ARGS__)#endif//_H_STRUCT

# 编译的目录结构
# Release : make
# Debug   : make D=-D_DEBUG
# Clean   : make clean

{"configurations": [{"name": "Linux","includePath": ["/usr/include/c++/7","/usr/include/x86_64-linux-gnu/c++/7","/usr/include/c++/7/backward","/usr/lib/gcc/x86_64-linux-gnu/7/include","/usr/local/include","/usr/lib/gcc/x86_64-linux-gnu/7/include-fixed","/usr/include/x86_64-linux-gnu","/usr/include","${workspaceRoot}","${workspaceRoot}/structc/base","${workspaceRoot}/structc/struct","${workspaceRoot}/structc/system"],"defines": ["_DEBUG","__GNUC__"],"intelliSenseMode": "clang-x64","browse": {"path": ["/usr/include/c++/7","/usr/include/x86_64-linux-gnu/c++/7","/usr/include/c++/7/backward","/usr/lib/gcc/x86_64-linux-gnu/7/include","/usr/local/include","/usr/lib/gcc/x86_64-linux-gnu/7/include-fixed","/usr/include/x86_64-linux-gnu","/usr/include","${workspaceRoot}"],"limitSymbolsToIncludedHeaders": true,"databaseFilename": ""},"compilerPath": "/usr/bin/clang","cStandard": "c11","cppStandard": "c++17"}],"version": 4
}

{// 使用 IntelliSense 了解相关属性。 // 悬停以查看现有属性的描述。// 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387"version": "0.2.0","configurations": [{"name": "(gdb) Launch","type": "cppdbg","request": "launch","program": "${workspaceFolder}/Out/main.exe","args": [],"stopAtEntry": false,"cwd": "${workspaceFolder}","environment": [],"externalConsole": true,"preLaunchTask": "Debug","MIMode": "gdb","setupCommands": [{"description": "Enable pretty-printing for gdb","text": "-enable-pretty-printing","ignoreFailures": true}]}]
}

4. F5 -> tasks.json

{// See https://go.microsoft.com/fwlink/?LinkId=733558// for the documentation about the tasks.json format"version": "2.0.0","tasks": [{"type"    : "shell","label"   : "Debug","command" : "make D=-D_DEBUG"}]
}

wzhi@wzc:~/structc/structc$ tree -L 1
.
├── base
├── conf
├── main
├── README.md
├── struct
├── structc.vcxproj
├── structc.vcxproj.filters
├── structc.vcxproj.user
├── system
└── test

wzhi@wzc:~/structc/structc/main$ tree
.
├── main.c
├── main_init.c
├── main_run.c
└── main_test.c

#include "head.h"//
// main - 程序的总入口, 从扯开始
// argc     : 输入参数个数
// argv     : 参数集
// return   : 返回程序退出的状态码
//
int main(int argc, char * argv[]) {//// 初始化 ... ...// ... ...
    EXTERN_RUN(main_init);//// make D=-D_DEBUG// main_test 单元测试才会启动//
#ifdef _DEBUGEXTERN_RUN(main_test);
#endif// ... // ... 启动当前项目运行的主函数//
    EXTERN_RUN(main_run, argc, argv);return EXIT_SUCCESS;
}

//
// EXTERN_RUN - 简单的声明, 并立即使用的宏
// ftest    : 需要执行的函数名称
// ...      : 可变参数, 保留
//
#define EXTERN_RUN(ftest, ...)                                  \
do {                                                            \extern void ftest();                                        \ftest (__VA_ARGS__);                                        \
} while(0)

#include "head.h"//
// TEST - 用于单元测试函数, 执行并输出运行时间
// ftest    : 需要执行的测试函数名称
// ...      : 可变参数, 保留
//
#define TEST(ftest, ...)                                         \
do {                                                             \extern void ftest();                                         \clock_t $s = clock();                                        \ftest (##__VA_ARGS__);                                       \double $e = (double)clock();                                 \printf(STR(ftest)" run time:%lfs\n", ($e-$s)/CLOCKS_PER_SEC);\
} while(0)//
// main_test - *_test is here run
// return   : void
//
void main_test(void) {//// 开始你的表演, 单元测试//
EXTERN_RUN(uv_tty_test);
}

#include <uv.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>//
// 测试 libuv tty 操作控制台
// 输出一段有颜色的文字
//
void uv_tty_test(void) {uv_tty_t tty;uv_buf_t buf[3];unsigned i, len = sizeof buf / sizeof *buf;uv_loop_t * loop = uv_default_loop();// 目前只对 tty 控制台处理if (uv_guess_handle(1) != UV_TTY) {fprintf(stderr, "uv_guess_handle(1) != UV_TTY!\n");exit(EXIT_FAILURE);}uv_tty_init(loop, &tty, 1, 0);uv_tty_set_mode(&tty, UV_TTY_MODE_NORMAL);// 开始发送消息buf[0].base = "\033[46;37m";buf[1].base = u8"(✿◡‿◡) 喵酱 ((●'-'●)) 比 ♥ 里~ \n";buf[2].base = "\033[0m";for (i = 0; i < len; ++i)buf[i].len = (int)strlen(buf[i].base);uv_try_write((uv_stream_t *)&tty, buf, len);// 重置终端行为
    uv_tty_reset_mode();uv_run(loop, UV_RUN_DEFAULT);
}

#ifndef _H_SOCKET
#define _H_SOCKET#include <time.h>
#include <fcntl.h>
#include "struct.h"
#include <signal.h>
#include <sys/types.h>#ifdef __GNUC__#include <netdb.h>
#include <unistd.h>
#include <sys/un.h>
#include <sys/uio.h>
#include <sys/time.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/tcp.h>
#include <sys/resource.h>//
// This is used instead of -1, since the. by WinSock
// On now linux EAGAIN and EWOULDBLOCK may be the same value
// connect 链接中, linux 是 EINPROGRESS，winds 是 WSAEWOULDBLOCK
//
typedef int socket_t;#define INVALID_SOCKET          (~0)
#define SOCKET_ERROR            (-1)// socket_init - 初始化 socket 库初始化方法
inline void socket_init(void) {// 管道破裂, 忽略 SIGPIPE 信号
    signal(SIGPIPE, SIG_IGN);
}inline int socket_close(socket_t s) {return close(s);
}// socket_set_block     - 设置套接字是阻塞
// socket_set_nonblock  - 设置套接字是非阻塞
inline int socket_set_block(socket_t s) {int mode = fcntl(s, F_GETFL, 0);return fcntl(s, F_SETFL, mode & ~O_NONBLOCK);
}inline int socket_set_nonblock(socket_t s) {int mode = fcntl(s, F_GETFL, 0);return fcntl(s, F_SETFL, mode | O_NONBLOCK);
}// socket_recv      - 读取数据
// socket_send      - 写入数据
inline int socket_recv(socket_t s, void * buf, int sz) {return (int)read(s, buf, sz);
}inline int socket_send(socket_t s, const void * buf, int sz) {return (int)write(s, buf, sz);
}#endif#ifdef _MSC_VER#include <ws2tcpip.h>#undef  errno
#define errno                   WSAGetLastError()
#undef  strerror
#define strerror                ((char * (*)(int))strerr)#undef  EINTR
#define EINTR                   WSAEINTR
#undef  EAGAIN
#define EAGAIN                  WSAEWOULDBLOCK
#undef  EINPROGRESS
#define EINPROGRESS             WSAEWOULDBLOCK/** WinSock 2 extension -- manifest constants for shutdown()*/
#define SHUT_RD                 SD_RECEIVE
#define SHUT_WR                 SD_SEND
#define SHUT_RDWR               SD_BOTH#define SO_REUSEPORT            SO_REUSEADDRtypedef SOCKET socket_t;
typedef int socklen_t;//
// gettimeofday - Linux sys/time.h 中得到微秒时间实现
// tv       : 返回结果包含秒数和微秒数
// tz       : 包含的时区, winds 上这个变量没有作用
// return   : 默认返回 0
//
extern int gettimeofday(struct timeval * tv, void * tz);//
// strerr - linux 上替代 strerror, winds 替代 FormatMessage
// error    : linux 是 errno, winds 可以是 WSAGetLastError() ...
// return   : system os 拔下来的提示常量字符串
//
extern const char * strerr(int err);// socket_init - 初始化 socket 库初始化方法
inline void socket_init(void) {WSADATA wsad;WSAStartup(WINSOCK_VERSION, &wsad);
}// socket_close     - 关闭上面创建后的句柄
inline int socket_close(socket_t s) {return closesocket(s);
}// socket_set_block     - 设置套接字是阻塞
// socket_set_nonblock  - 设置套接字是非阻塞
inline int socket_set_block(socket_t s) {u_long mode = 0;return ioctlsocket(s, FIONBIO, &mode);
}inline int socket_set_nonblock(socket_t s) {u_long mode = 1;return ioctlsocket(s, FIONBIO, &mode);
}// socket_recv      - 读取数据
// socket_send      - 写入数据
inline int socket_recv(socket_t s, void * buf, int sz) {return sz > 0 ? recv(s, buf, sz, 0) : 0;
}inline int socket_send(socket_t s, const void * buf, int sz) {return send(s, buf, sz, 0);
}#endif//
// 通用 sockaddr_in ipv4 地址
//
typedef struct sockaddr_in sockaddr_t[1];// socket_dgram     - 创建 UDP socket
// socket_stream    - 创建 TCP socket
inline socket_t socket_dgram(void) {return socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
}inline socket_t socket_stream(void) {return socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
}// socket_set_reuse - 开启端口和地址复用
// socket_set_keepalive - 开启心跳包检测, 默认2h 5次
inline int socket_set_enable(socket_t s, int optname) {int ov = 1;return setsockopt(s, SOL_SOCKET, optname, (void *)&ov, sizeof ov);
}inline int socket_set_reuse(socket_t s) {return socket_set_enable(s, SO_REUSEPORT);
}inline int socket_set_keepalive(socket_t s) {return socket_set_enable(s, SO_KEEPALIVE);
}// socket_set_rcvtimeo - 设置接收数据毫秒超时时间
// socket_set_sndtimeo - 设置发送数据毫秒超时时间
inline int socket_set_time(socket_t s, int ms, int optname) {struct timeval ov = { 0,0 };if (ms > 0) {ov.tv_sec = ms / 1000;ov.tv_usec = (ms % 1000) * 1000;}return setsockopt(s, SOL_SOCKET, optname, (void *)&ov, sizeof ov);
}inline int socket_set_rcvtimeo(socket_t s, int ms) {return socket_set_time(s, ms, SO_RCVTIMEO);
}inline int socket_set_sndtimeo(socket_t s, int ms) {return socket_set_time(s, ms, SO_SNDTIMEO);
}// socket_get_error - 得到当前socket error 值, 0 表示正确, 其它都是错误
inline int socket_get_error(socket_t s) {int err;socklen_t len = sizeof(err);int r = getsockopt(s, SOL_SOCKET, SO_ERROR, (void *)&err, &len);return r < 0 ? errno : err;
}// socket_recvfrom  - recvfrom 接受函数
// socket_sendto    - sendto 发送函数
inline int socket_recvfrom(socket_t s, void * buf, int len, int flags, sockaddr_t in) {socklen_t inlen = sizeof (sockaddr_t);return recvfrom(s, buf, len, flags, (struct sockaddr *)in, &inlen);
}inline int socket_sendto(socket_t s, const void * buf, int len, int flags, const sockaddr_t to) {return sendto(s, buf, len, flags, (const struct sockaddr *)to, sizeof(sockaddr_t));
}//
// socket_recvn     - socket 接受 sz 个字节
// socket_sendn     - socket 发送 sz 个字节
//
extern int socket_recvn(socket_t s, void * buf, int sz);
extern int socket_sendn(socket_t s, const void * buf, int sz);// socket_bind          - bind    绑定函数
// socket_listen        - listen  监听函数
// socket_accept        - accept  等接函数
// socket_connect       - connect 链接函数
inline int socket_bind(socket_t s, const sockaddr_t addr) {return bind(s, (const struct sockaddr *)addr, sizeof(sockaddr_t));
}inline int socket_listen(socket_t s) {return listen(s, SOMAXCONN);
}inline socket_t socket_accept(socket_t s, sockaddr_t addr) {socklen_t len = sizeof (sockaddr_t);return accept(s, (struct sockaddr *)addr, &len);
}inline int socket_connect(socket_t s, const sockaddr_t addr) {return connect(s, (const struct sockaddr *)addr, sizeof(sockaddr_t));
}//
// socket_binds     - 端口绑定返回绑定好的 socket fd, 返回 INVALID_SOCKET or PF_INET PF_INET6
// socket_listens   - 端口监听返回监听好的 socket fd.
//
extern socket_t socket_binds(const char * ip, uint16_t port, uint8_t protocol, int * family);
extern socket_t socket_listens(const char * ip, uint16_t port, int backlog);//
// socket_addr -socket_recv 通过 ip, port 构造 ipv4 结构
//
extern int socket_addr(const char * ip, uint16_t port, sockaddr_t addr);// socket_pton - 返回 ip 串
inline char * socket_pton(sockaddr_t addr, char ip[INET_ADDRSTRLEN]) {return (char *)inet_ntop(AF_INET, &addr->sin_addr, ip, INET_ADDRSTRLEN);
}//
// socket_host - 通过 ip:port 串得到 socket addr 结构
// host     : ip:port 串
// addr     : 返回最终生成的地址
// return   : >= EBase 表示成功
//
extern int socket_host(const char * host, sockaddr_t addr);//
// socket_tcp - 创建 TCP 详细套接字
// host     : ip:port 串
// return   : 返回监听后套接字
//
extern socket_t socket_tcp(const char * host);//
// socket_udp - 创建 UDP 详细套接字
// host     : ip:port 串
// return   : 返回绑定后套接字
//
extern socket_t socket_udp(const char * host);//
// socket_connects - 返回链接后的阻塞套接字
// host     : ip:port 串
// return   : 返回链接后阻塞套接字
//
extern socket_t socket_connects(const char * host);//
// socket_connectos - 返回链接后的非阻塞套接字
// host     : ip:port 串
// ms       : 链接过程中毫秒数
// return   : 返回链接后非阻塞套接字
//
extern socket_t socket_connectos(const char * host, int ms);#endif//_H_SOCKET

#include "socket.h"#ifdef _MSC_VER//
// gettimeofday - Linux sys/time.h 中得到微秒时间实现
// tv       : 返回结果包含秒数和微秒数
// tz       : 包含的时区, winds 上这个变量没有作用
// return   : 默认返回 0
//
int
gettimeofday(struct timeval * tv, void * tz) {struct tm m;SYSTEMTIME se;GetLocalTime(&se);m.tm_year = se.wYear - 1900;m.tm_mon = se.wMonth - 1;m.tm_mday = se.wDay;m.tm_hour = se.wHour;m.tm_min = se.wMinute;m.tm_sec = se.wSecond;m.tm_isdst = -1; // 不考虑夏令时
tv->tv_sec = (long)mktime(&m);tv->tv_usec = se.wMilliseconds * 1000;return 0;
}#endif//
// socket_recvn     - socket 接受 sz 个字节
// socket_sendn     - socket 发送 sz 个字节
//

int
socket_recvn(socket_t s, void * buf, int sz) {int r, n = sz;while (n > 0) {r = recv(s, buf, n, 0);if (r == 0) break;if (r == SOCKET_ERROR) {if (errno == EINTR)continue;return SOCKET_ERROR;}n -= r;buf = (char *)buf + r;}return sz - n;
}int
socket_sendn(socket_t s, const void * buf, int sz) {int r, n = sz;while (n > 0) {r = send(s, buf, n, 0);if (r == 0) break;if (r == SOCKET_ERROR) {if (errno == EINTR)continue;return SOCKET_ERROR;}n -= r;buf = (char *)buf + r;}return sz - n;
}//
// socket_addr - 通过 ip, port 构造 ipv4 结构
//
int
socket_addr(const char * ip, uint16_t port, sockaddr_t addr) {addr->sin_family = AF_INET;addr->sin_port = htons(port);addr->sin_addr.s_addr = inet_addr(ip);if (addr->sin_addr.s_addr == INADDR_NONE) {struct hostent * host = gethostbyname(ip);if (!host || !host->h_addr)return EParam;// 尝试一种, 默认 ipv4memcpy(&addr->sin_addr, host->h_addr, host->h_length);}memset(addr->sin_zero, 0, sizeof addr->sin_zero);return SBase;
}//
// socket_binds     - 端口绑定返回绑定好的 socket fd, 返回 INVALID_SOCKET or PF_INET PF_INET6
// socket_listens   - 端口监听返回监听好的 socket fd.
//
socket_t
socket_binds(const char * ip, uint16_t port, uint8_t protocol, int * family) {socket_t fd;char ports[sizeof "65535"];struct addrinfo * addr = NULL, hint = { 0 };if (NULL == ip || *ip == '\0')ip = "0.0.0.0"; // default INADDR_ANY
sprintf(ports, "%hu", port);hint.ai_family = AF_UNSPEC;if (protocol == IPPROTO_TCP)hint.ai_socktype = SOCK_STREAM;else {assert(protocol == IPPROTO_UDP);hint.ai_socktype = SOCK_DGRAM;}hint.ai_protocol = protocol;if (getaddrinfo(ip, ports, &hint, &addr))return INVALID_SOCKET;fd = socket(addr->ai_family, addr->ai_socktype, 0);if (fd == INVALID_SOCKET)goto err_free;if (socket_set_reuse(fd))goto err_close;if (bind(fd, addr->ai_addr, (int)addr->ai_addrlen))goto err_close;// Success return ip familyif (family)*family = addr->ai_family;freeaddrinfo(addr);return fd;err_close:socket_close(fd);
err_free:freeaddrinfo(addr);return INVALID_SOCKET;
}socket_t
socket_listens(const char * ip, uint16_t port, int backlog) {socket_t fd = socket_binds(ip, port, IPPROTO_TCP, NULL);if (INVALID_SOCKET != fd && listen(fd, backlog)) {socket_close(fd);return INVALID_SOCKET;}return fd;
}// host_parse - 解析 host 内容
static int host_parse(const char * host, char ip[BUFSIZ], uint16_t * pprt) {int port = 0;char * st = ip;if (!host || !*host || *host == ':')strcpy(ip, "0.0.0.0");else {char c;// 简单检查字符串是否合法size_t n = strlen(host);if (n >= BUFSIZ)RETURN(EParam, "host err %s", host);// 寻找分号while ((c = *host++) != ':' && c)*ip++ = c;*ip = '\0';if (c == ':') {if (n > ip - st + sizeof "65535")RETURN(EParam, "host port err %s", host);port = atoi(host);// 有些常识数字, 不一定是魔法 ... :)if (port <= 1024 || port > 65535)RETURN(EParam, "host port err %s, %d", host, port);}}*pprt = port;return SBase;
}//
// socket_host - 通过 ip:port 串得到 socket addr 结构
// host     : ip:port 串
// addr     : 返回最终生成的地址
// return   : >= EBase 表示成功
//
int
socket_host(const char * host, sockaddr_t addr) {uint16_t port; char ip[BUFSIZ];if (host_parse(host, ip, &port) < SBase)return EParam;// 开始构造 addrif (NULL == addr) {sockaddr_t nddr;return socket_addr(ip, port, nddr);}return socket_addr(ip, port, addr);
}//
// socket_tcp - 创建 TCP 详细套接字
// host     : ip:port 串
// return   : 返回监听后套接字
//
socket_t
socket_tcp(const char * host) {uint16_t port; char ip[BUFSIZ];if (host_parse(host, ip, &port) < SBase)return EParam;return socket_listens(ip, port, SOMAXCONN);
}//
// socket_udp - 创建 UDP 详细套接字
// host     : ip:port 串
// return   : 返回绑定后套接字
//
socket_t
socket_udp(const char * host) {uint16_t port; char ip[BUFSIZ];if (host_parse(host, ip, &port) < SBase)return EParam;return socket_binds(ip, port, IPPROTO_UDP, NULL);
}//
// socket_connects - 返回链接后的阻塞套接字
// host     : ip:port 串
// return   : 返回链接后阻塞套接字
//
socket_t
socket_connects(const char * host) {sockaddr_t addr;socket_t s = socket_stream();if (INVALID_SOCKET == s) {RETURN(s, "socket_stream is error");}// 解析配置成功后尝试链接if (socket_host(host, addr) >= SBase)if (socket_connect(s, addr) >= SBase)return s;socket_close(s);RETURN(INVALID_SOCKET, "socket_connects %s", host);
}//
// socket_connecto      - connect 超时链接, 返回非阻塞 socket
//
static int socket_connecto(socket_t s, const sockaddr_t addr, int ms) {int n, r;struct timeval to;fd_set rset, wset, eset;// 还是阻塞的connectif (ms < 0) return socket_connect(s, addr);// 非阻塞登录, 先设置非阻塞模式r = socket_set_nonblock(s);if (r < SBase) return r;// 尝试连接, connect 返回 -1 并且 errno == EINPROGRESS 表示正在建立链接r = socket_connect(s, addr);// connect 链接中, linux 是 EINPROGRESS，winds 是 WSAEWOULDBLOCKif (r >= SBase || errno != EINPROGRESS) {socket_set_block(s);return r;}// 超时 timeout, 直接返回结果 EBase = -1 错误if (ms == 0) {socket_set_block(s);return EBase;}FD_ZERO(&rset); FD_SET(s, &rset);FD_ZERO(&wset); FD_SET(s, &wset);FD_ZERO(&eset); FD_SET(s, &eset);to.tv_sec = ms / 1000;to.tv_usec = (ms % 1000) * 1000;n = select((int)s + 1, &rset, &wset, &eset, &to);// 超时直接滚if (n <= 0) {socket_set_block(s);return EBase;}// 当连接成功时候,描述符会变成可写if (n == 1 && FD_ISSET(s, &wset)){socket_set_block(s);return SBase;}// 当连接建立遇到错误时候, 描述符变为即可读又可写if (FD_ISSET(s, &eset) || n == 2) {socklen_t len = sizeof n;// 只要最后没有 error 那就 链接成功if (!getsockopt(s, SOL_SOCKET, SO_ERROR, (char *)&n, &len) && !n)r = SBase;}socket_set_block(s);return r;
}//
// socket_connectos - 返回链接后的非阻塞套接字
// host     : ip:port 串
// ms       : 链接过程中毫秒数
// return   : 返回链接后非阻塞套接字
//
socket_t
socket_connectos(const char * host, int ms) {sockaddr_t addr;socket_t s = socket_stream();if (INVALID_SOCKET == s) {RETURN(s, "socket_stream is error");}// 解析配置成功后尝试链接if (socket_host(host, addr) >= SBase)if (socket_connecto(s, addr, ms) >= SBase)return s;socket_close(s);RETURN(INVALID_SOCKET, "socket_connectos %s", host);
}

#ifndef _H_ATOM
#define _H_ATOM#include "atomic.h"//
// atom_t 自旋锁类型
// [static] atom_t o = 0;
//   atom_lock(o);
//  - One Man RPG
// atom_unlock(o);
//
typedef volatile long atom_t;// atom_acquire - 维护优化后读写代码不在其前
#define atom_acquire()      atomic_fence(ATOMIC_ACQUIRE)
// atom_release - 维护优化后读写代码不在其后
#define atom_release()      atomic_fence(ATOMIC_RELEASE)
// atom_seq_cst - 维护优化后读写代码前后不动
#define atom_seq_cst()      atomic_fence(ATOMIC_SEQ_CST)#ifdef __GNUC__#define atom_trylock(o)     (!__sync_lock_test_and_set(&(o), 1))#define atom_lock(o)        while(__sync_lock_test_and_set(&(o), 1))#define atom_unlock(o)      __sync_lock_release(&(o))// 内存屏障, 维持代码顺序
#define atom_sync()         __sync_synchronize()// v += a ; return v;
#define atom_add(v, a)      __sync_add_and_fetch(&(v), (a))
// type tmp = v ; v = a; return tmp;
#define atom_set(v, a)      __sync_lock_test_and_set(&(v), (a))
// v &= a; return v;
#define atom_and(v, a)      __sync_and_and_fetch(&(v), (a))
// return ++v;
#define atom_inc(v)         __sync_add_and_fetch(&(v), 1)
// return --v;
#define atom_dec(v)         __sync_sub_and_fetch(&(v), 1)
// bool b = v == c; b ? v=a : ; return b;
#define atom_cas(v, c, a)   __sync_bool_compare_and_swap(&(v), (c), (a))#endif#ifdef _MSC_VER#include <intrin.h>
#include <intrin0.h>/* Interlocked intrinsic mapping for _nf/_acq/_rel */
#if defined(_M_ARM) || defined(_M_ARM64)
#define _ACQUIRE(x) ATOMIC_CONCAT(x, _acq)
#else /* defined(_M_ARM) || defined(_M_ARM64) */
#define _ACQUIRE(x) x
#endif /* defined(_M_ARM) || defined(_M_ARM64) */#define atom_trylock(o)     (!_ACQUIRE(_interlockedbittestandset)(&(o), 0))#define atom_lock(o)        while(_ACQUIRE(_interlockedbittestandset)(&(o), 0))inline void store_release(atom_t * x) {/* store _Value atomically with release memory order */
#if defined(_M_ARM) || defined(_M_ARM64)__dmb(0xB /* _ARM_BARRIER_ISH or _ARM64_BARRIER_ISH*/);__iso_volatile_store32((volatile int *)x, 0);
#else_ReadWriteBarrier();*x = 0;
#endif
}#define atom_unlock(o)      store_release(&(o))// 保证代码优化后不乱序执行
#define atom_sync()         MemoryBarrier()// v 和 a 都是 long 这样数据
#define atom_add(v, a)      InterlockedAdd((volatile LONG *)&(v), (LONG)(a))
#define atom_set(v, a)      InterlockedExchange((volatile LONG *)&(v), (LONG)(a))
#define atom_and(v, a)      InterlockedAnd((volatile LONG *)&(v), (LONG)(a))
#define atom_inc(v)         InterlockedIncrement((volatile LONG *)&(v))
#define atom_dec(v)         InterlockedDecrement((volatile LONG *)&(v))
//
// 对于 InterlockedCompareExchange(v, c, a) 等价于下面
// long tmp = v ; v == a ? v = c : ; return tmp;
//
// 咱们的 atom_cas(v, c, a) 等价于下面
// long tmp = v ; v == c ? v = a : ; return tmp;
//
#define atom_cas(v, c, a)   ((LONG)(c) == InterlockedCompareExchange((volatile LONG *)&(v), (LONG)(a), (LONG)(c)))#endif#endif//_H_ATOM