Mercurial > pidgin3 / file comparison

comparison: src/protocols/icq/filesession.c

src/protocols/icq/filesession.c

changeset 2086
007508451e2c
child 2496
55a3120eb8e6
equal deleted inserted replaced
2085:4659ae7bc69c 2086:007508451e2c
1 /* -*- Mode: C; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2
3 /*
4 * $Id: filesession.c 2096 2001-07-31 01:00:39Z warmenhoven $
5 *
6 * Copyright (C) 1998-2001, Denis V. Dmitrienko <denis@null.net> and
7 * Bill Soudan <soudan@kde.org>
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., 675 Mass Ave, Cambridge, MA 02139, USA.
22 *
23 */
24
25 #include <stdlib.h>
26 #include <fcntl.h>
27 #include <sys/stat.h>
28
29 #ifdef _MSVC_
30 #include <io.h>
31 #define open _open
32 #define close _close
33 #define read _read
34 #define write _write
35 #endif
36
37 #include "icqlib.h"
38 #include "filesession.h"
39 #include "stdpackets.h"
40
41 icq_FileSession *icq_FileSessionNew(icq_Link *icqlink)
42 {
43 icq_FileSession *p=(icq_FileSession *)malloc(sizeof(icq_FileSession));
44
45 if (p)
46 {
47 p->status=0;
48 p->id=0L;
49 p->icqlink=icqlink;
50 p->tcplink=NULL;
51 p->current_fd=-1;
52 p->current_file_num=0;
53 p->current_file_progress=0;
54 p->current_file_size=0;
55 p->files=0L;
56 p->current_speed=100;
57 p->total_bytes=0;
58 p->total_files=0;
59 p->total_transferred_bytes=0;
60 p->working_dir[0]=0;
61 p->user_data=NULL;
62 icq_ListInsert(icqlink->d->icq_FileSessions, 0, p);
63 }
64
65 return p;
66 }
67
68 void icq_FileSessionDelete(void *pv)
69 {
70 icq_FileSession *p=(icq_FileSession *)pv;
71
72 invoke_callback(p->icqlink, icq_FileNotify)(p, FILE_NOTIFY_CLOSE, 0,
73 NULL);
74
75 if(p->files) {
76 char **p2=p->files;
77 while(*p2)
78 free(*(p2++));
79 free(p->files);
80 p->files=NULL;
81 }
82
83 if (p->current_fd > -1 ) {
84 close(p->current_fd);
85 p->current_fd=-1;
86 }
87
88 free(p);
89 }
90
91 int _icq_FindFileSession(void *p, va_list data)
92 {
93 icq_FileSession *psession=(icq_FileSession *)p;
94 DWORD uin=va_arg(data, DWORD);
95 unsigned long id=va_arg(data, unsigned long);
96
97 return (psession->remote_uin == uin) && ( id ? (psession->id == id) : 1 );
98
99 }
100
101 icq_FileSession *icq_FindFileSession(icq_Link *icqlink, DWORD uin,
102 unsigned long id)
103 {
104 return icq_ListTraverse(icqlink->d->icq_FileSessions, _icq_FindFileSession,
105 uin, id);
106 }
107
108 void icq_FileSessionSetStatus(icq_FileSession *p, int status)
109 {
110 if(status!=p->status)
111 {
112 p->status=status;
113 if(p->id)
114 invoke_callback(p->icqlink, icq_FileNotify)(p, FILE_NOTIFY_STATUS,
115 status, NULL);
116 if (status == FILE_STATUS_SENDING)
117 icq_SocketSetHandler(p->tcplink->socket, ICQ_SOCKET_WRITE,
118 icq_FileSessionSendData, p);
119 else
120 icq_SocketSetHandler(p->tcplink->socket, ICQ_SOCKET_WRITE, NULL, NULL);
121 }
122 }
123
124 void icq_FileSessionSetHandle(icq_FileSession *p, const char *handle)
125 {
126 strncpy(p->remote_handle, handle, 64);
127 }
128
129 void icq_FileSessionSetCurrentFile(icq_FileSession *p, const char *filename)
130 {
131 #ifdef _WIN32
132 struct _stat file_status;
133 #else
134 struct stat file_status;
135 #endif
136 char file[1024];
137
138 strcpy(file, p->working_dir);
139 strcat(file, filename);
140
141 if (p->current_fd>-1) {
142 close(p->current_fd);
143 p->current_fd=-1;
144 }
145
146 strncpy(p->current_file, file, 64);
147 p->current_file_progress=0;
148
149 /* does the file already exist? */
150 #ifdef _WIN32
151 if (_stat(file, &file_status)==0) {
152 #else
153 if (stat(file, &file_status)==0) {
154 #endif
155 p->current_file_progress=file_status.st_size;
156 p->total_transferred_bytes+=file_status.st_size;
157 #ifdef _WIN32
158 p->current_fd=open(file, _O_WRONLY | _O_APPEND | _O_BINARY);
159 #else
160 p->current_fd=open(file, O_WRONLY | O_APPEND);
161 #endif
162 } else {
163 #ifdef _WIN32
164 p->current_fd=open(file, _O_WRONLY | _O_CREAT | _O_BINARY,
165 _S_IREAD|_S_IWRITE);
166 #else
167 p->current_fd=open(file, O_WRONLY | O_CREAT, S_IRWXU);
168 #endif
169 }
170
171 /* make sure we have a valid filehandle */
172 if (p->current_fd == -1)
173 perror("couldn't open file: ");
174
175 }
176
177 void icq_FileSessionPrepareNextFile(icq_FileSession *p)
178 {
179 int i=0;
180 char **files=p->files;
181
182 p->current_file_num++;
183
184 while(*files) {
185 i++;
186 if(i==p->current_file_num)
187 break;
188 else
189 files++;
190 }
191
192 if(*files) {
193 #ifdef _WIN32
194 struct _stat file_status;
195 #else
196 struct stat file_status;
197 #endif
198
199 if (p->current_fd>-1) {
200 close(p->current_fd);
201 p->current_fd=-1;
202 }
203
204 #ifdef _WIN32
205 if (_stat(*files, &file_status)==0) {
206 char *basename=*files;
207 char *pos=strrchr(basename, '\\');
208 #else
209 if (stat(*files, &file_status)==0) {
210 char *basename=*files;
211 char *pos=strrchr(basename, '/');
212 #endif
213 if(pos) basename=pos+1;
214 strncpy(p->current_file, basename, 64);
215 p->current_file_progress=0;
216 p->current_file_size=file_status.st_size;
217 #ifdef _WIN32
218 p->current_fd=open(*files, _O_RDONLY | _O_BINARY);
219 #else
220 p->current_fd=open(*files, O_RDONLY);
221 #endif
222 }
223
224 /* make sure we have a valid filehandle */
225 if (p->current_fd == -1)
226 perror("couldn't open file: ");
227 }
228 }
229
230 void icq_FileSessionSendData(icq_FileSession *p)
231 {
232 /* for now just send a packet at a time */
233 char buffer[2048];
234 int count=read(p->current_fd, buffer, 2048);
235
236 if(count>0) {
237 icq_Packet *p2=icq_TCPCreateFile06Packet(count, buffer);
238 icq_TCPLinkSend(p->tcplink, p2);
239 p->total_transferred_bytes+=count;
240 p->current_file_progress+=count;
241 icq_FileSessionSetStatus(p, FILE_STATUS_SENDING);
242
243 invoke_callback(p->icqlink, icq_FileNotify)(p, FILE_NOTIFY_DATAPACKET,
244 count, buffer);
245 }
246
247 /* done transmitting if read returns less that 2048 bytes */
248 if(count<2048)
249 icq_FileSessionClose(p);
250
251 return;
252 }
253
254 /* public */
255
256 void icq_FileSessionSetSpeed(icq_FileSession *p, int speed)
257 {
258 icq_Packet *packet=icq_TCPCreateFile05Packet(speed);
259
260 icq_TCPLinkSend(p->tcplink, packet);
261 }
262
263 void icq_FileSessionClose(icq_FileSession *p)
264 {
265 icq_TCPLink *plink=p->tcplink;
266
267 /* TODO: handle closing already unallocated filesession? */
268
269 /* if we're attached to a tcplink, unattach so the link doesn't try
270 * to close us, and then close the tcplink */
271 if (plink)
272 {
273 plink->session=0L;
274 icq_TCPLinkClose(plink);
275 }
276
277 icq_ListRemove(p->icqlink->d->icq_FileSessions, p);
278 icq_FileSessionDelete(p);
279 }
280
281 void icq_FileSessionSetWorkingDir(icq_FileSession *p, const char *dir)
282 {
283 int length = sizeof(p->working_dir);
284 strncpy(p->working_dir, dir, length);
285 p->working_dir[length-1]='\0';
286 }
287
288 void icq_FileSessionSetFiles(icq_FileSession *p, char **files)
289 {
290 p->files=files;
291 }
292

Mercurial Repository: pidgin3

mercurial