2005-03-24 Marcus Brinkmann <marcus@g10code.de>
[gpgme.git] / assuan / assuan-socket-connect.c
1 /* assuan-socket-connect.c - Assuan socket based client
2  *      Copyright (C) 2002, 2003 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 <stddef.h>
24 #include <stdio.h>
25 #include <errno.h>
26 #include <sys/types.h>
27 #include <sys/socket.h>
28 #include <sys/un.h>
29 #include <unistd.h>
30
31 #include "assuan-defs.h"
32
33 #define LOG(format, args...) \
34         fprintf (assuan_get_assuan_log_stream (), \
35                  assuan_get_assuan_log_prefix (), \
36                  "%s" format , ## args)
37
38 static int
39 do_finish (ASSUAN_CONTEXT ctx)
40 {
41   if (ctx->inbound.fd != -1)
42     {
43       close (ctx->inbound.fd);
44     }
45   ctx->inbound.fd = -1;
46   ctx->outbound.fd = -1;
47   return 0;
48 }
49
50 static void
51 do_deinit (ASSUAN_CONTEXT ctx)
52 {
53   do_finish (ctx);
54 }
55 /* Make a connection to the Unix domain socket NAME and return a new
56    Assuan context in CTX.  SERVER_PID is currently not used but may
57    become handy in the future.  */
58 AssuanError
59 assuan_socket_connect (ASSUAN_CONTEXT *r_ctx,
60                        const char *name, pid_t server_pid)
61 {
62   static struct assuan_io io = { _assuan_simple_read,
63                                  _assuan_simple_write };
64
65   AssuanError err;
66   ASSUAN_CONTEXT ctx;
67   int fd;
68   struct sockaddr_un srvr_addr;
69   size_t len;
70
71   if (!r_ctx || !name)
72     return ASSUAN_Invalid_Value;
73   *r_ctx = NULL;
74
75   /* we require that the name starts with a slash, so that we can
76      alter reuse this function for other socket types */
77   if (*name != '/')
78     return ASSUAN_Invalid_Value;
79   if (strlen (name)+1 >= sizeof srvr_addr.sun_path)
80     return ASSUAN_Invalid_Value;
81
82   err = _assuan_new_context (&ctx); 
83   if (err)
84       return err;
85   ctx->pid = server_pid; /* save it in case we need it later */
86   ctx->deinit_handler = do_deinit;
87   ctx->finish_handler = do_finish;
88
89   fd = socket (PF_LOCAL, SOCK_STREAM, 0);
90   if (fd == -1)
91     {
92       LOG ("can't create socket: %s\n", strerror (errno));
93       _assuan_release_context (ctx);
94       return ASSUAN_General_Error;
95     }
96
97   memset (&srvr_addr, 0, sizeof srvr_addr);
98   srvr_addr.sun_family = AF_LOCAL;
99   len = strlen (srvr_addr.sun_path) + 1;
100   memcpy (srvr_addr.sun_path, name, len);
101   len += (offsetof (struct sockaddr_un, sun_path));
102
103   if (connect (fd, (struct sockaddr *) &srvr_addr, len) == -1)
104     {
105       LOG ("can't connect to `%s': %s\n", name, strerror (errno));
106       _assuan_release_context (ctx);
107       close (fd);
108       return ASSUAN_Connect_Failed;
109     }
110
111   ctx->inbound.fd = fd;
112   ctx->outbound.fd = fd;
113   ctx->io = &io;
114
115   /* initial handshake */
116   {
117     int okay, off;
118
119     err = _assuan_read_from_server (ctx, &okay, &off);
120     if (err)
121       LOG ("can't connect to server: %s\n", assuan_strerror (err));
122     else if (okay != 1)
123       {
124         LOG ("can't connect to server: `");
125         _assuan_log_sanitized_string (ctx->inbound.line);
126         fprintf (assuan_get_assuan_log_stream (), "'\n");
127         err = ASSUAN_Connect_Failed;
128       }
129   }
130
131   if (err)
132     {
133       assuan_disconnect (ctx); 
134     }
135   else
136     *r_ctx = ctx;
137   return 0;
138 }