chiark / gitweb /
Expunge revision histories in files.
[mLib] / conn.c
CommitLineData
97f65b00 1/* -*-c-*-
2 *
8656dc50 3 * $Id: conn.c,v 1.10 2004/04/08 01:36:11 mdw Exp $
97f65b00 4 *
5 * Nonblocking connect handling
6 *
a3ddb778 7 * (c) 1999 Straylight/Edgeware
97f65b00 8 */
9
10/*----- Licensing notice --------------------------------------------------*
11 *
12 * This file is part of the mLib utilities library.
13 *
14 * mLib is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU Library General Public License as
16 * published by the Free Software Foundation; either version 2 of the
17 * License, or (at your option) any later version.
18 *
19 * mLib is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU Library General Public License for more details.
23 *
24 * You should have received a copy of the GNU Library General Public
25 * License along with mLib; if not, write to the Free
26 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
27 * MA 02111-1307, USA.
28 */
29
97f65b00 30/*----- Header files ------------------------------------------------------*/
31
32#include <errno.h>
5feb52af 33#include <limits.h>
97f65b00 34#include <stdio.h>
35#include <stdlib.h>
36#include <string.h>
37
38#include <sys/types.h>
97f65b00 39#include <sys/socket.h>
97f65b00 40
41#include <unistd.h>
42#include <fcntl.h>
43
44#include "conn.h"
45#include "sel.h"
46
47/*----- Main code ---------------------------------------------------------*/
48
49/* --- @conn_connect@ --- *
50 *
51 * Arguments: @int fd@ = file descriptor to try to connect
52 * @unsigned mode@ = what we can do to the file
53 * @void *p@ = pointer to connection context
54 *
55 * Returns: ---
56 *
57 * Use: Handles select results for pending connections.
58 */
59
60static void conn_connect(int fd, unsigned mode, void *p)
61{
5feb52af 62#ifndef PATH_MAX
63# define PATH_MAX 1024
64#endif
65
97f65b00 66 conn *c = p;
5feb52af 67 char buf[PATH_MAX + 8]; /* Big enough */
d320826d 68 size_t sinsz;
97f65b00 69
5feb52af 70 sinsz = sizeof(buf);
89f167ab 71 sel_rmfile(&c->writer);
5feb52af 72 if (getpeername(fd, (struct sockaddr *)buf, &sinsz) < 0) {
97f65b00 73 int err;
d320826d 74 size_t errsz = sizeof(err);
97f65b00 75 if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errsz) == 0)
76 errno = err;
97f65b00 77 close(fd);
89f167ab 78 c->func(-1, c->p);
97f65b00 79 } else
80 c->func(fd, c->p);
97f65b00 81}
82
f128ce74 83/* --- @conn_fd@ --- *
84 *
85 * Arguments: @conn *c@ = pointer to connection block
86 * @sel_state *s@ = pointer to select state to attach to
87 * @int fd@ = file descriptor of socket
88 * @void (*func)(int fd, void *p) = handler function
89 * @void *p@ = argument for the handler function
90 *
91 * Returns: ---
92 *
93 * Use: Sets up a nonblocking connect job. The socket should have a
94 * connect pending for it already.
95 */
96
97void conn_fd(conn *c, sel_state *s, int fd,
98 void (*func)(int /*fd*/, void */*p*/),
99 void *p)
100{
101 c->func = func;
102 c->p = p;
103 sel_initfile(s, &c->writer, fd, SEL_WRITE, conn_connect, c);
104 sel_addfile(&c->writer);
105}
106
97f65b00 107/* --- @conn_init@ --- *
108 *
109 * Arguments: @conn *c@ = pointer to connection block
110 * @sel_state *s@ = pointer to select state to attach to
5feb52af 111 * @int fd@ = file descriptor of socket to connect
112 * @struct sockaddr *dst@ = destination address
113 * @int dsz@ = size of destination address
97f65b00 114 * @void (*func)(int fd, void *p) = handler function
115 * @void *p@ = argument for the handler function
116 *
ff34c6e7 117 * Returns: Zero on success, nonzero on failure.
97f65b00 118 *
5feb52af 119 * Use: Sets up a nonblocking connect job. The socket should already
120 * be bound if you care about that sort of thing. When the
121 * connection completes, the handler function is called with the
122 * connected socket as an argument. If the connect fails rather
123 * than completes, the socket is closed, and the handler is
124 * informed of this by being passed a negative file descriptor.
125 * In either case, the select job is then removed.
97f65b00 126 */
127
ff34c6e7 128int conn_init(conn *c, sel_state *s, int fd,
129 struct sockaddr *dst, int dsz,
130 void (*func)(int /*fd*/, void */*p*/),
131 void *p)
97f65b00 132{
5feb52af 133 int f;
97f65b00 134
750c61c0 135 if ((f = fcntl(fd, F_GETFL)) < 0 || fcntl(fd, F_SETFL, f | O_NONBLOCK))
136 goto fail;
97f65b00 137
c9bb3f9e 138 if (!connect(fd, dst, dsz))
139 func(fd, p);
140 else if (errno != EINPROGRESS)
141 goto fail;
f128ce74 142 else
143 conn_fd(c, s, fd, func, p);
ff34c6e7 144 return (0);
750c61c0 145
146fail:
147 close(fd);
148 return (-1);
97f65b00 149}
150
151/* --- @conn_kill@ --- *
152 *
153 * Arguments: @conn *c@ = pointer to connection to dispose of
154 *
155 * Returns: ---
156 *
157 * Use: Disposes of a connection when it's not wanted any more.
158 */
159
160void conn_kill(conn *c)
161{
162 if (c->writer.fd != -1) {
163 close(c->writer.fd);
164 sel_rmfile(&c->writer);
165 c->writer.fd = -1;
166 }
167}
168
169/*----- That's all, folks -------------------------------------------------*/