server
/**
* select server端测试代码
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <time.h>
#define PORT 8888
#define ADDR "127.0.0.1"
#define MAX_FD 1024
int server_init()
{
struct sockaddr_in server_addr, client_addr;
socklen_t addr_len;
int listen_fd, client_fd;
fd_set rfds;
struct timeval timeout;
int result;
int maxfd;
listen_fd = socket(AF_INET, SOCK_STREAM, 0);
if (listen_fd < 0) {
perror("create listen fd fail\n");
return -1;
}
server_addr.sin_addr.s_addr = inet_addr(ADDR);
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(PORT);
if (bind(listen_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
perror("bind fail\n");
return -1;
}
listen(listen_fd, 5);
timeout.tv_sec = 30;
timeout.tv_usec = 0;
FD_ZERO(&rfds);
FD_SET(listen_fd, &rfds);
maxfd = MAX_FD;
fprintf(stdout, "server start, listen on port %u....\n", PORT);
while(1) {
int i = 0;
result = select(maxfd, &rfds, NULL, NULL, NULL);
if (result < 0) {
perror("select fail\n");
return -1;
}
if (result == 0) {
fprintf(stdout, "select timeout\n");
continue;
}
for (i = 0;i <= result; i++) {
if (FD_ISSET(listen_fd, &rfds)) {
fprintf(stdout, "server fd can read\n");
client_fd = accept(listen_fd, (struct sockaddr*)&client_addr,
&addr_len);
if (client_fd < 0) {
perror("accept fail\n");
return -1;
}
fprintf(stdout, "server recv a client, addr(%s), port(%u)\n",
inet_ntoa(client_addr.sin_addr.s_addr), ntohs(client_addr.sin_port));
FD_SET(client_fd, &rfds);
if (FD_ISSET(client_fd, &rfds)) {
fprintf(stdout, "client can read:\n");
close(client_fd);
}
}
}
}
}
int main()
{
server_init();
}
#include <sched.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/epoll.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define SERV_PORT 8888
#define SERV_ADDR "127.0.0.1"
#define BUF_SIZE 1024
void connect_server()
{
int client_fd;
struct sockaddr_in serv_addr;
char send_buf[BUF_SIZE] = {0};
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(SERV_PORT);
serv_addr.sin_addr.s_addr = inet_addr(SERV_ADDR);
client_fd = socket(AF_INET, SOCK_STREAM, 0);
if (client_fd < 0)
{
fprintf(stderr, "create socket fail\n");
exit(1);
}
if (connect(client_fd, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) < 0)
{
fprintf(stderr, "connect to server fail\n");
exit(1);
}
close(client_fd);
sleep(9);
printf("client close\n");
}
int main()
{
connect_server();
}