2005-10-01 Marcus Brinkmann <marcus@g10code.de>
[gpgme.git] / assuan / assuan-socket-server.c
1 /* assuan-socket-server.c - Assuan socket based server
2  *      Copyright (C) 2002 Free Software Foundation, Inc.
3  *
4  * This file is part of Assuan.
5  *
6  * Assuan is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of
9  * the License, or (at your option) any later version.
10  *
11  * Assuan is distributed in the hope that it will be useful, but
12  * WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA 
19  */
20
21 #include <config.h>
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <errno.h>
25 #include <unistd.h>
26 #include <sys/types.h>
27 #ifndef HAVE_W32_SYSTEM
28 #include <sys/socket.h>
29 #include <sys/un.h>
30 #else
31 #include <windows.h>
32 #endif
33
34 #include "assuan-defs.h"
35
36 static int
37 accept_connection_bottom (assuan_context_t ctx)
38 {
39   int fd = ctx->connected_fd;
40
41 #ifdef HAVE_SO_PEERCRED
42   {
43     /* This overrides any already set PID if the function returns a
44        valid one. */
45     struct ucred cr; 
46     int cl = sizeof cr;
47
48     if ( !getsockopt (fd, SOL_SOCKET, SO_PEERCRED, &cr, &cl)
49          && cr.pid != (pid_t)-1 && cr.pid ) 
50       ctx->pid = cr.pid;
51   }
52 #endif
53
54   ctx->inbound.fd = fd;
55   ctx->inbound.eof = 0;
56   ctx->inbound.linelen = 0;
57   ctx->inbound.attic.linelen = 0;
58   ctx->inbound.attic.pending = 0;
59
60   ctx->outbound.fd = fd;
61   ctx->outbound.data.linelen = 0;
62   ctx->outbound.data.error = 0;
63   
64   ctx->confidential = 0;
65
66   return 0;
67 }
68
69
70 static int
71 accept_connection (assuan_context_t ctx)
72 {
73   int fd;
74   struct sockaddr_un clnt_addr;
75   size_t len = sizeof clnt_addr;
76
77   fd = accept (ctx->listen_fd, (struct sockaddr*)&clnt_addr, &len );
78   if (fd == -1)
79     {
80       ctx->os_errno = errno;
81       return ASSUAN_Accept_Failed;
82     }
83
84   ctx->connected_fd = fd;
85   return accept_connection_bottom (ctx);
86 }
87
88 static int
89 finish_connection (assuan_context_t ctx)
90 {
91   if (ctx->inbound.fd != -1)
92     {
93       _assuan_close (ctx->inbound.fd);
94     }
95   ctx->inbound.fd = -1;
96   ctx->outbound.fd = -1;
97   return 0;
98 }
99
100
101 static void
102 deinit_socket_server (assuan_context_t ctx)
103 {
104   finish_connection (ctx);
105 }
106
107 static struct assuan_io io = { _assuan_simple_read,
108                                _assuan_simple_write };
109
110 /* Initialize a server for the socket LISTEN_FD which has already be
111    put into listen mode */
112 int
113 assuan_init_socket_server (assuan_context_t *r_ctx, int listen_fd)
114 {
115   assuan_context_t ctx;
116   int rc;
117
118   *r_ctx = NULL;
119   ctx = xtrycalloc (1, sizeof *ctx);
120   if (!ctx)
121     return ASSUAN_Out_Of_Core;
122   ctx->is_server = 1;
123   ctx->input_fd = -1;
124   ctx->output_fd = -1;
125
126   ctx->inbound.fd = -1;
127   ctx->outbound.fd = -1;
128
129   ctx->listen_fd = listen_fd;
130   ctx->connected_fd = -1;
131   ctx->deinit_handler = deinit_socket_server;
132   ctx->accept_handler = accept_connection;
133   ctx->finish_handler = finish_connection;
134
135   ctx->io = &io;
136
137   rc = _assuan_register_std_commands (ctx);
138   if (rc)
139     xfree (ctx);
140   else
141     *r_ctx = ctx;
142   return rc;
143 }
144
145 /* Initialize a server using the already accepted socket FD. */
146 int
147 assuan_init_connected_socket_server (assuan_context_t *r_ctx, int fd)
148 {
149   assuan_context_t ctx;
150   int rc;
151
152   *r_ctx = NULL;
153   ctx = xtrycalloc (1, sizeof *ctx);
154   if (!ctx)
155     return ASSUAN_Out_Of_Core;
156   ctx->is_server = 1;
157   ctx->pipe_mode = 1; /* we want a second accept to indicate EOF */
158   ctx->input_fd = -1;
159   ctx->output_fd = -1;
160
161   ctx->inbound.fd = -1;
162   ctx->outbound.fd = -1;
163
164   ctx->io = &io;
165
166   ctx->listen_fd = -1;
167   ctx->connected_fd = fd;
168   ctx->deinit_handler = deinit_socket_server;
169   ctx->accept_handler = accept_connection_bottom;
170   ctx->finish_handler = finish_connection;
171
172   rc = _assuan_register_std_commands (ctx);
173   if (rc)
174     xfree (ctx);
175   else
176     *r_ctx = ctx;
177   return rc;
178 }
179
180