minix/test/select/test04_cli.c
2009-11-09 10:26:00 +00:00

148 lines
3.5 KiB
C

/*
* Test name: test04_cli.c
*
* Objective: Test a simple UDP client
*
* Description: Implements a simple echo client using the UDP protocol. First
* it waits until it is possible to write (which is always).
*/
#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/select.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(char *host, long port, udp_io_hdr_t *header)
{
/* configures UDP connection */
char *udp_device;
struct hostent *hp;
int netfd;
nwio_udpopt_t udpopt;
ipaddr_t dirhost;
int result;
/* get host address */
if ((hp = gethostbyname(host)) == (struct hostent*) NULL)
{
fprintf(stderr,"Unknown host\n");
return(-1);
}
memcpy((char *)&dirhost, (char *)hp->h_addr, hp->h_length);
/* Get UDP device */
if (( udp_device = getenv("UDP_DEVICE") ) == NULL)
udp_device = UDP_DEVICE;
/* Get UDP connection */
if ((netfd = open(udp_device, O_RDWR)) < 0)
{
fprintf(stderr,"Error opening UDP device\n");
return -1;
}
/* Configure UDP connection */
udpopt.nwuo_flags = NWUO_COPY | NWUO_LP_SEL | NWUO_EN_LOC | NWUO_DI_BROAD
| NWUO_RP_SET | NWUO_RA_SET | NWUO_RWDATALL | NWUO_DI_IPOPT;
udpopt.nwuo_remaddr = dirhost;
udpopt.nwuo_remport = (udpport_t) htons(port);
if ((result = ioctl(netfd, NWIOSUDPOPT, &udpopt) ) <0)
{
fprintf(stderr, "Error establishing communication\n");
printf("Error: %d\n",result);
close(netfd);
return -1;
}
/* Get configuration for UDP comm */
if ((result = ioctl(netfd, NWIOGUDPOPT, &udpopt) ) < 0)
{
fprintf(stderr,"Error getting configuration\n");
printf("Error: %d\n", result);
close(netfd);
return -1;
}
header->uih_src_addr = udpopt.nwuo_locaddr;
header->uih_dst_addr = udpopt.nwuo_remaddr;
header->uih_src_port = udpopt.nwuo_locport;
header->uih_dst_port = udpopt.nwuo_remport;
return netfd;
}
int main(int argc,char *argv[]) {
int fd;
ssize_t data_read;
udp_buffer_t buffer_send, buffer_rec;
fd_set fds_write;
int ret;
/* Check parameters */
if (argc !=2) {
fprintf(stderr,"Usage: %s host\n", argv[0]);
exit(-1);
}
if ((fd = udp_conf(argv[1], PORT, &buffer_send.header) ) < 0)
exit(-1);
/* init fd_set */
FD_ZERO(&fds_write);
FD_SET(fd, &fds_write);
while (1)
{
/* Wait until it is possible to write with select */
ret = select(4, NULL, &fds_write, NULL, NULL);
if (ret < 0) {
fprintf(stderr, "Error on select waiting for write: %d\n", errno);
exit(-1);
}
if (!FD_ISSET(fd, &fds_write)) {
fprintf(stderr, "Error: The net connection is not ready for writing (?)\n");
exit(-1);
}
/* Get a string and send it */
printf("Ready to write...\n");
printf("Send data: ");
gets(buffer_send.data);
write(fd, &buffer_send, sizeof(udp_buffer_t));
/* If data sent is exit then break */
if (!strcmp(buffer_send.data,"exit"))
break;
/* Get server response */
data_read = read(fd, &buffer_rec, sizeof(udp_buffer_t));
printf("Received: %s\n\n", buffer_rec.data);
}
/* Close UDP communication */
close(fd);
}