198 lines
5.3 KiB
C
Executable file
198 lines
5.3 KiB
C
Executable file
/*
|
|
* Test name: test05_srv.c
|
|
*
|
|
* Objective: Test an impatient UDP server with timeouts
|
|
*
|
|
* Description: Implements an echo server using the UDP protocol. It is
|
|
* based on test04_srv, but it has a timeout value.
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/select.h>
|
|
#include <sys/asynchio.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <net/netlib.h>
|
|
#include <net/gen/netdb.h>
|
|
#include <net/gen/in.h>
|
|
#include <net/gen/udp.h>
|
|
#include <net/gen/udp_hdr.h>
|
|
#include <net/gen/udp_io.h>
|
|
#include <net/hton.h>
|
|
|
|
#define PORT 6000L
|
|
|
|
/* type for received data */
|
|
typedef struct
|
|
{
|
|
udp_io_hdr_t header;
|
|
char data[1024];
|
|
} udp_buffer_t;
|
|
|
|
int udp_conf(long port) {
|
|
|
|
char *udp_device;
|
|
int netfd;
|
|
nwio_udpopt_t udpopt;
|
|
|
|
/* Get default UDP device */
|
|
if ((udp_device = getenv("UDP_DEVICE")) == NULL)
|
|
udp_device = UDP_DEVICE;
|
|
|
|
/* Open UDP connection */
|
|
if ((netfd = open(udp_device, O_RDWR)) < 0)
|
|
{
|
|
fprintf(stderr,"Error opening UDP connection\n");
|
|
return -1;
|
|
}
|
|
|
|
/* Configure UDP connection */
|
|
udpopt.nwuo_flags = NWUO_COPY | NWUO_LP_SET | NWUO_EN_LOC | NWUO_DI_BROAD
|
|
| NWUO_RP_ANY | NWUO_RA_ANY | NWUO_RWDATALL | NWUO_DI_IPOPT;
|
|
|
|
udpopt.nwuo_locport = (udpport_t) htons(port);
|
|
|
|
if ((ioctl(netfd, NWIOSUDPOPT, &udpopt))<0)
|
|
{
|
|
fprintf(stderr,"Error configuring the connection\n");
|
|
close(netfd);
|
|
return -1;
|
|
}
|
|
|
|
/* Get conf options */
|
|
if ((ioctl(netfd, NWIOGUDPOPT, &udpopt))<0)
|
|
{
|
|
fprintf(stderr,"Error getting the conf\n");
|
|
close(netfd);
|
|
return -1;
|
|
}
|
|
|
|
return netfd;
|
|
}
|
|
|
|
|
|
int main(int argc,char *argv[]) {
|
|
int fd;
|
|
ssize_t data_read;
|
|
udp_buffer_t buffer;
|
|
ipaddr_t tmp_addr;
|
|
udpport_t tmp_port;
|
|
int ret;
|
|
fd_set fds_read;
|
|
struct timeval timeout;
|
|
ipaddr_t client_addr, this_addr;
|
|
udpport_t client_port;
|
|
|
|
if ((fd = udp_conf(PORT)) < 0) {
|
|
fprintf(stderr, "Error configuring UDP connection\n");
|
|
exit(-1);
|
|
}
|
|
printf("Waiting for messages on port: %ld\n", PORT);
|
|
fflush(stdout);
|
|
|
|
|
|
/* get a first message so we know who is the client and we can harass it
|
|
afterwards */
|
|
|
|
/* Initialize fd_set */
|
|
FD_ZERO(&fds_read);
|
|
FD_SET(fd, &fds_read);
|
|
/* Set timeout structure */
|
|
timeout.tv_sec = 3;
|
|
timeout.tv_usec = 0;
|
|
ret = select(4, &fds_read, NULL, NULL, NULL);
|
|
|
|
if (ret < 0) {
|
|
fprintf(stderr, "Error in select\n");
|
|
exit(-1);
|
|
}
|
|
if (!FD_ISSET(fd, &fds_read)) {
|
|
fprintf(stderr, "Error: Should be receiving some data from network(?)\n");
|
|
exit(-1);
|
|
}
|
|
printf("Ready to receive...\n");
|
|
/* Read received data */
|
|
data_read = read(fd, &buffer, sizeof(udp_buffer_t));
|
|
printf("Received data: %s\n", buffer.data);
|
|
|
|
/* Can exit if the received string == exit */
|
|
if (!strcmp(buffer.data,"exit"))
|
|
exit(0);
|
|
|
|
/* Send data back, swap addresses */
|
|
tmp_addr = buffer.header.uih_src_addr;
|
|
buffer.header.uih_src_addr = buffer.header.uih_dst_addr;
|
|
buffer.header.uih_dst_addr = tmp_addr;
|
|
/* save address of both ends */
|
|
client_addr = tmp_addr;
|
|
this_addr = buffer.header.uih_src_addr;
|
|
|
|
/* Swap ports */
|
|
tmp_port = buffer.header.uih_src_port;
|
|
buffer.header.uih_src_port = buffer.header.uih_dst_port;
|
|
buffer.header.uih_dst_port = tmp_port;
|
|
/* save client port */
|
|
client_port = tmp_port;
|
|
|
|
/* Write the same back */
|
|
write(fd, &buffer, data_read);
|
|
|
|
while (1)
|
|
{
|
|
|
|
/* Initialize fd_set */
|
|
FD_ZERO(&fds_read);
|
|
FD_SET(fd, &fds_read);
|
|
/* Set timeout structure */
|
|
timeout.tv_sec = 3;
|
|
timeout.tv_usec = 0;
|
|
/* Wait for data available to be read (timeout) */
|
|
ret = select(4, &fds_read, NULL, NULL, &timeout);
|
|
if (ret < 0) {
|
|
fprintf(stderr, "Error on select: %d", errno);
|
|
exit(-1);
|
|
}
|
|
/* if timeout */
|
|
if (ret == 0) {
|
|
/* Send angry msg to client asking for more */
|
|
printf("Tired of waiting, send client an angry message\n");
|
|
buffer.header.uih_src_addr = this_addr;
|
|
buffer.header.uih_dst_addr = client_addr;
|
|
buffer.header.uih_src_port = PORT;
|
|
buffer.header.uih_dst_port = client_port;
|
|
strcpy(buffer.data, "Hey! I want to receive some data!\n");
|
|
write(fd, &buffer, sizeof(udp_buffer_t));
|
|
}
|
|
/* If receive data from network */
|
|
if (FD_ISSET(fd, &fds_read)) {
|
|
printf("Ready to receive...\n");
|
|
/* Read received data */
|
|
data_read = read(fd, &buffer, sizeof(udp_buffer_t));
|
|
printf("Received data: %s\n", buffer.data);
|
|
|
|
/* Can exit if the received string == exit */
|
|
if (!strcmp(buffer.data,"exit"))
|
|
break;
|
|
|
|
/* Send data back, swap addresses */
|
|
tmp_addr = buffer.header.uih_src_addr;
|
|
buffer.header.uih_src_addr = buffer.header.uih_dst_addr;
|
|
buffer.header.uih_dst_addr = tmp_addr;
|
|
|
|
/* Swap ports */
|
|
tmp_port = buffer.header.uih_src_port;
|
|
buffer.header.uih_src_port = buffer.header.uih_dst_port;
|
|
buffer.header.uih_dst_port = tmp_port;
|
|
|
|
/* Write the same back */
|
|
write(fd, &buffer, data_read);
|
|
}
|
|
}
|
|
close(fd);
|
|
}
|