usock
便是對網絡套接字鏈接的簡單封裝。網絡
類型標誌:socket
#define USOCK_TCP 0 #define USOCK_UDP 1 #define USOCK_SERVER 0x0100 #define USOCK_NOCLOEXEC 0x0200 #define USOCK_NONBLOCK 0x0400 #define USOCK_NUMERIC 0x0800 #define USOCK_IPV6ONLY 0x2000 #define USOCK_IPV4ONLY 0x4000 #define USOCK_UNIX 0x8000
接口函數:函數
/** * 建立一個sock * * @param type - 類型標誌 * @param host - 做爲server表示綁定本地地址;做爲client表示需鏈接的地址 * @param service - 端口 * @return - sock fd > 0; 錯誤 < 0 */ int usock(int type, const char *host, const char *service);
客戶端代碼:code
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/types.h> #include <sys/socket.h> #include "libubox/usock.h" int main(int argc, char** argv) { struct sockaddr cli_addr; socklen_t len = sizeof(struct sockaddr); int type = USOCK_TCP | USOCK_NOCLOEXEC | USOCK_IPV4ONLY; const char *host = "127.0.0.1"; const char *service = "8000"; char recv_buf[1024] = {0}; // printf("port: %s\n", usock_port(8888)); int c_fd = usock(type, host, service); if(c_fd < 0) { perror("usock"); return -1; } send(c_fd, "helloworld", 10, 0); sleep(2); close(c_fd); return 0; }
服務端代碼:server
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include "libubox/usock.h" int main(int argc, char** argv) { struct sockaddr_in cli_addr; socklen_t len = sizeof(struct sockaddr); int type = USOCK_TCP | USOCK_SERVER | USOCK_NOCLOEXEC | USOCK_IPV4ONLY; const char *host = "127.0.0.1"; const char *service = "8000"; char recv_buf[1024] = {0}; int connect_fd, u_fd; u_fd = usock(type, host, service); if(u_fd < 0) { perror("usock"); return -1; } while (1) { connect_fd = accept(u_fd, (struct sockaddr *)(&cli_addr), &len); if (connect_fd < 0) { perror("accept"); return -1; } printf("client_addr: %s\n", inet_ntoa(cli_addr.sin_addr)); recv(connect_fd, recv_buf, 1024, 0); printf("recv %s\n", recv_buf); close(connect_fd); } return 0; }