socketprogrammingexample

#include
#include
#include
#include
#include
#include
#include
#include
#include "server.h"int catch_signal(int sig, void (*handler) (int)) {struct sigaction action;action.sa_handler &＃61; handler;sigemptyset(&action.sa_mask);// sigemptyset() return 0 on success and -1 on error.action.sa_flags &＃61; 0;return sigaction(sig, &action, NULL);// sigaction() returns 0 on success and -1 on error.
}// recv()注意:
// 1. The characters are not terminated with a \0 character.
// 2. When someone types text in telnet, the string always ends \r\n.
// 3. The recv() will return the number of characters, or –1 if there&＃39;s an error, or
// 0 if the client has closed the connection.
// 4. You&＃39;re not guaranteed to receive all the characters in a single call to recv().// This reads all the characters until it reaches &＃39;\n&＃39;.
int read_in(int socket, char *buf, int len) {char *s &＃61; buf;int slen &＃61; len;int c &＃61; recv(socket, s, slen, 0);// Keep reading until there are no more characters or you reach &＃39;\n&＃39;.while ( (c > 0) && (s[c-1] !&＃61; &＃39;\n&＃39;) ) {s &＃43;&＃61; c; slen -&＃61; c;c &＃61; recv(socket, s, slen, 0);}if (c <0)// In case there&＃39;s an errorreturn c;else if (c &＃61;&＃61; 0)// Nothing read; send back an empty stringbuf[0] &＃61; &＃39;\0&＃39;;else// Replace the last character(here is &＃39;\n&＃39;) with a &＃39;\0&＃39;.s[c-1] &＃61; &＃39;\0&＃39;;return len-slen;
}void error(char * msg, int exit_val) {fprintf(stderr, "%s: %s\n", msg, strerror(errno));// if exit_val &＃61;&＃61; 0, not exit the program.if (exit_val)exit(exit_val);
}int open_listener_socket() {// Create an Internet streaming socket.int s &＃61; socket(PF_INET, SOCK_STREAM, 0);if (s &＃61;&＃61; -1)error("Can&＃39;t open socket", 1);return s;
}void bind_to_port(int socket, int port) {struct sockaddr_in name;name.sin_family &＃61; PF_INET;name.sin_port &＃61; (in_port_t) htons(port);name.sin_addr.s_addr &＃61; htonl(INADDR_ANY);// Yes, reuse the socket (so you can restart the server without problems).int reuse &＃61; 1;if (setsockopt(socket, SOL_SOCKET, SO_REUSEADDR, (char *) &reuse, sizeof(int)) &＃61;&＃61; -1)error("Can&＃39;t set the reuse option on the socket", 1);int c &＃61; bind(socket, (struct sockaddr *) &name, sizeof(name));if(c &＃61;&＃61; -1)error("Can&＃39;t bind to socket", 1);
}void listen_to_socket(int socket, int queue) {int l &＃61; listen(socket, queue);if(l &＃61;&＃61; -1)error("Can&＃39;t listen", 1);
}// wait client connect to this server. return the secondary socketfd.
int accept_socket(int sockfd, struct sockaddr *addr, socklen_t *addrlen) {int connect_d &＃61; accept(sockfd, addr, addrlen);if (connect_d &＃61;&＃61; -1)error("Can&＃39;t open secondary socket", 1);return connect_d;
}// Send a string to a client.
int say(int socket, char *s) {int result &＃61; send(socket, s, strlen(s), 0);if(result &＃61;&＃61; -1)// when error don&＃39;t exit.error("Error talking to the client", 0);return result;
}// This will store the main listener socket for the server.
int listener_d;// If someone hits Ctrl-C when the server is running,
// this function will close the socket before the program ends.
void handle_shutdown(int sig) {if(listener_d)close(listener_d);fprintf(stderr, "Signal:%i, Bye!\n", sig);exit(0);
}int main(int argc, char *argv[]) {int port;int queue;char opt;// get option argumentwhile ((opt &＃61; getopt(argc, argv, "hp:q:")) !&＃61; EOF) {switch(opt) {case &＃39;h&＃39;:fprintf(stdout, "%s -p $port_num -q $queue_num\n", argv[0]);exit(1);case &＃39;p&＃39;:port &＃61; atoi(optarg);break;case &＃39;q&＃39;:queue &＃61; atoi(optarg);break;default:fprintf(stderr, "Unknown option: %s\n", optarg);return 1;}}// if argument count is not 5, exit main function.if(argc !&＃61; 5) {fprintf(stdout, "%s -p $port_num -q $queue_num\n", argv[0]);return 1;}// 注册SIGINT 到信号处理函数handle_shutdown.if(catch_signal(SIGINT, handle_shutdown) &＃61;&＃61; -1)error("Can&＃39;t set the interrupt handler", 1);// 打开一个data stream socket.listener_d &＃61; open_listener_socket();// 绑定到端口portbind_to_port(listener_d, port);// 监听, 并指定queue长度listen_to_socket(listener_d, queue);// 准备接收客户端连接struct sockaddr_storage client_addr;unsigned int address_size &＃61; sizeof(client_addr);puts("Waiting for connection");char buf[255];// accept_socket等待客户端连接.while(1) {// accept将创建the secondary socketfd.int connect_d &＃61; accept_socket(listener_d, (struct sockaddr *) &client_addr, &address_size);// 子进程处理客户端请求if ( !fork() ) {close(listener_d);if (say(connect_d, "Internet Knock-Knock Protocol Server\r\nVersion 1.0\r\nKnock! Knock!\r\n> ") !&＃61; -1) {read_in(connect_d, buf, sizeof(buf));}if (strncasecmp("Who&＃39;s there?", buf, 12))say(connect_d, "You should say &＃39;Who&＃39;s there?&＃39;!");else {if (say(connect_d, "Oscar\r\n> ") !&＃61; -1) {read_in(connect_d, buf, sizeof(buf));if (strncasecmp("Oscar who?", buf, 10))say(connect_d, "You should say &＃39;Oscar who?&＃39;!\r\n");elsesay(connect_d, "Oscar silly question, you get a silly answer\r\n");}}// 子进程处理完请求后关闭the secondary socketfd. 并退出.close(connect_d);exit(0);}// 主进程不与客户端继续交互, 所以直接关闭the secondary socketfd. 继续while循环, 等待客户端连接.close(connect_d);}return 0;
}
2. 编译, 执行.

[root&＃64;db-172-16-3-150 zzz]# gcc -O3 -Wall -Wextra -Werror -I. -g ./server.c -o server
[root&＃64;db-172-16-3-150 zzz]# ./server -h
./server -p $port_num -q $queue_num
[root&＃64;db-172-16-3-150 zzz]# ./server -p 30000 -q 10 // 因为本例用子进程处理客户端请求, 主进程及时的close(connect_d), 所以这个queue不需要太大. 0就可以了.
Waiting for connection

其他会话1 : 

[root&＃64;db-172-16-3-150 zzz]# telnet 172.16.3.150 30000
Trying 172.16.3.150...
Connected to db-172-16-3-150.sky-mobi.com (172.16.3.150).
Escape character is &＃39;^]&＃39;.
Internet Knock-Knock Protocol Server
Version 1.0
Knock! Knock!
> hello
You should say &＃39;Who&＃39;s there?&＃39;!Connection closed by foreign host.

其他会话2 : 

[root&＃64;db-172-16-3-150 zzz]# telnet 172.16.3.150 30000
Trying 172.16.3.150...
Connected to db-172-16-3-150.sky-mobi.com (172.16.3.150).
Escape character is &＃39;^]&＃39;.
Internet Knock-Knock Protocol Server
Version 1.0
Knock! Knock!
> Who&＃39;s there?
Oscar
> Oscar who?
Oscar silly question, you get a silly answer
Connection closed by foreign host.

【注意】

1. NAMElisten - listen for connections on a socketSYNOPSIS#include int listen(int sockfd, int backlog);
NOTESThe behaviour of the backlog parameter on TCP sockets changed with Linux 2.2. Now it specifies the queuelength for completely established sockets waiting to be accepted, instead of the number of incomplete connec-tion requests. The maximum length of the queue for incomplete sockets can be set using the tcp_max_syn_backlogsysctl. When synCOOKIEs are enabled there is no logical maximum length and this sysctl setting is ignored.See tcp(7) for more information.