服务器端:
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#define SERV_PORT 9877
#define SA struct sockaddr
int main(int argc, char**argv)
{
int sockfd;
struct sockaddr_in servaddr, cliaddr;
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(SERV_PORT);
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
bind(sockfd, (SA*)&servaddr, sizeof(servaddr));
dg_echo(sockfd, (SA*)&cliaddr, sizeof(cliaddr));
return 0;
}
客户端:
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#define SERV_PORT 9877
#define SA struct sockaddr
int main(int argc, char** argv)
{
int sockfd;
struct sockaddr_in servaddr;
if(argc != 2)
{
err_quit("usage: udpcli <IPaddress>");
}
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(SERV_PORT);
inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
dg_cli(stdin, sockfd, (SA*)&servaddr, sizeof(servaddr));
return 0;
}
dg_cli.c
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#ifndef MAXLINE
#define MAXLINE 4096
#endif
#define SA struct sockaddr
void dg_cli(FILE* fp, int sockfd, const SA* pservaddr, socklen_t servlen)
{
int n;
char sendline[MAXLINE], recvline[MAXLINE+1];
while(fgets(sendline, MAXLINE, fp) != NULL)
{
sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
n = recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);
recvline[n] = 0;
fputs(recvline, stdout);
}
}
dg_echo.c
#include <stdio.h>
#include <unistd.h>
#include <sys/socket.h>
#ifndef MAXLINE
#define MAXLINE 4096
#endif
#define SA struct sockaddr
void dg_echo(int sockfd, SA* pcliaddr, socklen_t clilen)
{
int n;
socklen_t len;
char mesg[MAXLINE];
for(; ;)
{
len = clilen;
n = recvfrom(sockfd, mesg, MAXLINE, 0, pcliaddr, &len);
sendto(sockfd, mesg, n, 0, pcliaddr, len);
}
}
error.h
#ifndef ERROR_H
#define ERROR_H
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <unistd.h>
#include <syslog.h>
#include <errno.h>
#include "constant.h"
static void err_doit(int, int, const char *, va_list);
void err_ret(const char *fmt, ...);
void err_sys(const char *fmt, ...);
void err_dump(const char *fmt, ...);
void err_msg(const char *fmt, ...);
void err_quit(const char *fmt, ...);
#endif
error.c
#include "error.h"
int daemon_proc; /* set nonzero by daemon_init() */
static void err_doit(int, int, const char *, va_list);
/* Nonfatal error related to system call
* * Print message and return */
void err_ret(const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
err_doit(1, LOG_INFO, fmt, ap);
va_end(ap);
return;
}
/* Fatal error related to system call
* * Print message and terminate */
void err_sys(const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
err_doit(1, LOG_ERR, fmt, ap);
va_end(ap);
exit(1);
}
/* Fatal error related to system call
* * Print message, dump core, and terminate */
void err_dump(const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
err_doit(1, LOG_ERR, fmt, ap);
va_end(ap);
abort(); /* dump core and terminate */
exit(1); /* shouldn't get here */
}
/* Fatal error unrelated to system call
* * Print message and terminate */
void err_quit(const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
err_doit(0, LOG_ERR, fmt, ap);
va_end(ap);
exit(1);
}
/* Print message and return to caller
* * Caller specifies "errnoflag" and "level" */
static void err_doit(int errnoflag, int level, const char *fmt, va_list ap)
{
int errno_save,n;
char buf[MAXLINE + 1];
errno_save = errno; /* value caller might want printed */
#ifdef HAVE_VSNPRINTF
vsnprintf(buf, MAXLINE, fmt, ap); /* safe */
#else
vsprintf(buf, fmt, ap); /* not safe */
#endif
n = strlen(buf);
if (errnoflag)
snprintf(buf + n, MAXLINE - n, ": %s", strerror(errno_save));
strcat(buf, "\n");
if (daemon_proc) {
syslog(level, buf);
}
else
{
fflush(stdout); /* in case stdout and stderr are the same */
fputs(buf, stderr);
fflush(stderr);
}
return;
}
constant.h
#ifndef Constant_H
#define Constant_H
#ifndef MAXLINE
#define MAXLINE 4096
#endif
#ifndef LISTENQ
#define LISTENQ 1024
#endif
#ifndef BUFFSIZE
#define BUFFSIZE 8092
#endif
#endif
makefile
bins = udpserv udpcli
objs = udpserv.o udpcli.o dg_echo.o dg_cli.o error.o
srcs = udpserv.c udpcli.c dg_echo.c dg_cli.c error.c error.h
$(bins):$(objs)
gcc -o udpserv udpserv.o dg_echo.o
gcc -o udpcli udpcli.o dg_cli.o error.o
$(objs):$(srcs)
gcc -c udpserv.c
gcc -c udpcli.c
gcc -c dg_echo.c
gcc -c dg_cli.c
gcc -c error.c error.h
clean:
rm -rf $(bins) *.o *.h.gch