服务器端:
#include <stdio.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <string.h>
#define LISTENQ 1024
#define SERV_PORT 9877
#define SA struct sockaddr
int main(int argc, char** argv)
{
int listenfd, connfd;
pid_t childpid;
socklen_t clilen;
struct sockaddr_in cliaddr, servaddr;
listenfd = socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(SERV_PORT);
bind(listenfd, (SA*)&servaddr, sizeof(servaddr));
listen(listenfd, LISTENQ);
for(;;){
clilen = sizeof(cliaddr);
connfd = accept(listenfd, (SA*)&cliaddr, &clilen);
if((childpid = fork()) == 0){
close(listenfd);
str_echo(connfd);
exit(0);
}
close(connfd);
}
}
#include <stdio.h>
#include <errno.h>
#define MAXLINE 1024
void str_echo(int sockfd)
{
ssize_t n;
char buf[MAXLINE];
again:
while((n = read(sockfd, buf, MAXLINE)) > 0)
writen(sockfd, buf, n);
if(n < 0 && errno == EINTR){
goto again;
}
else if(n < 0){
printf("str_echo: read error");
}
}
#include <stdio.h>
#include <errno.h>
ssize_t writen(int fd, void* vptr, size_t n)
{
size_t nleft;
ssize_t nwritten;
const char* ptr;
ptr = vptr;
nleft = n;
while(nleft > 0){
if((nwritten = write(fd, ptr, nleft)) <= 0){
if(nwritten < 0 && errno == EINTR){
nwritten = 0;
}
else{
return(-1);
}
}
nleft -= nwritten;
ptr += nwritten;
}
return(n);
}
客户端程序:
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <string.h>
#define SER_PORT 9877
#define SA struct sockaddr
int main(int argc, char** argv)
{
int sockfd;
struct sockaddr_in servaddr;
if(argc != 2){
printf("param");
}
sockfd = socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(SER_PORT);
inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
connect(sockfd, (SA*)&servaddr, sizeof(servaddr));
str_cli(stdin, sockfd);
exit(0);
}
#include <sys/types.h>
#include <stdio.h>
#include <string.h>
#define MAXLINE 1024
void str_cli(FILE* fp, int sockfd)
{
char sendline[MAXLINE] = {0}, recvline[MAXLINE] = {0};
while(fgets(sendline, MAXLINE, fp) != NULL){
writen(sockfd, sendline, strlen(sendline));
if(readline(sockfd, recvline, MAXLINE) == 0)
printf("str_cli: server terminated");
fputs(recvline, stdout);
}
}
#include <stdio.h>
#include <errno.h>
#define MAXLINE 4096
static int read_cnt;
static char* read_ptr;
static char read_buf[MAXLINE];
static ssize_t my_read(int fd, char* ptr)
{
if(read_cnt <= 0){
again:
if((read_cnt = read(fd, read_buf, sizeof(read_buf))) < 0){
if(errno == EINTR){
goto again;
}
return (-1);
}
else if(read_cnt == 0){
return (0);
}
read_ptr = read_buf;
}
read_cnt--;
*ptr = *read_ptr++;
return(1);
}
ssize_t readline(int fd, void* vptr, size_t maxlen)
{
ssize_t n, rc;
char c, *ptr;
ptr = vptr;
for(n = 1; n < maxlen; n++){
if((rc = my_read(fd, &c)) == 1){
*ptr++ = c;
if(c == '\n'){
break;
}
}
else if(rc == 0){
*ptr = 0;
return (n -1);
}
else{
return (-1);
}
}
*ptr = 0;
return(n);
}
ssize_t readlinebuf(void** vptrptr)
{
if(read_cnt){
*vptrptr = read_ptr;
}
return(read_cnt);
}
Makefile文件
CC = gcc
OBJS = ser01.o str_echo.o writen.o cli01.o str_cli.o readline.o
EXEC = ser01 cli01
$(EXEC): $(OBJS)
$(CC) -g -o ser01 ser01.o str_echo.o writen.o
$(CC) -g -o cli01 cli01.o str_cli.o writen.o readline.o
ser01.o: ser01.c str_echo.c writen.c
$(CC) -g -c $<
cli01.o: cli01.c str_cli.c writen.c readline.c
$(CC) -g -c $<
str_echo.o: str_echo.c writen.c
$(CC) -g -c $<
str_cli.o: str_cli.c writen.c readline.c
$(CC) -g -c $<
writen.o: writen.c
$(CC) -g -c $<
readline.o: readline.c
$(CC) -g -c $<
clean:
rm -rf $(EXEC) $(EXECLI) *.o