begin to work on alias.[hc]
[apps/madmutt.git] / lib-sys / mutt_tunnel.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 2000 Manoj Kasichainula <manoj@io.com>
4  * Copyright (C) 2001,2005 Brendan Cully <brendan@kublai.com>
5  *
6  * This file is part of mutt-ng, see http://www.muttng.org/.
7  * It's licensed under the GNU General Public License,
8  * please see the file GPL in the top level source directory.
9  */
10
11 #if HAVE_CONFIG_H
12 # include "config.h"
13 #endif
14
15 #include <netinet/in.h>
16 #include <sys/types.h>
17 #include <sys/socket.h>
18 #include <sys/wait.h>
19 #include <fcntl.h>
20 #include <errno.h>
21
22 #include <lib-lib/mem.h>
23 #include <lib-lib/str.h>
24 #include <lib-lib/macros.h>
25
26 #include "mutt.h"
27
28 #include "exit.h"
29 #include "mutt_socket.h"
30 #include "mutt_tunnel.h"
31 #include "mutt_signal.h"
32
33 /* -- data types -- */
34 typedef struct {
35   pid_t pid;
36   int readfd;
37   int writefd;
38 } TUNNEL_DATA;
39
40 /* forward declarations */
41 static int tunnel_socket_open (CONNECTION *);
42 static int tunnel_socket_close (CONNECTION *);
43 static int tunnel_socket_read (CONNECTION * conn, char *buf, size_t len);
44 static int tunnel_socket_write (CONNECTION * conn, const char *buf,
45                                 size_t len);
46
47 /* -- public functions -- */
48 int mutt_tunnel_socket_setup (CONNECTION * conn)
49 {
50   conn->conn_open = tunnel_socket_open;
51   conn->conn_close = tunnel_socket_close;
52   conn->conn_read = tunnel_socket_read;
53   conn->conn_write = tunnel_socket_write;
54
55   return 0;
56 }
57
58 static int tunnel_socket_open (CONNECTION * conn)
59 {
60   TUNNEL_DATA *tunnel;
61   int pid;
62   int rc;
63   int pin[2], pout[2];
64
65   tunnel = p_new(TUNNEL_DATA, 1);
66   conn->sockdata = tunnel;
67
68   mutt_message (_("Connecting with \"%s\"..."), Tunnel);
69
70   if ((rc = pipe (pin)) == -1) {
71     mutt_perror ("pipe");
72     return -1;
73   }
74   if ((rc = pipe (pout)) == -1) {
75     mutt_perror ("pipe");
76     return -1;
77   }
78
79   mutt_block_signals_system ();
80   if ((pid = fork ()) == 0) {
81     mutt_unblock_signals_system (0);
82     if (dup2 (pout[0], STDIN_FILENO) < 0 || dup2 (pin[1], STDOUT_FILENO) < 0)
83       _exit (127);
84     close (pin[0]);
85     close (pin[1]);
86     close (pout[0]);
87     close (pout[1]);
88     close (STDERR_FILENO);
89
90     /* Don't let the subprocess think it can use the controlling tty */
91     setsid ();
92
93     execl ("/bin/sh", "sh", "-c", Tunnel, NULL);
94     _exit (127);
95   }
96   mutt_unblock_signals_system (1);
97
98   if (pid == -1) {
99     close (pin[0]);
100     close (pin[1]);
101     close (pout[0]);
102     close (pout[1]);
103     mutt_perror ("fork");
104     return -1;
105   }
106   if (close (pin[1]) < 0 || close (pout[0]) < 0)
107     mutt_perror ("close");
108
109   fcntl (pin[0], F_SETFD, FD_CLOEXEC);
110   fcntl (pout[1], F_SETFD, FD_CLOEXEC);
111
112   tunnel->readfd = pin[0];
113   tunnel->writefd = pout[1];
114   tunnel->pid = pid;
115
116   conn->fd = 42;                /* stupid hack */
117
118   return 0;
119 }
120
121 static int tunnel_socket_close (CONNECTION * conn)
122 {
123   TUNNEL_DATA *tunnel = (TUNNEL_DATA *) conn->sockdata;
124   int status;
125
126   close (tunnel->readfd);
127   close (tunnel->writefd);
128   waitpid (tunnel->pid, &status, 0);
129   if (!WIFEXITED(status) || WEXITSTATUS(status)) {
130     mutt_error(_("Tunnel to %s returned error %d (%s)"),
131                conn->account.host, WEXITSTATUS(status),
132                m_strsysexit(WEXITSTATUS(status)));
133     mutt_sleep (2);
134   }
135   p_delete(&conn->sockdata);
136   return 0;
137 }
138
139 static int tunnel_socket_read (CONNECTION * conn, char *buf, size_t len)
140 {
141   TUNNEL_DATA *tunnel = (TUNNEL_DATA *) conn->sockdata;
142   int rc;
143
144   rc = read (tunnel->readfd, buf, len);
145   if (rc == -1) {
146     mutt_error (_("Tunnel error talking to %s: %s"), conn->account.host,
147                 strerror (errno));
148     mutt_sleep (1);
149   }
150
151   return rc;
152 }
153
154 static int tunnel_socket_write (CONNECTION * conn, const char *buf,
155                                 size_t len)
156 {
157   TUNNEL_DATA *tunnel = (TUNNEL_DATA *) conn->sockdata;
158   int rc;
159
160   rc = write (tunnel->writefd, buf, len);
161   if (rc == -1) {
162     mutt_error (_("Tunnel error talking to %s: %s"), conn->account.host,
163                 strerror (errno));
164     mutt_sleep (1);
165   }
166
167   return rc;
168 }