comparison: src/protocols/irc/dcc_send.c
src/protocols/irc/dcc_send.c
- branch
- gaim
- changeset 20470
- 77693555855f
- parent 13071
- b98e72d4089a
- parent 20469
- b2836a24d81e
- child 20471
- 1966704b3e42
equal
deleted
inserted
replaced
| 13071:b98e72d4089a | 20470:77693555855f |
|---|---|
| 1 /** | |
| 2 * @file dcc_send.c Functions used in sending files with DCC SEND | |
| 3 * | |
| 4 * gaim | |
| 5 * | |
| 6 * Copyright (C) 2004, Timothy T Ringenbach <omarvo@hotmail.com> | |
| 7 * Copyright (C) 2003, Robbert Haarman <gaim@inglorion.net> | |
| 8 * | |
| 9 * This program is free software; you can redistribute it and/or modify | |
| 10 * it under the terms of the GNU General Public License as published by | |
| 11 * the Free Software Foundation; either version 2 of the License, or | |
| 12 * (at your option) any later version. | |
| 13 * | |
| 14 * This program is distributed in the hope that it will be useful, | |
| 15 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 17 * GNU General Public License for more details. | |
| 18 * | |
| 19 * You should have received a copy of the GNU General Public License | |
| 20 * along with this program; if not, write to the Free Software | |
| 21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
| 22 */ | |
| 23 | |
| 24 #include "internal.h" | |
| 25 #include "irc.h" | |
| 26 #include "debug.h" | |
| 27 #include "ft.h" | |
| 28 #include "notify.h" | |
| 29 #include "network.h" | |
| 30 | |
| 31 /*************************************************************************** | |
| 32 * Functions related to receiving files via DCC SEND | |
| 33 ***************************************************************************/ | |
| 34 | |
| 35 struct irc_xfer_rx_data { | |
| 36 gchar *ip; | |
| 37 }; | |
| 38 | |
| 39 static void irc_dccsend_recv_destroy(GaimXfer *xfer) | |
| 40 { | |
| 41 struct irc_xfer_rx_data *xd = xfer->data; | |
| 42 | |
| 43 if (xd->ip != NULL) | |
| 44 g_free(xd->ip); | |
| 45 | |
| 46 g_free(xd); | |
| 47 } | |
| 48 | |
| 49 /* | |
| 50 * This function is called whenever data is received. | |
| 51 * It sends the acknowledgement (in the form of a total byte count as an | |
| 52 * unsigned 4 byte integer in network byte order) | |
| 53 */ | |
| 54 static void irc_dccsend_recv_ack(GaimXfer *xfer, const guchar *data, size_t size) { | |
| 55 unsigned long l; | |
| 56 | |
| 57 l = htonl(xfer->bytes_sent); | |
| 58 write(xfer->fd, &l, sizeof(l)); | |
| 59 } | |
| 60 | |
| 61 static void irc_dccsend_recv_init(GaimXfer *xfer) { | |
| 62 struct irc_xfer_rx_data *xd = xfer->data; | |
| 63 | |
| 64 gaim_xfer_start(xfer, -1, xd->ip, xfer->remote_port); | |
| 65 g_free(xd->ip); | |
| 66 xd->ip = NULL; | |
| 67 } | |
| 68 | |
| 69 /* This function makes the necessary arrangements for receiving files */ | |
| 70 void irc_dccsend_recv(struct irc_conn *irc, const char *from, const char *msg) { | |
| 71 GaimXfer *xfer; | |
| 72 struct irc_xfer_rx_data *xd; | |
| 73 gchar **token; | |
| 74 struct in_addr addr; | |
| 75 GString *filename; | |
| 76 int i = 0; | |
| 77 guint32 nip; | |
| 78 | |
| 79 token = g_strsplit(msg, " ", 0); | |
| 80 if (!token[0] || !token[1] || !token[2]) { | |
| 81 g_strfreev(token); | |
| 82 return; | |
| 83 } | |
| 84 | |
| 85 filename = g_string_new(""); | |
| 86 if (token[0][0] == '"') { | |
| 87 if (!strchr(&(token[0][1]), '"')) { | |
| 88 g_string_append(filename, &(token[0][1])); | |
| 89 for (i = 1; token[i]; i++) | |
| 90 if (!strchr(token[i], '"')) { | |
| 91 g_string_append_printf(filename, " %s", token[i]); | |
| 92 } else { | |
| 93 g_string_append_len(filename, token[i], strlen(token[i]) - 1); | |
| 94 break; | |
| 95 } | |
| 96 } else { | |
| 97 g_string_append_len(filename, &(token[0][1]), strlen(&(token[0][1])) - 1); | |
| 98 } | |
| 99 } else { | |
| 100 g_string_append(filename, token[0]); | |
| 101 } | |
| 102 | |
| 103 if (!token[i] || !token[i+1] || !token[i+2]) { | |
| 104 g_strfreev(token); | |
| 105 g_string_free(filename, TRUE); | |
| 106 return; | |
| 107 } | |
| 108 i++; | |
| 109 | |
| 110 xfer = gaim_xfer_new(irc->account, GAIM_XFER_RECEIVE, from); | |
| 111 xd = g_new0(struct irc_xfer_rx_data, 1); | |
| 112 xfer->data = xd; | |
| 113 | |
| 114 gaim_xfer_set_filename(xfer, filename->str); | |
| 115 xfer->remote_port = atoi(token[i+1]); | |
| 116 | |
| 117 nip = strtoul(token[i], NULL, 10); | |
| 118 if (nip) { | |
| 119 addr.s_addr = htonl(nip); | |
| 120 xd->ip = g_strdup(inet_ntoa(addr)); | |
| 121 } else { | |
| 122 xd->ip = g_strdup(token[i]); | |
| 123 } | |
| 124 gaim_debug(GAIM_DEBUG_INFO, "irc", "Receiving file from %s\n", | |
| 125 xd->ip); | |
| 126 gaim_xfer_set_size(xfer, token[i+2] ? atoi(token[i+2]) : 0); | |
| 127 | |
| 128 gaim_xfer_set_init_fnc(xfer, irc_dccsend_recv_init); | |
| 129 gaim_xfer_set_ack_fnc(xfer, irc_dccsend_recv_ack); | |
| 130 | |
| 131 gaim_xfer_set_end_fnc(xfer, irc_dccsend_recv_destroy); | |
| 132 gaim_xfer_set_request_denied_fnc(xfer, irc_dccsend_recv_destroy); | |
| 133 gaim_xfer_set_cancel_send_fnc(xfer, irc_dccsend_recv_destroy); | |
| 134 | |
| 135 gaim_xfer_request(xfer); | |
| 136 g_strfreev(token); | |
| 137 g_string_free(filename, TRUE); | |
| 138 } | |
| 139 | |
| 140 /******************************************************************* | |
| 141 * Functions related to sending files via DCC SEND | |
| 142 *******************************************************************/ | |
| 143 | |
| 144 struct irc_xfer_send_data { | |
| 145 gint inpa; | |
| 146 int fd; | |
| 147 guchar *rxqueue; | |
| 148 guint rxlen; | |
| 149 }; | |
| 150 | |
| 151 static void irc_dccsend_send_destroy(GaimXfer *xfer) | |
| 152 { | |
| 153 struct irc_xfer_send_data *xd = xfer->data; | |
| 154 | |
| 155 if (xd == NULL) | |
| 156 return; | |
| 157 | |
| 158 if (xd->inpa > 0) | |
| 159 gaim_input_remove(xd->inpa); | |
| 160 if (xd->fd != -1) | |
| 161 close(xd->fd); | |
| 162 | |
| 163 if (xd->rxqueue) | |
| 164 g_free(xd->rxqueue); | |
| 165 | |
| 166 g_free(xd); | |
| 167 } | |
| 168 | |
| 169 /* just in case you were wondering, this is why DCC is gay */ | |
| 170 static void irc_dccsend_send_read(gpointer data, int source, GaimInputCondition cond) | |
| 171 { | |
| 172 GaimXfer *xfer = data; | |
| 173 struct irc_xfer_send_data *xd = xfer->data; | |
| 174 char *buffer[16]; | |
| 175 int len; | |
| 176 | |
| 177 if ((len = read(source, buffer, sizeof(buffer))) <= 0) { | |
| 178 gaim_input_remove(xd->inpa); | |
| 179 xd->inpa = 0; | |
| 180 return; | |
| 181 } | |
| 182 | |
| 183 xd->rxqueue = g_realloc(xd->rxqueue, len + xd->rxlen); | |
| 184 memcpy(xd->rxqueue + xd->rxlen, buffer, len); | |
| 185 xd->rxlen += len; | |
| 186 | |
| 187 while (1) { | |
| 188 size_t acked; | |
| 189 | |
| 190 if (xd->rxlen < 4) | |
| 191 break; | |
| 192 | |
| 193 acked = ntohl(*((gint32 *)xd->rxqueue)); | |
| 194 | |
| 195 xd->rxlen -= 4; | |
| 196 if (xd->rxlen) { | |
| 197 unsigned char *tmp = g_memdup(xd->rxqueue + 4, xd->rxlen); | |
| 198 g_free(xd->rxqueue); | |
| 199 xd->rxqueue = tmp; | |
| 200 } else { | |
| 201 g_free(xd->rxqueue); | |
| 202 xd->rxqueue = NULL; | |
| 203 } | |
| 204 | |
| 205 if (acked >= gaim_xfer_get_size(xfer)) { | |
| 206 gaim_input_remove(xd->inpa); | |
| 207 xd->inpa = 0; | |
| 208 gaim_xfer_set_completed(xfer, TRUE); | |
| 209 gaim_xfer_end(xfer); | |
| 210 return; | |
| 211 } | |
| 212 | |
| 213 | |
| 214 } | |
| 215 } | |
| 216 | |
| 217 static gssize irc_dccsend_send_write(const guchar *buffer, size_t size, GaimXfer *xfer) | |
| 218 { | |
| 219 gssize s; | |
| 220 | |
| 221 s = MIN(gaim_xfer_get_bytes_remaining(xfer), size); | |
| 222 if (!s) | |
| 223 return 0; | |
| 224 | |
| 225 return write(xfer->fd, buffer, s); | |
| 226 } | |
| 227 | |
| 228 static void irc_dccsend_send_connected(gpointer data, int source, GaimInputCondition cond) { | |
| 229 GaimXfer *xfer = (GaimXfer *) data; | |
| 230 struct irc_xfer_send_data *xd = xfer->data; | |
| 231 int conn; | |
| 232 | |
| 233 conn = accept(xd->fd, NULL, 0); | |
| 234 if (conn == -1) { | |
| 235 /* Accepting the connection failed. This could just be related | |
| 236 * to the nonblocking nature of the listening socket, so we'll | |
| 237 * just try again next time */ | |
| 238 /* Let's print an error message anyway */ | |
| 239 gaim_debug_warning("irc", "accept: %s\n", strerror(errno)); | |
| 240 return; | |
| 241 } | |
| 242 | |
| 243 gaim_input_remove(xfer->watcher); | |
| 244 xfer->watcher = 0; | |
| 245 close(xd->fd); | |
| 246 xd->fd = -1; | |
| 247 | |
| 248 xd->inpa = gaim_input_add(conn, GAIM_INPUT_READ, irc_dccsend_send_read, xfer); | |
| 249 /* Start the transfer */ | |
| 250 gaim_xfer_start(xfer, conn, NULL, 0); | |
| 251 } | |
| 252 | |
| 253 static void | |
| 254 irc_dccsend_network_listen_cb(int sock, gpointer data) | |
| 255 { | |
| 256 GaimXfer *xfer = data; | |
| 257 struct irc_xfer_send_data *xd; | |
| 258 GaimConnection *gc; | |
| 259 struct irc_conn *irc; | |
| 260 const char *arg[2]; | |
| 261 char *tmp; | |
| 262 struct in_addr addr; | |
| 263 unsigned short int port; | |
| 264 | |
| 265 if (gaim_xfer_get_status(xfer) == GAIM_XFER_STATUS_CANCEL_LOCAL | |
| 266 || gaim_xfer_get_status(xfer) == GAIM_XFER_STATUS_CANCEL_REMOTE) { | |
| 267 gaim_xfer_unref(xfer); | |
| 268 return; | |
| 269 } | |
| 270 | |
| 271 xd = xfer->data; | |
| 272 gc = gaim_account_get_connection(gaim_xfer_get_account(xfer)); | |
| 273 irc = gc->proto_data; | |
| 274 | |
| 275 gaim_xfer_unref(xfer); | |
| 276 | |
| 277 if (sock < 0) { | |
| 278 gaim_notify_error(gc, NULL, _("File Transfer Failed"), | |
| 279 _("Gaim could not open a listening port.")); | |
| 280 gaim_xfer_cancel_local(xfer); | |
| 281 return; | |
| 282 } | |
| 283 | |
| 284 xd->fd = sock; | |
| 285 | |
| 286 port = gaim_network_get_port_from_fd(sock); | |
| 287 gaim_debug_misc("irc", "port is %hu\n", port); | |
| 288 /* Monitor the listening socket */ | |
| 289 xfer->watcher = gaim_input_add(sock, GAIM_INPUT_READ, | |
| 290 irc_dccsend_send_connected, xfer); | |
| 291 | |
| 292 /* Send the intended recipient the DCC request */ | |
| 293 arg[0] = xfer->who; | |
| 294 inet_aton(gaim_network_get_my_ip(irc->fd), &addr); | |
| 295 arg[1] = tmp = g_strdup_printf("\001DCC SEND \"%s\" %u %hu %" G_GSIZE_FORMAT "\001", | |
| 296 xfer->filename, ntohl(addr.s_addr), | |
| 297 port, xfer->size); | |
| 298 | |
| 299 irc_cmd_privmsg(gc->proto_data, "msg", NULL, arg); | |
| 300 g_free(tmp); | |
| 301 } | |
| 302 | |
| 303 /* | |
| 304 * This function is called after the user has selected a file to send. | |
| 305 */ | |
| 306 static void irc_dccsend_send_init(GaimXfer *xfer) { | |
| 307 GaimConnection *gc = gaim_account_get_connection(gaim_xfer_get_account(xfer)); | |
| 308 | |
| 309 xfer->filename = g_path_get_basename(xfer->local_filename); | |
| 310 | |
| 311 gaim_xfer_ref(xfer); | |
| 312 | |
| 313 /* Create a listening socket */ | |
| 314 if (!gaim_network_listen_range(0, 0, SOCK_STREAM, | |
| 315 irc_dccsend_network_listen_cb, xfer)) { | |
| 316 gaim_xfer_unref(xfer); | |
| 317 gaim_notify_error(gc, NULL, _("File Transfer Failed"), | |
| 318 _("Gaim could not open a listening port.")); | |
| 319 gaim_xfer_cancel_local(xfer); | |
| 320 } | |
| 321 | |
| 322 } | |
| 323 | |
| 324 GaimXfer *irc_dccsend_new_xfer(GaimConnection *gc, const char *who) { | |
| 325 GaimXfer *xfer; | |
| 326 struct irc_xfer_send_data *xd; | |
| 327 | |
| 328 /* Build the file transfer handle */ | |
| 329 xfer = gaim_xfer_new(gaim_connection_get_account(gc), GAIM_XFER_SEND, who); | |
| 330 | |
| 331 xd = g_new0(struct irc_xfer_send_data, 1); | |
| 332 xd->fd = -1; | |
| 333 xfer->data = xd; | |
| 334 | |
| 335 /* Setup our I/O op functions */ | |
| 336 gaim_xfer_set_init_fnc(xfer, irc_dccsend_send_init); | |
| 337 gaim_xfer_set_write_fnc(xfer, irc_dccsend_send_write); | |
| 338 gaim_xfer_set_end_fnc(xfer, irc_dccsend_send_destroy); | |
| 339 gaim_xfer_set_request_denied_fnc(xfer, irc_dccsend_send_destroy); | |
| 340 gaim_xfer_set_cancel_send_fnc(xfer, irc_dccsend_send_destroy); | |
| 341 | |
| 342 return xfer; | |
| 343 } | |
| 344 | |
| 345 /** | |
| 346 * Gaim calls this function when the user selects Send File from the | |
| 347 * buddy menu | |
| 348 * It sets up the GaimXfer struct and tells Gaim to go ahead | |
| 349 */ | |
| 350 void irc_dccsend_send_file(GaimConnection *gc, const char *who, const char *file) { | |
| 351 GaimXfer *xfer = irc_dccsend_new_xfer(gc, who); | |
| 352 | |
| 353 /* Perform the request */ | |
| 354 if (file) | |
| 355 gaim_xfer_request_accepted(xfer, file); | |
| 356 else | |
| 357 gaim_xfer_request(xfer); | |
| 358 } |
