Fix the bugs in the new macros. (Whoops.)

[mLib] / conn.c

/* -*-c-*-
 *
 * $Id: conn.c,v 1.2 1999/05/15 10:33:32 mdw Exp $
 *
 * Nonblocking connect handling
 *
 * (c) 1999 Straylight/Edgeware
 */

/*----- Licensing notice --------------------------------------------------* 
 *
 * This file is part of the mLib utilities library.
 *
 * mLib is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Library General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 * 
 * mLib is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Library General Public License for more details.
 * 
 * You should have received a copy of the GNU Library General Public
 * License along with mLib; if not, write to the Free
 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA 02111-1307, USA.
 */

/*----- Revision history --------------------------------------------------* 
 *
 * $Log: conn.c,v $
 * Revision 1.2  1999/05/15 10:33:32  mdw
 * Fix copyright notices.
 *
 * Revision 1.1  1999/05/14 21:01:14  mdw
 * Integrated `select' handling bits from the background resolver project.
 *
 */

/*----- Header files ------------------------------------------------------*/

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <sys/types.h>

#include <sys/socket.h>
#include <netinet/in.h>

#include <unistd.h>
#include <fcntl.h>

#include "conn.h"
#include "sel.h"

/*----- Main code ---------------------------------------------------------*/

/* --- @conn_connect@ --- *
 *
 * Arguments:	@int fd@ = file descriptor to try to connect
 *		@unsigned mode@ = what we can do to the file
 *		@void *p@ = pointer to connection context
 *
 * Returns:	---
 *
 * Use:		Handles select results for pending connections.
 */

static void conn_connect(int fd, unsigned mode, void *p)
{
  conn *c = p;
  struct sockaddr_in sin;
  int sinsz;

  sinsz = sizeof(sin);
  if (getpeername(fd, (struct sockaddr *)&sin, &sinsz) < 0) {
    int err;
    int errsz = sizeof(err);
    if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errsz) == 0)
      errno = err;
    c->func(-1, c->p);
    close(fd);
  } else
    c->func(fd, c->p);
  sel_rmfile(&c->writer);
}

/* --- @conn_init@ --- *
 *
 * Arguments:	@conn *c@ = pointer to connection block
 *		@sel_state *s@ = pointer to select state to attach to
 *		@unsigned long saddr@ = source IP address
 *		@unsigned short sport@ = source port
 *		@unsigned long daddr@ = destination IP address
 *		@unsigned short dport@ = destination port
 *		@void (*func)(int fd, void *p) = handler function
 *		@void *p@ = argument for the handler function
 *
 * Returns:	---
 *
 * Use:		Sets up a nonblocking connect job.
 */

void conn_init(conn *c, sel_state *s,
	       unsigned long saddr,
	       unsigned short sport,
	       unsigned long daddr,
	       unsigned long dport,
	       void (*func)(int /*fd*/, void */*p*/),
	       void *p)
{
  int fd;

  /* --- Make a socket to do the connecting with --- */

  c->writer.fd = -1;
  if ((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
    goto fail;

  /* --- Make the socket nonblocking --- */

  {
    int f;

    if ((f = fcntl(fd, F_GETFL)) < 0 ||
	fcntl(fd, F_SETFL, f | O_NONBLOCK))
      goto fail_close;
  }

  /* --- Set up the source address and bind it to the socket --- */

  {
    struct sockaddr_in sin;

    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_addr.s_addr = saddr;
    sin.sin_port = sport;
    if (bind(fd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
      goto fail_close;
  }

  /* --- Finally, set up the destination and try the connect --- */

  {
    struct sockaddr_in sin;

    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_addr.s_addr = daddr;
    sin.sin_port = dport;
    if (connect(fd, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
      if (errno != EINPROGRESS)
	goto fail_close;
      c->func = func;
      c->p = p;
      sel_initfile(s, &c->writer, fd, SEL_WRITE, conn_connect, c);
      sel_addfile(&c->writer);
    } else
      func(fd, p);
  }

  /* --- Everything is set up now --- */

  return;

  /* --- Something went pear-shaped --- */

fail_close:
  close(fd);
fail:
  func(-1, p);
}

/* --- @conn_kill@ --- *
 *
 * Arguments:	@conn *c@ = pointer to connection to dispose of
 *
 * Returns:	---
 *
 * Use:		Disposes of a connection when it's not wanted any more.
 */

void conn_kill(conn *c)
{
  if (c->writer.fd != -1) {
    close(c->writer.fd);
    sel_rmfile(&c->writer);
    c->writer.fd = -1;
  }
}

/*----- That's all, folks -------------------------------------------------*/