Select test-set imported from the minix contributions, with permission

to modify and distribute from Jose Manuel Gomez, the author (jmgomez@linuxmail.org).
2005-06-17 13:45:40 +00:00
parent f81ac6ae5c
commit ed2a024d47
22 changed files with 2509 additions and 0 deletions
@@ -0,0 +1,198 @@
+/*
+ * 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);
+}