Re: connect(2) for UDP

#include <string.h>

#include <sys/socket.h>
#include <netdb.h>
extern int h_errno;
#include <arpa/inet.h>
#include <sys/ioctl.h>

#include <cassert>

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>

using namespace std;

enum modes {binding, connecting} mode = binding;

char lower( char c ) {
	return tolower(c);
}

int
main( int argc, char *argv[] )
{
	int s1, s2, ch, port = 0, erc;
	sockaddr_in loc, svr;
	
	string hostname;

	while( (ch = getopt(argc, argv, "c:h:p:")) != -1 ) {
		switch (ch) {
		case 'c': {
				string option(optarg);
				
				transform(option.begin(), option.end(), option.begin(), lower);
				
				if( option == "connect" ) {
					mode = connecting;
				} else { 
					if( option != "bind" ) {
						cerr << "-c {bind|connect}\n";
						return 1;
					}
				}
						
			}
			break;
		case 'h':
			hostname = optarg;
			break;
		case 'p': {
				stringstream s(optarg);
				s >> port;
				assert(port != 0);
			}
			break;
		default:
			exit(1);
		}
	}
	
	if( hostname.empty() ) {
		cerr << "need hostname\n";
		return 1;
	}
	if( port == 0 ) {
		cerr << "need port\n";
		return 1;
	}
	
	hostent *ent = gethostbyname(hostname.c_str());
	
	if( !ent ) {
		herror(hostname.c_str());
		return 1;
	}
	
	memcpy( &svr.sin_addr.s_addr, ent->h_addr_list[0], sizeof(svr.sin_addr.s_addr) );
	
	cout << "address of " << hostname << " is " << inet_ntoa( *reinterpret_cast<in_addr*>(&svr.sin_addr.s_addr) ) << endl;

	svr.sin_family = PF_INET;
	svr.sin_port = htons(port);

	if( (s1 = socket(PF_INET, SOCK_DGRAM, 0)) < 0 ) {
		perror("could not create socket");
		return 1;
	}
	if( (s2 = socket(PF_INET, SOCK_DGRAM, 0)) < 0 ) {
		perror("could not create local socket");
		return 1;
	}


	unsigned long ioctl_nonblocking = 1;
	
	if( (erc = ioctl(s1, FIONBIO, &ioctl_nonblocking)) < 0) {
		perror("could not set nonblocking mode");
		return 1;
	}

	if( (erc = ioctl(s2, FIONBIO, &ioctl_nonblocking)) < 0) {
		perror("could not set nonblocking mode");
		return 1;
	}

	char cmd[3] = { 0x03, '\0' };
	if( 1434 != port ) 
		strcpy(cmd, "A\n");
	
	
	if( mode == binding ) {
		loc.sin_addr.s_addr = INADDR_ANY;
		loc.sin_port = htons((short) 4341);
		loc.sin_family = PF_INET;

		cout << "binding local UDP socket ... " << flush;

		if (bind(s2, (struct sockaddr *) &loc, sizeof(loc)) < 0) {
			perror("failed");
			return 1;
		}
		cout << "done\n";

		cout << "sending  to  " << hostname << ":" << port << " ... " << flush;

		if( (erc = sendto(s2, &cmd, strlen(cmd), 0, (const sockaddr*)&svr, sizeof(svr))) == -1 ) {
			perror("failed");
			return 1;
		}

		cout << "done.  (sent " << erc << ")\n";
	} 
	cout << "connecting UDP socket to " << hostname << ":" << port << " ... " << flush;

	if( (erc = connect(s1, (const sockaddr*)&svr, sizeof(svr))) < 0 ) {
		perror("failed");
		return 1;
	}
	cout << "done\n";

	cout << "writing  to  " << hostname << ":" << port << " ... " << flush;

	if( (erc = write(s1, &cmd, strlen(cmd))) == -1 ) {
		perror("failed");
		return 1;
	}

	cout << "done.  (wrote " << erc << ")\n";
	
	fd_set fds;
	FD_ZERO(&fds);
	FD_SET(s1, &fds);
	FD_SET(s2, &fds);

	struct timeval timeout;
	bzero(&timeout, sizeof(timeout));
	timeout.tv_sec = 5;
	timeout.tv_usec = 0;
	
	cout << "reading from " << hostname << ":" << port << " ... " << flush;
	
	if( (erc = select(s2 + 1, &fds, NULL, NULL, &timeout)) == -1 ) {
		perror("select(2) failed");
		return 1;
	}

	if( 0 == erc ) {
		cerr << "timed out, sorry.\n";
		return 1;
	}
	
	cout << erc << " fd ready on " << flush;
	
	int s;
	char buffer[2048];
	
	if( FD_ISSET(s1, &fds) ) {
		s = s1;
		cout << "connected ... " << flush;
	}
	if( FD_ISSET(s2, &fds) ) {
		s = s2;
		cout << "bound ... " << flush;
	}
	
	while( (erc = read(s, &buffer, sizeof(buffer))) == -1 ) {
		perror("could not read");
		return 1;
	}

	cout << "done.  (read " << erc << " bytes)\n" << flush;
	
	for( char *p=buffer; p < buffer + erc; p++ ) {
		if( *p == ';' ) 
			*p = '\n';
		if( isprint(*p) || isspace(*p) )
			putchar(*p);
	}
}