服务器端代码,接收客户端请求,创建子进程,输出接收的数据,同时将数据返回给客户端
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#include <stdlib.h>
#define SIN_PORT 9999
#define BUFLEN 256
void str_echo(int fd);
int main(int argc, char **argv)
{
int listenfd, connfd;
pid_t child;
struct sockaddr_in servaddr;
char buf[BUFLEN];
time_t ticks;
listenfd = socket(AF_INET, SOCK_STREAM, 0);
if (listenfd < 0) {
printf("socket error :%s\n", strerror(errno));
return -1;
}
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(SIN_PORT);
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0) {
printf("bind error:%s\n", strerror(errno));
close(listenfd);
return -1;
}
if (listen(listenfd, 3) < 0) {
printf("listen error:%s\n", strerror(errno));
close(listenfd);
return -1;
}
struct sockaddr_in clientaddr;
int len;
for (;;) {
len = sizeof(clientaddr);
connfd = accept(listenfd, (struct sockaddr*)&clientaddr, &len);
if (connfd < 0) {
printf("accept error:%s\n", strerror(errno));
close(listenfd);
return -1;
}
if ((child = fork()) == 0) {
close(listenfd);
str_echo(connfd);
return 0;
}
/*
printf("connection from %s, port:%d\n", inet_ntop(AF_INET, &clientaddr.sin_addr, buf, sizeof(buf)), htons(clientaddr.sin_port));
ticks = time(NULL);
snprintf(buf, sizeof(buf), "%.24s\r\n", ctime(&ticks));
write(connfd, buf, strlen(buf));
*/
close(connfd);
}
return 0;
}
size_t readn(int fd, void *ptr, size_t n)
{
char *p = ptr;
size_t nleft = n;
size_t nread;
while (nleft > 0) {
if ((nread = read(fd, p, nleft)) < 0) {
if (errno == EINTR) nread = 0;
else return -1;
} else if (nread == 0) break;
nleft -= nread;
p += nread;
}
return n - nleft;
}
size_t readline(int fd, void *ptr, size_t maxsize)
{
char *p = ptr;
size_t rc, n;
char c;
for (n = 1; n < maxsize; n++) {
again:
if ((rc = read(fd, &c, 1)) == 1) {
*p++ = c;
if (c == '\n') break;
} else if (rc == 0) {
if (n == 1) return 0;
else break;
} else {
if (errno == EINTR) goto again;
return -1;
}
}
*p = 0;
return n;
}
size_t writen(int fd, void *ptr, size_t n)
{
char *p = ptr;
size_t nleft = n, nwriten;
while (nleft > 0) {
if ((nwriten = write(fd, p, nleft)) <= 0) {
if (errno == EINTR) nwriten = 0;
else return -1;
}
p += nwriten;
nleft -= nwriten;
}
return n;
}
void str_echo(int fd)
{
char recvline[BUFLEN];
int n;
for (;;) {
if ((n = readline(fd, recvline, BUFLEN)) == 0) return;
printf("received buf=%s", recvline);
writen(fd, recvline, n);
}
}
客户端代码,从标准输入读入数据,发送给客户端,读取服务器端发过来的数据,输出到标准输出
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#define SIN_PORT 9999
#define BUF_LEN 256
size_t readline(int fd, void *ptr, int maxsize)
{
char *p = ptr;
char c;
int rc;
int n;
for (n = 1; n < maxsize; n++) {
again:
if ((rc = read(fd, &c, 1)) == 1) {
*p++ = c;
if (c == '\n') break;
} else if (rc == 0) {
if (n == 1) return 0;
else break;
}else {
if (errno == EINTR) goto again;
else return -1;
}
}
*p = 0;
return n;
}
size_t readn(int fd, void *ptr, size_t n)
{
char *p = ptr;
int nleft = n;
int nread;
while (nleft > 0) {
if ((nread = read(fd, p, nleft)) < 0) {
if (errno == EINTR) nread = 0;
else return -1;
} else if (nread == 0) break;
nleft -= nread;
p += nread;
}
return n - nleft;
}
size_t writen(int fd, void *ptr, size_t n)
{
size_t nwriten;
size_t nleft = n;
char *p = ptr;
while (nleft > 0) {
if ((nwriten = write(fd, p, nleft)) <= 0) {
if (errno == EINTR) nwriten = 0;
else return -1;
}
nleft -= nwriten;
p += nwriten;
}
return n;
}
void str_cli(FILE *file, int fd)
{
char recvline[BUF_LEN], sendline[BUF_LEN];
while (fgets(sendline, BUF_LEN, file) != NULL) {
writen(fd, sendline, strlen(sendline));
if (readline(fd, recvline, BUF_LEN) == 0) {
printf("readline error:%s\n", strerror(errno));
return ;
}
fputs(recvline, stdout);
}
}
int main(int argc, char** argv)
{
struct sockaddr_in serveraddr;
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
printf("socket error:%s\n", strerror(errno));
return -1;
}
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(SIN_PORT);
if (inet_pton(AF_INET, argv[1], &serveraddr.sin_addr) != 1) {
printf("inet_pton error:%s\n", strerror(errno));
close(sockfd);
return -1;
}
if (connect(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr)) < 0) {
printf("connect error:%s\n", strerror(errno));
close(sockfd);
return -1;
}
str_cli(stdin, sockfd);
exit(0);
}