Linux C socket 封装

/**************************************************************************
*                        Linux C socket 封装
* 声明：
*      同样的代码当然没必要敲很多遍了，一遍就够了，封起来，什么时候要用，
*  什么时候就来这里拿。
*
*                                   2015-7-4 晴 深圳 南山平山村 曾剑锋
*************************************************************************/

\\\\\\\\-*- 目录 -*-////////
|  一、cat zengjf_socket.h
|  二、cat zengjf_socket.c
\\\\\\\\\\\\\\//////////////

一、cat zengjf_socket.h
#ifndef __ZENGJF_SOCKET_H__
#define __ZENGJF_SOCKET_H__

#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 <unistd.h>

#define handle_error(msg) { perror(msg); exit(EXIT_FAILURE); }

/**
* 客户端socket
* 参数说明：
*      1.ip：服务器ip地址，点分十进制的ip字符串
*      2.port：服务器端口号，字串的形式
*      3.srvaddr：连接到服务器端的sockaddr_in
*/
int socketClient(char* ip, char* port, struct sockaddr_in* srvaddr);
/**
* 服务器端socket
* 参数说明：
*      1.port：监听端口号，是字串的形式
*      2.srvaddr：服务器的sockaddr_in
*
*
*      服务器端socket默认能够同时监听30个连接
*/
int socketServer(char* port, struct sockaddr_in* srvaddr);

#endif // __ZENGJF_SOCKET_H__

二、cat zengjf_socket.c
#include "zengjf_socket.h"

int socketClient(char* ip, char* port, struct sockaddr_in* srvaddr){

int sfd;

// create a socket
if ((sfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
handle_error("can not create socketClient.");
}

// set the server socket address
srvaddr->sin_family = AF_INET;
srvaddr->sin_port = htons(atoi(port));
inet_aton(ip, &(srvaddr->sin_addr));

// connect
if (connect(sfd, (struct sockaddr *)srvaddr, sizeof(struct sockaddr)) == -1){
handle_error("can not connect to server.");
}

return sfd;
}

int socketServer(char* port, struct sockaddr_in* srvaddr){

int srv_fd;
//the default listen count
int listenCount = 30;

// create a socket
if ((srv_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
handle_error("can not create socketServer.");
}

//reuse the socket address
int sockopt = 1;
setsockopt(srv_fd, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(sockopt));

// set the socket address
srvaddr->sin_family = AF_INET;
srvaddr->sin_port = htons(atoi(port));
srvaddr->sin_addr.s_addr = htonl(INADDR_ANY);

// bind the socket with the address
if (bind(srv_fd, (struct sockaddr *)srvaddr, sizeof(struct sockaddr)) == -1){
handle_error("can not bind the address.");
}

// listening
// default was 30
if (listen(srv_fd, listenCount) == -1){
handle_error("can not listening to the port.");
}

return srv_fd;
}