THE WORLD OF SELECT()
o, yeah. find a good sample with detailed comment for select fun.
URL: http://www.lowtek.com/sockets/select.html
So just why am I so hyped on select() ?
One traditional way to write network servers is to have the main server block on accept() , waiting for a connection. Once a connection comes in, the server fork() s, the child process handles the connection and the main server is able to service new incoming requests.
With select() , instead of having a process for each request, there is usually only one process that "multi-plexes" all requests, servicing each request as much as it can.
So one main advantage of using select() is that your server will only require a single process to handle all requests. Thus, your server will not need shared memory or synchronization primitives for different 'tasks' to communicate.
One major disadvantage of using select() , is that your server cannot act like there's only one client, like with a fork() 'ing solution. For example, with a fork() 'ing solution, after the server fork() s, the child process works with the client as if there was only one client in the universe -- the child does not have to worry about new incoming connections or the existence of other sockets. With select() , the programming isn't as transparent.
Okay, so how do you use select() ?
select() works by blocking until something happens on a file descriptor (aka a socket). What's 'something'? Data coming in or being able to write to a file descriptor -- you tell select() what you want to be woken up by. How do you tell it? You fill up a fd_set structure with some macros.
Most select() -based servers look pretty much the same:
-
Fill up a fd_set structure with the file descriptors you want to know when data comes in on.
-
Fill up a fd_set structure with the file descriptors you want to know when you can write on.
-
Call select() and block until something happens.
-
Once select() returns, check to see if any of your file descriptors was the reason you woke up. If so, 'service' that file descriptor in whatever particular way your server needs to (i.e. read in a request for a Web page).
-
Repeat this process forever.
Quit with the pseudo-code, show me some real code!
Okay, let's take a look at a sample server included with Vic Metcalfe's Socket Programming FAQ (my comments are in red ):
/*
PLEASE READ THE FILE NB-APOLOGY !!!! There are some things you should
know about this source before you read it. Thanks.
Quang Ngo alerted me to a bug where the variable listnum in deal_with_data()
wasn't being passed in by parameter, thus it was always garbage. I have
quick-fixed this in the code below. - Spencer (October 12, 1999)
Non blocking server demo
By Vic Metcalfe ([email protected])
For the unix-socket-faq
*/
#include "sockhelp.h"
#include <ctype.h>
#include <sys/time.h>
#include <fcntl.h>
int sock; /* The socket file descriptor for our "listening"
socket */
int connectlist[5]; /* Array of connected sockets so we know who
we are talking to */
fd_set socks; /* Socket file descriptors we want to wake
up for, using select() */
int highsock; /* Highest #'d file descriptor, needed for select() */
void setnonblocking(sock)
int sock;
{
int opts;
opts = fcntl(sock,F_GETFL);
if (opts < 0) {
perror("fcntl(F_GETFL)");
exit(EXIT_FAILURE);
}
opts = (opts | O_NONBLOCK);
if (fcntl(sock,F_SETFL,opts) < 0) {
perror("fcntl(F_SETFL)");
exit(EXIT_FAILURE);
}
return;
}
void build_select_list() {
int listnum; /* Current item in connectlist for for loops */
/* First put together fd_set for select(), which will
consist of the sock veriable in case a new connection
is coming in, plus all the sockets we have already
accepted. */
/* FD_ZERO() clears out the fd_set called socks, so that
it doesn't contain any file descriptors. */
FD_ZERO(&socks);
/* FD_SET() adds the file descriptor "sock" to the fd_set,
so that select() will return if a connection comes in
on that socket (which means you have to do accept(), etc. */
FD_SET(sock,&socks);
/* Loops through all the possible connections and adds
those sockets to the fd_set */
for (listnum = 0; listnum < 5; listnum++) {
if (connectlist[listnum] != 0) {
FD_SET(connectlist[listnum],&socks);
if (connectlist[listnum] > highsock)
highsock = connectlist[listnum];
}
}
}
void handle_new_connection() {
int listnum; /* Current item in connectlist for for loops */
int connection; /* Socket file descriptor for incoming connections */
/* We have a new connection coming in! We'll
try to find a spot for it in connectlist. */
connection = accept(sock, NULL, NULL);
if (connection < 0) {
perror("accept");
exit(EXIT_FAILURE);
}
setnonblocking(connection);
for (listnum = 0; (listnum < 5) && (connection != -1); listnum ++)
if (connectlist[listnum] == 0) {
printf("/nConnection accepted: FD=%d; Slot=%d/n",
connection,listnum);
connectlist[listnum] = connection;
connection = -1;
}
if (connection != -1) {
/* No room left in the queue! */
printf("/nNo room left for new client./n");
sock_puts(connection,"Sorry, this server is too busy. "
Try again later!/r/n");
close(connection);
}
}
void deal_with_data(
int listnum /* Current item in connectlist for for loops */
) {
char buffer[80]; /* Buffer for socket reads */
char *cur_char; /* Used in processing buffer */
if (sock_gets(connectlist[listnum],buffer,80) < 0) {
/* Connection closed, close this end
and free up entry in connectlist */
printf("/nConnection lost: FD=%d; Slot=%d/n",
connectlist[listnum],listnum);
close(connectlist[listnum]);
connectlist[listnum] = 0;
} else {
/* We got some data, so upper case it
and send it back. */
printf("/nReceived: %s; ",buffer);
cur_char = buffer;
while (cur_char[0] != 0) {
cur_char[0] = toupper(cur_char[0]);
cur_char++;
}
sock_puts(connectlist[listnum],buffer);
sock_puts(connectlist[listnum],"/n");
printf("responded: %s/n",buffer);
}
}
void read_socks() {
int listnum; /* Current item in connectlist for for loops */
/* OK, now socks will be set with whatever socket(s)
are ready for reading. Lets first check our
"listening" socket, and then check the sockets
in connectlist. */
/* If a client is trying to connect() to our listening
socket, select() will consider that as the socket
being 'readable'. Thus, if the listening socket is
part of the fd_set, we need to accept a new connection. */
if (FD_ISSET(sock,&socks))
handle_new_connection();
/* Now check connectlist for available data */
/* Run through our sockets and check to see if anything
happened with them, if so 'service' them. */
for (listnum = 0; listnum < 5; listnum++) {
if (FD_ISSET(connectlist[listnum],&socks))
deal_with_data(listnum);
} /* for (all entries in queue) */
}
int main (argc, argv)
int argc;
char *argv[];
{
char *ascport; /* ASCII version of the server port */
int port; /* The port number after conversion from ascport */
struct sockaddr_in server_address; /* bind info structure */
int reuse_addr = 1; /* Used so we can re-bind to our port
while a previous connection is still
in TIME_WAIT state. */
struct timeval timeout; /* Timeout for select */
int readsocks; /* Number of sockets ready for reading */
/* Make sure we got a port number as a parameter */
if (argc < 2) {
printf("Usage: %s PORT/r/n",argv[0]);
exit(EXIT_FAILURE);
}
/* Obtain a file descriptor for our "listening" socket */
sock = socket(AF_INET, SOCK_STREAM, 0);
if (sock < 0) {
perror("socket");
exit(EXIT_FAILURE);
}
/* So that we can re-bind to it without TIME_WAIT problems */
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuse_addr,
sizeof(reuse_addr));
/* Set socket to non-blocking with our setnonblocking routine */
setnonblocking(sock);
/* Get the address information, and bind it to the socket */
ascport = argv[1]; /* Read what the user gave us */
port = atoport(ascport); /* Use function from sockhelp to
convert to an int */
memset((char *) &server_address, 0, sizeof(server_address));
server_address.sin_family = AF_INET;
server_address.sin_addr.s_addr = htonl(INADDR_ANY);
server_address.sin_port = port;
if (bind(sock, (struct sockaddr *) &server_address,
sizeof(server_address)) < 0 ) {
perror("bind");
close(sock);
exit(EXIT_FAILURE);
}
/* Set up queue for incoming connections. */
listen(sock,5);
/* Since we start with only one socket, the listening socket,
it is the highest socket so far. */
highsock = sock;
memset((char *) &connectlist, 0, sizeof(connectlist));
while (1) { /* Main server loop - forever */
build_select_list();
timeout.tv_sec = 1;
timeout.tv_usec = 0;
/* The first argument to select is the highest file
descriptor value plus 1. In most cases, you can
just pass FD_SETSIZE and you'll be fine. */
/* The second argument to select() is the address of
the fd_set that contains sockets we're waiting
to be readable (including the listening socket). */
/* The third parameter is an fd_set that you want to
know if you can write on -- this example doesn't
use it, so it passes 0, or NULL. The fourth parameter
is sockets you're waiting for out-of-band data for,
which usually, you're not. */
/* The last parameter to select() is a time-out of how
long select() should block. If you want to wait forever
until something happens on a socket, you'll probably
want to pass NULL. */
readsocks = select(highsock+1, &socks, (fd_set *) 0,
(fd_set *) 0, &timeout);
/* select() returns the number of sockets that had
things going on with them -- i.e. they're readable. */
/* Once select() returns, the original fd_set has been
modified so it now reflects the state of why select()
woke up. i.e. If file descriptor 4 was originally in
the fd_set, and then it became readable, the fd_set
contains file descriptor 4 in it. */
if (readsocks < 0) {
perror("select");
exit(EXIT_FAILURE);
}
if (readsocks == 0) {
/* Nothing ready to read, just show that
we're alive */
printf(".");
fflush(stdout);
} else
read_socks();
} /* while(1) */
} /* main */