simplify handle_SSH_KNOWNHOSTS
[ocurl.git] / curl-helper.c
blobb2ebba700eba7d9d77b0287f26a8470906c6c8b4
1 /***
2 *** curl-helper.c
3 ***
4 *** Copyright (c) 2003-2008, Lars Nilsson, <lars@quantumchamaeleon.com>
5 *** Copyright (c) 2009, ygrek, <ygrek@autistici.org>
6 ***/
8 #include <stdio.h>
9 #include <string.h>
10 #include <stdlib.h>
11 #include <stdarg.h>
12 #include <caml/config.h>
13 #ifdef HAS_UNISTD
14 #include <unistd.h>
15 #endif
16 /* suppress false gcc warning on seekFunction */
17 #define CURL_DISABLE_TYPECHECK
18 #include <curl/curl.h>
20 #ifndef CAML_NAME_SPACE
21 #define CAML_NAME_SPACE
22 #endif
23 #include <caml/alloc.h>
24 #include <caml/memory.h>
25 #include <caml/mlvalues.h>
26 #include <caml/callback.h>
27 #include <caml/fail.h>
28 #include <caml/unixsupport.h>
29 #include <caml/custom.h>
30 #include <caml/threads.h>
32 #ifndef CAMLdrop
33 #define CAMLdrop caml_local_roots = caml__frame
34 #endif
36 #ifdef HAVE_CONFIG_H
37 #include "config.h"
38 #else
39 #pragma message("No config file given.")
40 #endif
42 #ifdef __cplusplus
43 extern "C" {
44 #endif
46 #define Val_none Val_int(0)
48 static __inline value
49 Val_some( value v )
51 CAMLparam1( v );
52 CAMLlocal1( some );
53 some = caml_alloc(1, 0);
54 Store_field( some, 0, v );
55 CAMLreturn( some );
58 static value Val_pair(value v1, value v2)
60 CAMLparam2(v1,v2);
61 CAMLlocal1(pair);
62 pair = caml_alloc_small(2,0);
63 Field(pair,0) = v1;
64 Field(pair,1) = v2;
65 CAMLreturn(pair);
68 static value Val_cons(value list, value v) { return Val_pair(v,list); }
70 typedef struct Connection Connection;
71 typedef struct ConnectionList ConnectionList;
73 #define Connection_val(v) (*(Connection**)Data_custom_val(v))
75 typedef enum OcamlValues
77 Ocaml_WRITEFUNCTION,
78 Ocaml_READFUNCTION,
79 Ocaml_ERRORBUFFER,
80 Ocaml_POSTFIELDS,
81 Ocaml_HTTPHEADER,
82 Ocaml_HTTPPOST,
83 Ocaml_QUOTE,
84 Ocaml_POSTQUOTE,
85 Ocaml_HEADERFUNCTION,
86 Ocaml_PROGRESSFUNCTION,
87 Ocaml_DEBUGFUNCTION,
88 Ocaml_HTTP200ALIASES,
89 Ocaml_IOCTLFUNCTION,
90 Ocaml_SEEKFUNCTION,
91 Ocaml_OPENSOCKETFUNCTION,
93 Ocaml_URL,
94 Ocaml_PROXY,
95 Ocaml_USERPWD,
96 Ocaml_PROXYUSERPWD,
97 Ocaml_RANGE,
98 Ocaml_REFERER,
99 Ocaml_USERAGENT,
100 Ocaml_FTPPORT,
101 Ocaml_COOKIE,
102 Ocaml_HTTPPOSTSTRINGS,
103 Ocaml_SSLCERT,
104 Ocaml_SSLCERTTYPE,
105 Ocaml_SSLCERTPASSWD,
106 Ocaml_SSLKEY,
107 Ocaml_SSLKEYTYPE,
108 Ocaml_SSLKEYPASSWD,
109 Ocaml_SSLENGINE,
110 Ocaml_COOKIEFILE,
111 Ocaml_CUSTOMREQUEST,
112 Ocaml_INTERFACE,
113 Ocaml_CAINFO,
114 Ocaml_CAPATH,
115 Ocaml_RANDOM_FILE,
116 Ocaml_EGDSOCKET,
117 Ocaml_COOKIEJAR,
118 Ocaml_SSL_CIPHER_LIST,
119 Ocaml_PRIVATE,
120 Ocaml_NETRC_FILE,
121 Ocaml_FTP_ACCOUNT,
122 Ocaml_COOKIELIST,
123 Ocaml_FTP_ALTERNATIVE_TO_USER,
124 Ocaml_SSH_PUBLIC_KEYFILE,
125 Ocaml_SSH_PRIVATE_KEYFILE,
126 Ocaml_SSH_HOST_PUBLIC_KEY_MD5,
127 Ocaml_COPYPOSTFIELDS,
128 Ocaml_DNS_SERVERS,
129 Ocaml_MAIL_FROM,
130 Ocaml_MAIL_RCPT,
131 Ocaml_RESOLVE,
132 Ocaml_USERNAME,
133 Ocaml_PASSWORD,
134 Ocaml_LOGIN_OPTIONS,
135 Ocaml_CONNECT_TO,
136 Ocaml_POSTREDIR,
137 Ocaml_MIMEPOST,
138 Ocaml_SSH_KEYFUNCTION,
140 /* Not used, last for size */
141 OcamlValuesSize
142 } OcamlValue;
144 struct Connection
146 CURL *handle;
148 value ocamlValues;
150 size_t refcount; /* number of references to this structure */
152 char *curl_ERRORBUFFER;
153 char *curl_POSTFIELDS;
154 struct curl_slist *curl_HTTPHEADER;
155 struct curl_slist *httpPostBuffers;
156 struct curl_httppost *httpPostFirst;
157 struct curl_httppost *httpPostLast;
158 struct curl_slist *curl_RESOLVE;
159 struct curl_slist *curl_QUOTE;
160 struct curl_slist *curl_POSTQUOTE;
161 struct curl_slist *curl_HTTP200ALIASES;
162 struct curl_slist *curl_MAIL_RCPT;
163 struct curl_slist *curl_CONNECT_TO;
164 #if HAVE_DECL_CURLOPT_MIMEPOST
165 curl_mime *curl_MIMEPOST;
166 #endif
169 typedef struct CURLErrorMapping CURLErrorMapping;
171 struct CURLErrorMapping
173 char *name;
174 CURLcode error;
177 CURLErrorMapping errorMap[] =
179 {"CURLE_OK", CURLE_OK},
180 #if HAVE_DECL_CURLE_UNSUPPORTED_PROTOCOL
181 {"CURLE_UNSUPPORTED_PROTOCOL", CURLE_UNSUPPORTED_PROTOCOL},
182 #else
183 {"CURLE_UNSUPPORTED_PROTOCOL", -1},
184 #endif
185 #if HAVE_DECL_CURLE_FAILED_INIT
186 {"CURLE_FAILED_INIT", CURLE_FAILED_INIT},
187 #else
188 {"CURLE_FAILED_INIT", -1},
189 #endif
190 #if HAVE_DECL_CURLE_URL_MALFORMAT
191 {"CURLE_URL_MALFORMAT", CURLE_URL_MALFORMAT},
192 #else
193 {"CURLE_URL_MALFORMAT", -1},
194 #endif
195 #if HAVE_DECL_CURLE_URL_MALFORMAT_USER
196 {"CURLE_URL_MALFORMAT_USER", CURLE_URL_MALFORMAT_USER},
197 #else
198 {"CURLE_URL_MALFORMAT_USER", -1},
199 #endif
200 #if HAVE_DECL_CURLE_COULDNT_RESOLVE_PROXY
201 {"CURLE_COULDNT_RESOLVE_PROXY", CURLE_COULDNT_RESOLVE_PROXY},
202 #else
203 {"CURLE_COULDNT_RESOLVE_PROXY", -1},
204 #endif
205 #if HAVE_DECL_CURLE_COULDNT_RESOLVE_HOST
206 {"CURLE_COULDNT_RESOLVE_HOST", CURLE_COULDNT_RESOLVE_HOST},
207 #else
208 {"CURLE_COULDNT_RESOLVE_HOST", -1},
209 #endif
210 #if HAVE_DECL_CURLE_COULDNT_CONNECT
211 {"CURLE_COULDNT_CONNECT", CURLE_COULDNT_CONNECT},
212 #else
213 {"CURLE_COULDNT_CONNECT", -1},
214 #endif
215 #if HAVE_DECL_CURLE_FTP_WEIRD_SERVER_REPLY
216 {"CURLE_FTP_WEIRD_SERVER_REPLY", CURLE_FTP_WEIRD_SERVER_REPLY},
217 #else
218 {"CURLE_FTP_WEIRD_SERVER_REPLY", -1},
219 #endif
220 #if HAVE_DECL_CURLE_FTP_ACCESS_DENIED
221 {"CURLE_FTP_ACCESS_DENIED", CURLE_FTP_ACCESS_DENIED},
222 #else
223 {"CURLE_FTP_ACCESS_DENIED", -1},
224 #endif
225 #if HAVE_DECL_CURLE_FTP_USER_PASSWORD_INCORRECT
226 {"CURLE_FTP_USER_PASSWORD_INCORRECT", CURLE_FTP_USER_PASSWORD_INCORRECT},
227 #else
228 {"CURLE_FTP_USER_PASSWORD_INCORRECT", -1},
229 #endif
230 #if HAVE_DECL_CURLE_FTP_WEIRD_PASS_REPLY
231 {"CURLE_FTP_WEIRD_PASS_REPLY", CURLE_FTP_WEIRD_PASS_REPLY},
232 #else
233 {"CURLE_FTP_WEIRD_PASS_REPLY", -1},
234 #endif
235 #if HAVE_DECL_CURLE_FTP_WEIRD_USER_REPLY
236 {"CURLE_FTP_WEIRD_USER_REPLY", CURLE_FTP_WEIRD_USER_REPLY},
237 #else
238 {"CURLE_FTP_WEIRD_USER_REPLY", -1},
239 #endif
240 #if HAVE_DECL_CURLE_FTP_WEIRD_PASV_REPLY
241 {"CURLE_FTP_WEIRD_PASV_REPLY", CURLE_FTP_WEIRD_PASV_REPLY},
242 #else
243 {"CURLE_FTP_WEIRD_PASV_REPLY", -1},
244 #endif
245 #if HAVE_DECL_CURLE_FTP_WEIRD_227_FORMAT
246 {"CURLE_FTP_WEIRD_227_FORMAT", CURLE_FTP_WEIRD_227_FORMAT},
247 #else
248 {"CURLE_FTP_WEIRD_227_FORMAT", -1},
249 #endif
250 #if HAVE_DECL_CURLE_FTP_CANT_GET_HOST
251 {"CURLE_FTP_CANT_GET_HOST", CURLE_FTP_CANT_GET_HOST},
252 #else
253 {"CURLE_FTP_CANT_GET_HOST", -1},
254 #endif
255 #if HAVE_DECL_CURLE_FTP_CANT_RECONNECT
256 {"CURLE_FTP_CANT_RECONNECT", CURLE_FTP_CANT_RECONNECT},
257 #else
258 {"CURLE_FTP_CANT_RECONNECT", -1},
259 #endif
260 #if HAVE_DECL_CURLE_FTP_COULDNT_SET_BINARY
261 {"CURLE_FTP_COULDNT_SET_BINARY", CURLE_FTP_COULDNT_SET_BINARY},
262 #else
263 {"CURLE_FTP_COULDNT_SET_BINARY", -1},
264 #endif
265 #if HAVE_DECL_CURLE_PARTIAL_FILE
266 {"CURLE_PARTIAL_FILE", CURLE_PARTIAL_FILE},
267 #else
268 {"CURLE_PARTIAL_FILE", -1},
269 #endif
270 #if HAVE_DECL_CURLE_FTP_COULDNT_RETR_FILE
271 {"CURLE_FTP_COULDNT_RETR_FILE", CURLE_FTP_COULDNT_RETR_FILE},
272 #else
273 {"CURLE_FTP_COULDNT_RETR_FILE", -1},
274 #endif
275 #if HAVE_DECL_CURLE_FTP_WRITE_ERROR
276 {"CURLE_FTP_WRITE_ERROR", CURLE_FTP_WRITE_ERROR},
277 #else
278 {"CURLE_FTP_WRITE_ERROR", -1},
279 #endif
280 #if HAVE_DECL_CURLE_FTP_QUOTE_ERROR
281 {"CURLE_FTP_QUOTE_ERROR", CURLE_FTP_QUOTE_ERROR},
282 #else
283 {"CURLE_FTP_QUOTE_ERROR", -1},
284 #endif
285 #if HAVE_DECL_CURLE_HTTP_RETURNED_ERROR
286 {"CURLE_HTTP_RETURNED_ERROR", CURLE_HTTP_RETURNED_ERROR},
287 #else
288 {"CURLE_HTTP_RETURNED_ERROR", -1},
289 #endif
290 #if HAVE_DECL_CURLE_WRITE_ERROR
291 {"CURLE_WRITE_ERROR", CURLE_WRITE_ERROR},
292 #else
293 {"CURLE_WRITE_ERROR", -1},
294 #endif
295 #if HAVE_DECL_CURLE_MALFORMAT_USER
296 {"CURLE_MALFORMAT_USER", CURLE_MALFORMAT_USER},
297 #else
298 {"CURLE_MALFORMAT_USER", -1},
299 #endif
300 #if HAVE_DECL_CURLE_FTP_COULDNT_STOR_FILE
301 {"CURLE_FTP_COULDNT_STOR_FILE", CURLE_FTP_COULDNT_STOR_FILE},
302 #else
303 {"CURLE_FTP_COULDNT_STOR_FILE", -1},
304 #endif
305 #if HAVE_DECL_CURLE_READ_ERROR
306 {"CURLE_READ_ERROR", CURLE_READ_ERROR},
307 #else
308 {"CURLE_READ_ERROR", -1},
309 #endif
310 #if HAVE_DECL_CURLE_OUT_OF_MEMORY
311 {"CURLE_OUT_OF_MEMORY", CURLE_OUT_OF_MEMORY},
312 #else
313 {"CURLE_OUT_OF_MEMORY", -1},
314 #endif
315 #if HAVE_DECL_CURLE_OPERATION_TIMEOUTED
316 {"CURLE_OPERATION_TIMEOUTED", CURLE_OPERATION_TIMEOUTED},
317 #else
318 {"CURLE_OPERATION_TIMEOUTED", -1},
319 #endif
320 #if HAVE_DECL_CURLE_FTP_COULDNT_SET_ASCII
321 {"CURLE_FTP_COULDNT_SET_ASCII", CURLE_FTP_COULDNT_SET_ASCII},
322 #else
323 {"CURLE_FTP_COULDNT_SET_ASCII", -1},
324 #endif
325 #if HAVE_DECL_CURLE_FTP_PORT_FAILED
326 {"CURLE_FTP_PORT_FAILED", CURLE_FTP_PORT_FAILED},
327 #else
328 {"CURLE_FTP_PORT_FAILED", -1},
329 #endif
330 #if HAVE_DECL_CURLE_FTP_COULDNT_USE_REST
331 {"CURLE_FTP_COULDNT_USE_REST", CURLE_FTP_COULDNT_USE_REST},
332 #else
333 {"CURLE_FTP_COULDNT_USE_REST", -1},
334 #endif
335 #if HAVE_DECL_CURLE_FTP_COULDNT_GET_SIZE
336 {"CURLE_FTP_COULDNT_GET_SIZE", CURLE_FTP_COULDNT_GET_SIZE},
337 #else
338 {"CURLE_FTP_COULDNT_GET_SIZE", -1},
339 #endif
340 #if HAVE_DECL_CURLE_HTTP_RANGE_ERROR
341 {"CURLE_HTTP_RANGE_ERROR", CURLE_HTTP_RANGE_ERROR},
342 #else
343 {"CURLE_HTTP_RANGE_ERROR", -1},
344 #endif
345 #if HAVE_DECL_CURLE_HTTP_POST_ERROR
346 {"CURLE_HTTP_POST_ERROR", CURLE_HTTP_POST_ERROR},
347 #else
348 {"CURLE_HTTP_POST_ERROR", -1},
349 #endif
350 #if HAVE_DECL_CURLE_SSL_CONNECT_ERROR
351 {"CURLE_SSL_CONNECT_ERROR", CURLE_SSL_CONNECT_ERROR},
352 #else
353 {"CURLE_SSL_CONNECT_ERROR", -1},
354 #endif
355 #if HAVE_DECL_CURLE_BAD_DOWNLOAD_RESUME
356 {"CURLE_BAD_DOWNLOAD_RESUME", CURLE_BAD_DOWNLOAD_RESUME},
357 #else
358 {"CURLE_BAD_DOWNLOAD_RESUME", -1},
359 #endif
360 #if HAVE_DECL_CURLE_FILE_COULDNT_READ_FILE
361 {"CURLE_FILE_COULDNT_READ_FILE", CURLE_FILE_COULDNT_READ_FILE},
362 #else
363 {"CURLE_FILE_COULDNT_READ_FILE", -1},
364 #endif
365 #if HAVE_DECL_CURLE_LDAP_CANNOT_BIND
366 {"CURLE_LDAP_CANNOT_BIND", CURLE_LDAP_CANNOT_BIND},
367 #else
368 {"CURLE_LDAP_CANNOT_BIND", -1},
369 #endif
370 #if HAVE_DECL_CURLE_LDAP_SEARCH_FAILED
371 {"CURLE_LDAP_SEARCH_FAILED", CURLE_LDAP_SEARCH_FAILED},
372 #else
373 {"CURLE_LDAP_SEARCH_FAILED", -1},
374 #endif
375 #if HAVE_DECL_CURLE_LIBRARY_NOT_FOUND
376 {"CURLE_LIBRARY_NOT_FOUND", CURLE_LIBRARY_NOT_FOUND},
377 #else
378 {"CURLE_LIBRARY_NOT_FOUND", -1},
379 #endif
380 #if HAVE_DECL_CURLE_FUNCTION_NOT_FOUND
381 {"CURLE_FUNCTION_NOT_FOUND", CURLE_FUNCTION_NOT_FOUND},
382 #else
383 {"CURLE_FUNCTION_NOT_FOUND", -1},
384 #endif
385 #if HAVE_DECL_CURLE_ABORTED_BY_CALLBACK
386 {"CURLE_ABORTED_BY_CALLBACK", CURLE_ABORTED_BY_CALLBACK},
387 #else
388 {"CURLE_ABORTED_BY_CALLBACK", -1},
389 #endif
390 #if HAVE_DECL_CURLE_BAD_FUNCTION_ARGUMENT
391 {"CURLE_BAD_FUNCTION_ARGUMENT", CURLE_BAD_FUNCTION_ARGUMENT},
392 #else
393 {"CURLE_BAD_FUNCTION_ARGUMENT", -1},
394 #endif
395 #if HAVE_DECL_CURLE_BAD_CALLING_ORDER
396 {"CURLE_BAD_CALLING_ORDER", CURLE_BAD_CALLING_ORDER},
397 #else
398 {"CURLE_BAD_CALLING_ORDER", -1},
399 #endif
400 #if HAVE_DECL_CURLE_INTERFACE_FAILED
401 {"CURLE_INTERFACE_FAILED", CURLE_INTERFACE_FAILED},
402 #else
403 {"CURLE_INTERFACE_FAILED", -1},
404 #endif
405 #if HAVE_DECL_CURLE_BAD_PASSWORD_ENTERED
406 {"CURLE_BAD_PASSWORD_ENTERED", CURLE_BAD_PASSWORD_ENTERED},
407 #else
408 {"CURLE_BAD_PASSWORD_ENTERED", -1},
409 #endif
410 #if HAVE_DECL_CURLE_TOO_MANY_REDIRECTS
411 {"CURLE_TOO_MANY_REDIRECTS", CURLE_TOO_MANY_REDIRECTS},
412 #else
413 {"CURLE_TOO_MANY_REDIRECTS", -1},
414 #endif
415 #if HAVE_DECL_CURLE_UNKNOWN_TELNET_OPTION
416 {"CURLE_UNKNOWN_TELNET_OPTION", CURLE_UNKNOWN_TELNET_OPTION},
417 #else
418 {"CURLE_UNKNOWN_TELNET_OPTION", -1},
419 #endif
420 #if HAVE_DECL_CURLE_TELNET_OPTION_SYNTAX
421 {"CURLE_TELNET_OPTION_SYNTAX", CURLE_TELNET_OPTION_SYNTAX},
422 #else
423 {"CURLE_TELNET_OPTION_SYNTAX", -1},
424 #endif
425 #if HAVE_DECL_CURLE_SSL_PEER_CERTIFICATE
426 {"CURLE_SSL_PEER_CERTIFICATE", CURLE_SSL_PEER_CERTIFICATE},
427 #else
428 {"CURLE_SSL_PEER_CERTIFICATE", -1},
429 #endif
430 #if HAVE_DECL_CURLE_GOT_NOTHING
431 {"CURLE_GOT_NOTHING", CURLE_GOT_NOTHING},
432 #else
433 {"CURLE_GOT_NOTHING", -1},
434 #endif
435 #if HAVE_DECL_CURLE_SSL_ENGINE_NOTFOUND
436 {"CURLE_SSL_ENGINE_NOTFOUND", CURLE_SSL_ENGINE_NOTFOUND},
437 #else
438 {"CURLE_SSL_ENGINE_NOTFOUND", -1},
439 #endif
440 #if HAVE_DECL_CURLE_SSL_ENGINE_SETFAILED
441 {"CURLE_SSL_ENGINE_SETFAILED", CURLE_SSL_ENGINE_SETFAILED},
442 #else
443 {"CURLE_SSL_ENGINE_SETFAILED", -1},
444 #endif
445 #if HAVE_DECL_CURLE_SEND_ERROR
446 {"CURLE_SEND_ERROR", CURLE_SEND_ERROR},
447 #else
448 {"CURLE_SEND_ERROR", -1},
449 #endif
450 #if HAVE_DECL_CURLE_RECV_ERROR
451 {"CURLE_RECV_ERROR", CURLE_RECV_ERROR},
452 #else
453 {"CURLE_RECV_ERROR", -1},
454 #endif
455 #if HAVE_DECL_CURLE_SHARE_IN_USE
456 {"CURLE_SHARE_IN_USE", CURLE_SHARE_IN_USE},
457 #else
458 {"CURLE_SHARE_IN_USE", -1},
459 #endif
460 #if HAVE_DECL_CURLE_SSL_CERTPROBLEM
461 {"CURLE_SSL_CERTPROBLEM", CURLE_SSL_CERTPROBLEM},
462 #else
463 {"CURLE_SSL_CERTPROBLEM", -1},
464 #endif
465 #if HAVE_DECL_CURLE_SSL_CIPHER
466 {"CURLE_SSL_CIPHER", CURLE_SSL_CIPHER},
467 #else
468 {"CURLE_SSL_CIPHER", -1},
469 #endif
470 #if HAVE_DECL_CURLE_SSL_CACERT
471 {"CURLE_SSL_CACERT", CURLE_SSL_CACERT},
472 #else
473 {"CURLE_SSL_CACERT", -1},
474 #endif
475 #if HAVE_DECL_CURLE_BAD_CONTENT_ENCODING
476 {"CURLE_BAD_CONTENT_ENCODING", CURLE_BAD_CONTENT_ENCODING},
477 #else
478 {"CURLE_BAD_CONTENT_ENCODING", -1},
479 #endif
480 #if HAVE_DECL_CURLE_LDAP_INVALID_URL
481 {"CURLE_LDAP_INVALID_URL", CURLE_LDAP_INVALID_URL},
482 #else
483 {"CURLE_LDAP_INVALID_URL", -1},
484 #endif
485 #if HAVE_DECL_CURLE_FILESIZE_EXCEEDED
486 {"CURLE_FILESIZE_EXCEEDED", CURLE_FILESIZE_EXCEEDED},
487 #else
488 {"CURLE_FILESIZE_EXCEEDED", -1},
489 #endif
490 #if HAVE_DECL_CURLE_FTP_SSL_FAILED
491 {"CURLE_FTP_SSL_FAILED", CURLE_FTP_SSL_FAILED},
492 #else
493 {"CURLE_FTP_SSL_FAILED", -1},
494 #endif
495 #if HAVE_DECL_CURLE_SEND_FAIL_REWIND
496 {"CURLE_SEND_FAIL_REWIND", CURLE_SEND_FAIL_REWIND},
497 #else
498 {"CURLE_SEND_FAIL_REWIND", -1},
499 #endif
500 #if HAVE_DECL_CURLE_SSL_ENGINE_INITFAILED
501 {"CURLE_SSL_ENGINE_INITFAILED", CURLE_SSL_ENGINE_INITFAILED},
502 #else
503 {"CURLE_SSL_ENGINE_INITFAILED", -1},
504 #endif
505 #if HAVE_DECL_CURLE_LOGIN_DENIED
506 {"CURLE_LOGIN_DENIED", CURLE_LOGIN_DENIED},
507 #else
508 {"CURLE_LOGIN_DENIED", -1},
509 #endif
510 #if HAVE_DECL_CURLE_TFTP_NOTFOUND
511 {"CURLE_TFTP_NOTFOUND", CURLE_TFTP_NOTFOUND},
512 #else
513 {"CURLE_TFTP_NOTFOUND", -1},
514 #endif
515 #if HAVE_DECL_CURLE_TFTP_PERM
516 {"CURLE_TFTP_PERM", CURLE_TFTP_PERM},
517 #else
518 {"CURLE_TFTP_PERM", -1},
519 #endif
520 #if HAVE_DECL_CURLE_REMOTE_DISK_FULL
521 {"CURLE_REMOTE_DISK_FULL", CURLE_REMOTE_DISK_FULL},
522 #else
523 {"CURLE_REMOTE_DISK_FULL", -1},
524 #endif
525 #if HAVE_DECL_CURLE_TFTP_ILLEGAL
526 {"CURLE_TFTP_ILLEGAL", CURLE_TFTP_ILLEGAL},
527 #else
528 {"CURLE_TFTP_ILLEGAL", -1},
529 #endif
530 #if HAVE_DECL_CURLE_TFTP_UNKNOWNID
531 {"CURLE_TFTP_UNKNOWNID", CURLE_TFTP_UNKNOWNID},
532 #else
533 {"CURLE_TFTP_UNKNOWNID", -1},
534 #endif
535 #if HAVE_DECL_CURLE_REMOTE_FILE_EXISTS
536 {"CURLE_REMOTE_FILE_EXISTS", CURLE_REMOTE_FILE_EXISTS},
537 #else
538 {"CURLE_REMOTE_FILE_EXISTS", -1},
539 #endif
540 #if HAVE_DECL_CURLE_TFTP_NOSUCHUSER
541 {"CURLE_TFTP_NOSUCHUSER", CURLE_TFTP_NOSUCHUSER},
542 #else
543 {"CURLE_TFTP_NOSUCHUSER", -1},
544 #endif
545 #if HAVE_DECL_CURLE_CONV_FAILED
546 {"CURLE_CONV_FAILED", CURLE_CONV_FAILED},
547 #else
548 {"CURLE_CONV_FAILED", -1},
549 #endif
550 #if HAVE_DECL_CURLE_CONV_REQD
551 {"CURLE_CONV_REQD", CURLE_CONV_REQD},
552 #else
553 {"CURLE_CONV_REQD", -1},
554 #endif
555 #if HAVE_DECL_CURLE_SSL_CACERT_BADFILE
556 {"CURLE_SSL_CACERT_BADFILE", CURLE_SSL_CACERT_BADFILE},
557 #else
558 {"CURLE_SSL_CACERT_BADFILE", -1},
559 #endif
560 #if HAVE_DECL_CURLE_REMOTE_FILE_NOT_FOUND
561 {"CURLE_REMOTE_FILE_NOT_FOUND", CURLE_REMOTE_FILE_NOT_FOUND},
562 #else
563 {"CURLE_REMOTE_FILE_NOT_FOUND", -1},
564 #endif
565 #if HAVE_DECL_CURLE_SSH
566 {"CURLE_SSH", CURLE_SSH},
567 #else
568 {"CURLE_SSH", -1},
569 #endif
570 #if HAVE_DECL_CURLE_SSL_SHUTDOWN_FAILED
571 {"CURLE_SSL_SHUTDOWN_FAILED", CURLE_SSL_SHUTDOWN_FAILED},
572 #else
573 {"CURLE_SSL_SHUTDOWN_FAILED", -1},
574 #endif
575 #if HAVE_DECL_CURLE_AGAIN
576 {"CURLE_AGAIN", CURLE_AGAIN},
577 #else
578 {"CURLE_AGAIN", -1},
579 #endif
580 {NULL, (CURLcode)0}
583 typedef struct CURLOptionMapping CURLOptionMapping;
585 struct CURLOptionMapping
587 void (*optionHandler)(Connection *, value);
588 char *name;
589 OcamlValue ocamlValue;
592 static char* strdup_ml(value v)
594 char* p = NULL;
595 p = (char*)malloc(caml_string_length(v)+1);
596 memcpy(p,String_val(v),caml_string_length(v)+1); // caml strings have terminating zero
597 return p;
600 static value ml_copy_string(char const* p, size_t size)
602 value v = caml_alloc_string(size);
603 memcpy(&Byte(v,0),p,size);
604 return v;
607 /* prepends to the beginning of list */
608 static struct curl_slist* curl_slist_prepend_ml(struct curl_slist* list, value v)
610 /* FIXME check NULLs */
611 struct curl_slist* new_item = (struct curl_slist*)malloc(sizeof(struct curl_slist));
613 new_item->next = list;
614 new_item->data = strdup_ml(v);
616 return new_item;
619 static void free_curl_slist(struct curl_slist *slist)
621 if (NULL == slist)
622 return;
624 curl_slist_free_all(slist);
627 static void raiseError(Connection *conn, CURLcode code)
629 CAMLparam0();
630 CAMLlocal1(exceptionData);
631 value *exception;
632 char *errorString = "Unknown Error";
633 int i;
635 for (i = 0; errorMap[i].name != NULL; i++)
637 if (errorMap[i].error == code)
639 errorString = errorMap[i].name;
640 break;
644 exceptionData = caml_alloc_tuple(3);
646 Store_field(exceptionData, 0, Val_int(code));
647 Store_field(exceptionData, 1, Val_int(code));
648 Store_field(exceptionData, 2, caml_copy_string(errorString));
650 if (conn != NULL && conn->curl_ERRORBUFFER != NULL)
652 Store_field(Field(conn->ocamlValues, Ocaml_ERRORBUFFER), 0, caml_copy_string(conn->curl_ERRORBUFFER));
655 exception = caml_named_value("CurlException");
657 if (exception == NULL)
658 caml_failwith("CurlException not registered");
660 caml_raise_with_arg(*exception, exceptionData);
662 CAMLreturn0;
665 static void resetOcamlValues(Connection* connection)
667 int i;
669 for (i = 0; i < OcamlValuesSize; i++)
670 Store_field(connection->ocamlValues, i, Val_unit);
673 static Connection* allocConnection(CURL* h)
675 Connection* connection = (Connection *)malloc(sizeof(Connection));
677 connection->ocamlValues = caml_alloc(OcamlValuesSize, 0);
678 resetOcamlValues(connection);
679 caml_register_global_root(&connection->ocamlValues);
681 connection->handle = h;
682 curl_easy_setopt(h, CURLOPT_PRIVATE, connection);
684 connection->refcount = 0;
686 connection->curl_ERRORBUFFER = NULL;
687 connection->curl_POSTFIELDS = NULL;
688 connection->curl_HTTPHEADER = NULL;
689 connection->httpPostBuffers = NULL;
690 connection->httpPostFirst = NULL;
691 connection->httpPostLast = NULL;
692 connection->curl_QUOTE = NULL;
693 connection->curl_POSTQUOTE = NULL;
694 connection->curl_HTTP200ALIASES = NULL;
695 connection->curl_RESOLVE = NULL;
696 connection->curl_MAIL_RCPT = NULL;
697 connection->curl_CONNECT_TO = NULL;
698 #if HAVE_DECL_CURLOPT_MIMEPOST
699 connection->curl_MIMEPOST = NULL;
700 #endif
702 return connection;
705 static Connection *newConnection(void)
707 CURL* h;
709 caml_enter_blocking_section();
710 h = curl_easy_init();
711 caml_leave_blocking_section();
713 return allocConnection(h);
716 static void free_if(void* p) { if (NULL != p) free(p); }
718 static void removeConnection(Connection *connection, int finalization)
720 const char* fin_url = NULL;
722 if (!connection->handle)
724 return; /* already cleaned up */
727 if (finalization)
729 /* cannot engage OCaml runtime at finalization, just report leak */
730 if (CURLE_OK != curl_easy_getinfo(connection->handle, CURLINFO_EFFECTIVE_URL, &fin_url) || NULL == fin_url)
732 fin_url = "unknown";
734 fprintf(stderr,"Curl: handle %p leaked, conn %p, url %s\n", connection->handle, connection, fin_url);
735 fflush(stderr);
737 else
739 caml_enter_blocking_section();
740 curl_easy_cleanup(connection->handle);
741 caml_leave_blocking_section();
744 connection->handle = NULL;
746 caml_remove_global_root(&connection->ocamlValues);
748 free_if(connection->curl_ERRORBUFFER);
749 free_if(connection->curl_POSTFIELDS);
750 free_curl_slist(connection->curl_HTTPHEADER);
751 free_curl_slist(connection->httpPostBuffers);
752 if (connection->httpPostFirst != NULL)
753 curl_formfree(connection->httpPostFirst);
754 free_curl_slist(connection->curl_RESOLVE);
755 free_curl_slist(connection->curl_QUOTE);
756 free_curl_slist(connection->curl_POSTQUOTE);
757 free_curl_slist(connection->curl_HTTP200ALIASES);
758 free_curl_slist(connection->curl_MAIL_RCPT);
759 free_curl_slist(connection->curl_CONNECT_TO);
760 #if HAVE_DECL_CURLOPT_MIMEPOST
761 curl_mime_free(connection->curl_MIMEPOST);
762 #endif
765 static Connection* getConnection(CURL* h)
767 Connection* p = NULL;
769 if (CURLE_OK != curl_easy_getinfo(h, CURLINFO_PRIVATE, &p) || NULL == p)
771 caml_failwith("Unknown handle");
774 return p;
777 #if 1
778 static void checkConnection(Connection * connection)
780 (void)connection;
782 #else
783 static void checkConnection(Connection *connection)
785 if (connection != getConnection(connection->handle))
787 caml_failwith("Invalid Connection");
790 #endif
792 void op_curl_easy_finalize(value v)
794 Connection* conn = Connection_val(v);
795 /* same connection may be referenced by several different
796 OCaml values, see e.g. caml_curl_multi_remove_finished */
797 conn->refcount--;
798 if (0 == conn->refcount)
800 removeConnection(conn, 1);
801 free(conn);
805 int op_curl_easy_compare(value v1, value v2)
807 size_t p1 = (size_t)Connection_val(v1);
808 size_t p2 = (size_t)Connection_val(v2);
809 return (p1 == p2 ? 0 : (p1 > p2 ? 1 : -1)); /* compare addresses */
812 intnat op_curl_easy_hash(value v)
814 return (size_t)Connection_val(v); /* address */
817 static struct custom_operations curl_easy_ops = {
818 "ygrek.curl_easy",
819 op_curl_easy_finalize,
820 op_curl_easy_compare,
821 op_curl_easy_hash,
822 custom_serialize_default,
823 custom_deserialize_default,
824 #if defined(custom_compare_ext_default)
825 custom_compare_ext_default,
826 #endif
829 value caml_curl_alloc(Connection* conn)
831 value v = caml_alloc_custom(&curl_easy_ops, sizeof(Connection*), 0, 1);
832 Connection_val(v) = conn;
833 conn->refcount++;
834 return v;
837 static size_t cb_WRITEFUNCTION(char *ptr, size_t size, size_t nmemb, void *data)
839 caml_leave_blocking_section();
841 CAMLparam0();
842 CAMLlocal2(result, str);
843 Connection *conn = (Connection *)data;
845 checkConnection(conn);
847 str = ml_copy_string(ptr,size*nmemb);
849 result = caml_callback_exn(Field(conn->ocamlValues, Ocaml_WRITEFUNCTION), str);
851 size_t r = Is_exception_result(result) ? 0 : Int_val(result);
852 CAMLdrop;
854 caml_enter_blocking_section();
855 return r;
858 static size_t cb_READFUNCTION(void *ptr, size_t size, size_t nmemb, void *data)
860 caml_leave_blocking_section();
862 CAMLparam0();
863 CAMLlocal1(result);
864 Connection *conn = (Connection *)data;
865 size_t length;
867 checkConnection(conn);
869 result = caml_callback_exn(Field(conn->ocamlValues, Ocaml_READFUNCTION),
870 Val_int(size*nmemb));
872 if (Is_exception_result(result))
874 CAMLreturnT(size_t,CURL_READFUNC_ABORT);
877 length = caml_string_length(result);
879 size_t r;
881 if (length <= size*nmemb)
883 memcpy(ptr, String_val(result), length);
884 r = length;
886 else
888 r = CURL_READFUNC_ABORT;
890 CAMLdrop;
892 caml_enter_blocking_section();
893 return r;
896 static size_t cb_HEADERFUNCTION(char *ptr, size_t size, size_t nmemb, void *data)
898 caml_leave_blocking_section();
900 CAMLparam0();
901 CAMLlocal2(result,str);
902 Connection *conn = (Connection *)data;
904 checkConnection(conn);
906 str = ml_copy_string(ptr,size*nmemb);
908 result = caml_callback_exn(Field(conn->ocamlValues, Ocaml_HEADERFUNCTION), str);
910 size_t r = Is_exception_result(result) ? 0 : Int_val(result);
911 CAMLdrop;
913 caml_enter_blocking_section();
914 return r;
917 static int cb_PROGRESSFUNCTION(void *data,
918 double dlTotal,
919 double dlNow,
920 double ulTotal,
921 double ulNow)
923 caml_leave_blocking_section();
925 CAMLparam0();
926 CAMLlocal1(result);
927 CAMLlocalN(callbackData, 4);
928 Connection *conn = (Connection *)data;
930 checkConnection(conn);
932 callbackData[0] = caml_copy_double(dlTotal);
933 callbackData[1] = caml_copy_double(dlNow);
934 callbackData[2] = caml_copy_double(ulTotal);
935 callbackData[3] = caml_copy_double(ulNow);
937 result = caml_callbackN_exn(Field(conn->ocamlValues, Ocaml_PROGRESSFUNCTION),
938 4, callbackData);
940 int r = Is_exception_result(result) ? 1 : Bool_val(result);
941 CAMLdrop;
943 caml_enter_blocking_section();
944 return r;
947 static int cb_DEBUGFUNCTION(CURL *debugConnection,
948 curl_infotype infoType,
949 char *buffer,
950 size_t bufferLength,
951 void *data)
953 caml_leave_blocking_section();
955 CAMLparam0();
956 CAMLlocal3(camlDebugConnection, camlInfoType, camlMessage);
957 Connection *conn = (Connection *)data;
958 (void)debugConnection; /* not used */
960 checkConnection(conn);
962 camlDebugConnection = caml_curl_alloc(conn);
963 camlMessage = ml_copy_string(buffer,bufferLength);
964 camlInfoType = Val_long(infoType <= CURLINFO_SSL_DATA_OUT ? infoType : CURLINFO_END);
966 caml_callback3_exn(Field(conn->ocamlValues, Ocaml_DEBUGFUNCTION),
967 camlDebugConnection,
968 camlInfoType,
969 camlMessage);
971 CAMLdrop;
973 caml_enter_blocking_section();
974 return 0;
977 static curlioerr cb_IOCTLFUNCTION(CURL *ioctl,
978 int cmd,
979 void *data)
981 caml_leave_blocking_section();
983 CAMLparam0();
984 CAMLlocal3(camlResult, camlConnection, camlCmd);
985 Connection *conn = (Connection *)data;
986 curlioerr result = CURLIOE_OK;
987 (void)ioctl; /* not used */
989 checkConnection(conn);
991 if (cmd == CURLIOCMD_NOP)
992 camlCmd = Val_long(0);
993 else if (cmd == CURLIOCMD_RESTARTREAD)
994 camlCmd = Val_long(1);
995 else
996 caml_failwith("Invalid IOCTL Cmd!");
998 camlConnection = caml_curl_alloc(conn);
1000 camlResult = caml_callback2_exn(Field(conn->ocamlValues, Ocaml_IOCTLFUNCTION),
1001 camlConnection,
1002 camlCmd);
1004 if (Is_exception_result(camlResult))
1006 result = CURLIOE_FAILRESTART;
1008 else
1009 switch (Long_val(camlResult))
1011 case 0: /* CURLIOE_OK */
1012 result = CURLIOE_OK;
1013 break;
1015 case 1: /* CURLIOE_UNKNOWNCMD */
1016 result = CURLIOE_UNKNOWNCMD;
1017 break;
1019 case 2: /* CURLIOE_FAILRESTART */
1020 result = CURLIOE_FAILRESTART;
1021 break;
1023 default: /* Incorrect return value, but let's handle it */
1024 result = CURLIOE_FAILRESTART;
1025 break;
1027 CAMLdrop;
1029 caml_enter_blocking_section();
1030 return result;
1033 #if HAVE_DECL_CURLOPT_SEEKFUNCTION
1034 static int cb_SEEKFUNCTION(void *data,
1035 curl_off_t offset,
1036 int origin)
1038 caml_leave_blocking_section();
1040 CAMLparam0();
1041 CAMLlocal3(camlResult, camlOffset, camlOrigin);
1042 Connection *conn = (Connection *)data;
1044 camlOffset = caml_copy_int64(offset);
1046 if (origin == SEEK_SET)
1047 camlOrigin = Val_long(0);
1048 else if (origin == SEEK_CUR)
1049 camlOrigin = Val_long(1);
1050 else if (origin == SEEK_END)
1051 camlOrigin = Val_long(2);
1052 else
1053 caml_failwith("Invalid seek code");
1055 camlResult = caml_callback2_exn(Field(conn->ocamlValues,
1056 Ocaml_SEEKFUNCTION),
1057 camlOffset,
1058 camlOrigin);
1060 int result;
1061 if (Is_exception_result(camlResult))
1062 result = CURL_SEEKFUNC_FAIL;
1063 else
1064 switch (Int_val(camlResult))
1066 case 0: result = CURL_SEEKFUNC_OK; break;
1067 case 1: result = CURL_SEEKFUNC_FAIL; break;
1068 case 2: result = CURL_SEEKFUNC_CANTSEEK; break;
1069 default: caml_failwith("Invalid seek result");
1071 CAMLdrop;
1073 caml_enter_blocking_section();
1074 return result;
1076 #endif
1078 #if HAVE_DECL_CURLOPT_OPENSOCKETFUNCTION
1079 static int cb_OPENSOCKETFUNCTION(void *data,
1080 curlsocktype purpose,
1081 struct curl_sockaddr *addr)
1083 caml_leave_blocking_section();
1085 CAMLparam0();
1086 CAMLlocal1(result);
1087 Connection *conn = (Connection *)data;
1088 int sock = -1;
1089 (void)purpose; /* not used */
1091 sock = socket(addr->family, addr->socktype, addr->protocol);
1093 if (-1 != sock)
1095 /* FIXME windows */
1096 result = caml_callback_exn(Field(conn->ocamlValues, Ocaml_OPENSOCKETFUNCTION), Val_int(sock));
1097 if (Is_exception_result(result))
1099 close(sock);
1100 sock = -1;
1103 CAMLdrop;
1105 caml_enter_blocking_section();
1106 return ((sock == -1) ? CURL_SOCKET_BAD : sock);
1108 #endif
1110 static int cb_SSH_KEYFUNCTION(CURL *easy,
1111 const struct curl_khkey *knownkey,
1112 const struct curl_khkey *foundkey,
1113 enum curl_khmatch match,
1114 void *clientp)
1116 caml_leave_blocking_section();
1118 CAMLparam0();
1119 CAMLlocal4(known, found, mismatch, result);
1120 Connection *conn = (Connection *)clientp;
1121 int res = CURLKHSTAT_REJECT;
1123 switch (match) {
1124 case CURLKHMATCH_OK:
1125 found = ml_copy_string(foundkey->key, foundkey->len ? foundkey->len : strlen(foundkey->key));
1126 result = caml_callback2_exn(Field(conn->ocamlValues, Ocaml_SSH_KEYFUNCTION), Val_int(0), found);
1127 break;
1128 case CURLKHMATCH_MISMATCH:
1129 found = ml_copy_string(foundkey->key, foundkey->len ? foundkey->len : strlen(foundkey->key));
1130 known = ml_copy_string(knownkey->key, knownkey->len ? knownkey->len : strlen(knownkey->key));
1131 mismatch = caml_alloc_small(1, 0);
1132 Field(mismatch, 0) = found;
1133 result = caml_callback2_exn(Field(conn->ocamlValues, Ocaml_SSH_KEYFUNCTION), mismatch, found);
1134 break;
1135 case CURLKHMATCH_MISSING:
1136 found = ml_copy_string(foundkey->key, foundkey->len ? foundkey->len : strlen(foundkey->key));
1137 result = caml_callback2_exn(Field(conn->ocamlValues, Ocaml_SSH_KEYFUNCTION), Val_int(1), found);
1138 break;
1139 default:
1140 caml_failwith("Invalid CURL_SSH_KEYFUNCTION argument");
1141 break;
1144 if (!Is_exception_result(result)) {
1145 switch (Int_val(result)) {
1146 case 0:
1147 res = CURLKHSTAT_FINE_ADD_TO_FILE;
1148 break;
1149 case 1:
1150 res = CURLKHSTAT_FINE;
1151 break;
1152 case 2:
1153 res = CURLKHSTAT_REJECT;
1154 break;
1155 case 3:
1156 res = CURLKHSTAT_DEFER;
1157 break;
1158 default:
1159 caml_failwith("Invalid CURLOPT_SSH_KEYFUNCTION return value");
1160 break;
1164 CAMLdrop;
1166 caml_enter_blocking_section();
1167 return res;
1171 ** curl_global_init helper function
1174 value caml_curl_global_init(value initOption)
1176 CAMLparam1(initOption);
1178 switch (Long_val(initOption))
1180 case 0: /* CURLINIT_GLOBALALL */
1181 CAMLreturn(Val_long(curl_global_init(CURL_GLOBAL_ALL)));
1182 break;
1184 case 1: /* CURLINIT_GLOBALSSL */
1185 CAMLreturn(Val_long(curl_global_init(CURL_GLOBAL_SSL)));
1186 break;
1188 case 2: /* CURLINIT_GLOBALWIN32 */
1189 CAMLreturn(Val_long(curl_global_init(CURL_GLOBAL_WIN32)));
1190 break;
1192 case 3: /* CURLINIT_GLOBALNOTHING */
1193 CAMLreturn(Val_long(curl_global_init(CURL_GLOBAL_NOTHING)));
1194 break;
1196 default:
1197 caml_failwith("Invalid Initialization Option");
1198 break;
1201 /* Keep compiler happy, we should never get here due to caml_failwith() */
1202 CAMLreturn(Val_unit);
1206 ** curl_global_cleanup helper function
1209 value caml_curl_global_cleanup(void)
1211 CAMLparam0();
1213 curl_global_cleanup();
1215 CAMLreturn(Val_unit);
1219 ** curl_easy_init helper function
1221 value caml_curl_easy_init(void)
1223 CAMLparam0();
1224 CAMLlocal1(result);
1226 result = caml_curl_alloc(newConnection());
1228 CAMLreturn(result);
1231 value caml_curl_easy_reset(value conn)
1233 CAMLparam1(conn);
1234 Connection *connection = Connection_val(conn);
1236 checkConnection(connection);
1237 curl_easy_reset(connection->handle);
1238 curl_easy_setopt(connection->handle, CURLOPT_PRIVATE, connection);
1239 resetOcamlValues(connection);
1241 CAMLreturn(Val_unit);
1245 ** curl_easy_setopt helper utility functions
1248 #define SETOPT_FUNCTION(name) \
1249 static void handle_##name##FUNCTION(Connection *conn, value option) \
1251 CAMLparam1(option); \
1252 CURLcode result = CURLE_OK; \
1253 Store_field(conn->ocamlValues, Ocaml_##name##FUNCTION, option); \
1254 result = curl_easy_setopt(conn->handle, CURLOPT_##name##FUNCTION, cb_##name##FUNCTION); \
1255 if (result != CURLE_OK) raiseError(conn, result); \
1256 result = curl_easy_setopt(conn->handle, CURLOPT_##name##DATA, conn); \
1257 if (result != CURLE_OK) raiseError(conn, result); \
1258 CAMLreturn0; \
1261 SETOPT_FUNCTION( WRITE)
1262 SETOPT_FUNCTION( READ)
1263 SETOPT_FUNCTION( HEADER)
1264 SETOPT_FUNCTION( PROGRESS)
1265 SETOPT_FUNCTION( DEBUG)
1266 SETOPT_FUNCTION( SSH_KEY)
1268 #if HAVE_DECL_CURLOPT_SEEKFUNCTION
1269 SETOPT_FUNCTION( SEEK)
1270 #endif
1272 #if HAVE_DECL_CURLOPT_IOCTLFUNCTION
1273 SETOPT_FUNCTION( IOCTL)
1274 #endif
1276 #if HAVE_DECL_CURLOPT_OPENSOCKETFUNCTION
1277 SETOPT_FUNCTION( OPENSOCKET)
1278 #endif
1280 static void handle_slist(Connection *conn, struct curl_slist** slist, OcamlValue caml_option, CURLoption curl_option, value option)
1282 CAMLparam1(option);
1283 CURLcode result = CURLE_OK;
1285 Store_field(conn->ocamlValues, caml_option, option);
1287 free_curl_slist(*slist);
1288 *slist = NULL;
1290 while (Val_emptylist != option)
1292 *slist = curl_slist_append(*slist, String_val(Field(option, 0)));
1294 option = Field(option, 1);
1297 result = curl_easy_setopt(conn->handle, curl_option, *slist);
1299 if (result != CURLE_OK)
1300 raiseError(conn, result);
1302 CAMLreturn0;
1305 static long convert_bit_list(long *map, size_t map_size, value option)
1307 CAMLparam1(option);
1308 long bits = 0;
1309 int index;
1311 while (Val_emptylist != option)
1313 index = Int_val(Field(option, 0));
1314 if ((index < 0) || ((size_t)index >= map_size))
1315 caml_invalid_argument("convert_bit_list");
1317 bits |= map[index];
1319 option = Field(option, 1);
1322 CAMLreturnT(long, bits);
1325 #define SETOPT_STRING(name) \
1326 static void handle_##name(Connection *conn, value option) \
1328 CAMLparam1(option); \
1329 CURLcode result = CURLE_OK; \
1331 Store_field(conn->ocamlValues, Ocaml_##name, option); \
1333 result = curl_easy_setopt(conn->handle, CURLOPT_##name, String_val(option)); \
1335 if (result != CURLE_OK) \
1336 raiseError(conn, result); \
1338 CAMLreturn0; \
1341 #define SETOPT_VAL_(func_name, curl_option, conv_val) \
1342 static void func_name(Connection *conn, value option) \
1344 CAMLparam1(option); \
1345 CURLcode result = CURLE_OK; \
1347 result = curl_easy_setopt(conn->handle, curl_option, conv_val(option)); \
1349 if (result != CURLE_OK) \
1350 raiseError(conn, result); \
1352 CAMLreturn0; \
1355 #define SETOPT_VAL(name, conv) SETOPT_VAL_(handle_##name, CURLOPT_##name, conv)
1356 #define SETOPT_BOOL(name) SETOPT_VAL(name, Bool_val)
1357 #define SETOPT_LONG(name) SETOPT_VAL(name, Long_val)
1358 #define SETOPT_INT64(name) SETOPT_VAL(name, Int64_val)
1360 #if HAVE_DECL_CURLOPT_MIMEPOST
1362 static void new_part(Connection* conn, curl_mime* mime, value v_part)
1364 value v_encoding = Field(v_part, 0);
1365 value v_headers = Field(v_part, 1);
1366 value v_subparts = Field(v_part, 2);
1367 value v_data = Field(v_part, 3);
1368 value v_str = Field(v_data, 0);
1370 struct curl_slist *headers = NULL;
1371 CURLcode result;
1373 curl_mimepart *part = curl_mime_addpart(mime);
1375 switch (Int_val(v_encoding)) {
1376 case 0:
1377 result = curl_mime_encoder(part, "8bit");
1378 break;
1379 case 1:
1380 result = curl_mime_encoder(part, "binary");
1381 break;
1382 case 2:
1383 result = curl_mime_encoder(part, "7bit");
1384 break;
1385 case 3:
1386 result = curl_mime_encoder(part, "quoted-printable");
1387 break;
1388 case 4:
1389 result = curl_mime_encoder(part, "base64");
1390 break;
1391 case 5:
1392 result = CURLE_OK;
1393 break;
1394 default:
1395 caml_failwith("Invalid MIMEPOST encoding value");
1396 break;
1399 if (result != CURLE_OK) {
1400 raiseError(conn, result);
1403 while (v_headers != Val_emptylist) {
1404 headers = curl_slist_append(headers, String_val(Field(v_headers, 0)));
1405 v_headers = Field(v_headers, 1);
1408 result = curl_mime_headers(part, headers, 1);
1410 if (result != CURLE_OK) {
1411 raiseError(conn, result);
1414 switch (Tag_val(v_data)) {
1415 case 0:
1416 result = curl_mime_data(part, String_val(v_str), caml_string_length(v_str));
1417 break;
1418 case 1:
1419 result = curl_mime_filedata(part, String_val(v_str));
1420 break;
1421 default:
1422 caml_failwith("Invalid MIMEPOST data value");
1423 break;
1426 if (result != CURLE_OK) {
1427 raiseError(conn, result);
1430 if (v_subparts != Val_emptylist) {
1431 curl_mime *mime = curl_mime_init(conn->handle);
1433 while (v_subparts != Val_emptylist) {
1434 new_part(conn, mime, Field(v_subparts, 0));
1435 v_subparts = Field(v_subparts, 1);
1438 result = curl_mime_subparts(part, mime);
1440 if (result != CURLE_OK) {
1441 raiseError(conn, result);
1446 static void handle_MIMEPOST(Connection* conn, value v_subparts)
1448 CAMLparam1(v_subparts);
1449 curl_mime *mime = curl_mime_init(conn->handle);
1450 CURLcode result;
1452 Store_field(conn->ocamlValues, Ocaml_MIMEPOST, v_subparts);
1454 curl_mime_free(conn->curl_MIMEPOST);
1455 conn->curl_MIMEPOST = mime;
1457 while (v_subparts != Val_emptylist) {
1458 new_part(conn, mime, Field(v_subparts, 0));
1459 v_subparts = Field(v_subparts, 1);
1462 result = curl_easy_setopt(conn->handle, CURLOPT_MIMEPOST, mime);
1464 if (result != CURLE_OK) {
1465 raiseError(conn, result);
1468 CAMLreturn0;
1471 #endif
1473 #define SETOPT_SLIST(name) \
1474 static void handle_##name(Connection* conn, value option) \
1476 handle_slist(conn,&(conn->curl_##name),Ocaml_##name,CURLOPT_##name,option); \
1479 SETOPT_STRING( URL)
1480 SETOPT_LONG( INFILESIZE)
1481 SETOPT_STRING( PROXY)
1482 SETOPT_LONG( PROXYPORT)
1483 SETOPT_BOOL( HTTPPROXYTUNNEL)
1484 SETOPT_BOOL( VERBOSE)
1485 SETOPT_BOOL( HEADER)
1486 SETOPT_BOOL( NOPROGRESS)
1488 #if HAVE_DECL_CURLOPT_NOSIGNAL
1489 SETOPT_BOOL( NOSIGNAL)
1490 #endif
1492 SETOPT_BOOL( NOBODY)
1493 SETOPT_BOOL( FAILONERROR)
1494 SETOPT_BOOL( UPLOAD)
1495 SETOPT_BOOL( POST)
1496 SETOPT_BOOL( FTPLISTONLY)
1497 SETOPT_BOOL( FTPAPPEND)
1500 static void handle_NETRC(Connection *conn, value option)
1502 CAMLparam1(option);
1503 CURLcode result = CURLE_OK;
1504 long netrc;
1506 switch (Long_val(option))
1508 case 0: /* CURL_NETRC_OPTIONAL */
1509 netrc = CURL_NETRC_OPTIONAL;
1510 break;
1512 case 1:/* CURL_NETRC_IGNORED */
1513 netrc = CURL_NETRC_IGNORED;
1514 break;
1516 case 2: /* CURL_NETRC_REQUIRED */
1517 netrc = CURL_NETRC_REQUIRED;
1518 break;
1520 default:
1521 caml_failwith("Invalid NETRC Option");
1522 break;
1525 result = curl_easy_setopt(conn->handle,
1526 CURLOPT_NETRC,
1527 netrc);
1529 if (result != CURLE_OK)
1530 raiseError(conn, result);
1532 CAMLreturn0;
1535 #if HAVE_DECL_CURLOPT_ENCODING
1536 static void handle_ENCODING(Connection *conn, value option)
1538 CAMLparam1(option);
1539 CURLcode result = CURLE_OK;
1541 switch (Long_val(option))
1543 case 0: /* CURL_ENCODING_NONE */
1544 result = curl_easy_setopt(conn->handle,
1545 CURLOPT_ENCODING,
1546 "identity");
1547 break;
1549 case 1: /* CURL_ENCODING_DEFLATE */
1550 result = curl_easy_setopt(conn->handle,
1551 CURLOPT_ENCODING,
1552 "deflate");
1553 break;
1555 case 2: /* CURL_ENCODING_GZIP */
1556 result = curl_easy_setopt(conn->handle,
1557 CURLOPT_ENCODING,
1558 "gzip");
1559 break;
1561 case 3: /* CURL_ENCODING_ANY */
1562 result = curl_easy_setopt(conn->handle,
1563 CURLOPT_ENCODING,
1564 "");
1565 break;
1567 default:
1568 caml_failwith("Invalid Encoding Option");
1569 break;
1572 if (result != CURLE_OK)
1573 raiseError(conn, result);
1575 CAMLreturn0;
1577 #endif
1580 SETOPT_BOOL( FOLLOWLOCATION)
1581 SETOPT_BOOL( TRANSFERTEXT)
1582 SETOPT_BOOL( PUT)
1583 SETOPT_STRING( USERPWD)
1584 SETOPT_STRING( PROXYUSERPWD)
1585 SETOPT_STRING( RANGE)
1587 static void handle_ERRORBUFFER(Connection *conn, value option)
1589 CAMLparam1(option);
1590 CURLcode result = CURLE_OK;
1592 Store_field(conn->ocamlValues, Ocaml_ERRORBUFFER, option);
1594 if (conn->curl_ERRORBUFFER != NULL)
1595 free(conn->curl_ERRORBUFFER);
1597 conn->curl_ERRORBUFFER = (char*)malloc(sizeof(char) * CURL_ERROR_SIZE);
1599 result = curl_easy_setopt(conn->handle,
1600 CURLOPT_ERRORBUFFER,
1601 conn->curl_ERRORBUFFER);
1603 if (result != CURLE_OK)
1604 raiseError(conn, result);
1606 CAMLreturn0;
1609 SETOPT_LONG( TIMEOUT)
1611 static void handle_POSTFIELDS(Connection *conn, value option)
1613 CAMLparam1(option);
1614 CURLcode result = CURLE_OK;
1616 Store_field(conn->ocamlValues, Ocaml_POSTFIELDS, option);
1618 if (conn->curl_POSTFIELDS != NULL)
1619 free(conn->curl_POSTFIELDS);
1621 conn->curl_POSTFIELDS = strdup_ml(option);
1623 result = curl_easy_setopt(conn->handle,
1624 CURLOPT_POSTFIELDS,
1625 conn->curl_POSTFIELDS);
1627 if (result != CURLE_OK)
1628 raiseError(conn, result);
1630 CAMLreturn0;
1633 SETOPT_LONG( POSTFIELDSIZE)
1634 SETOPT_STRING( REFERER)
1635 SETOPT_STRING( USERAGENT)
1636 SETOPT_STRING( FTPPORT)
1637 SETOPT_LONG( LOW_SPEED_LIMIT)
1638 SETOPT_LONG( LOW_SPEED_TIME)
1639 SETOPT_LONG( RESUME_FROM)
1640 SETOPT_STRING( COOKIE)
1642 SETOPT_SLIST( HTTPHEADER)
1644 static void handle_HTTPPOST(Connection *conn, value option)
1646 CAMLparam1(option);
1647 CAMLlocal3(listIter, formItem, contentType);
1648 CURLcode result = CURLE_OK;
1650 listIter = option;
1652 Store_field(conn->ocamlValues, Ocaml_HTTPPOST, option);
1654 free_curl_slist(conn->httpPostBuffers);
1655 if (conn->httpPostFirst != NULL)
1656 curl_formfree(conn->httpPostFirst);
1658 conn->httpPostBuffers = NULL;
1659 conn->httpPostFirst = NULL;
1660 conn->httpPostLast = NULL;
1662 while (!Is_long(listIter))
1664 formItem = Field(listIter, 0);
1666 switch (Tag_val(formItem))
1668 case 0: /* CURLFORM_CONTENT */
1669 if (Wosize_val(formItem) < 3)
1671 caml_failwith("Incorrect CURLFORM_CONTENT parameters");
1674 if (Is_long(Field(formItem, 2)) &&
1675 Long_val(Field(formItem, 2)) == 0)
1677 curl_formadd(&conn->httpPostFirst,
1678 &conn->httpPostLast,
1679 CURLFORM_COPYNAME,
1680 String_val(Field(formItem, 0)),
1681 CURLFORM_NAMELENGTH,
1682 caml_string_length(Field(formItem, 0)),
1683 CURLFORM_COPYCONTENTS,
1684 String_val(Field(formItem, 1)),
1685 CURLFORM_CONTENTSLENGTH,
1686 caml_string_length(Field(formItem, 1)),
1687 CURLFORM_END);
1689 else if (Is_block(Field(formItem, 2)))
1691 contentType = Field(formItem, 2);
1693 curl_formadd(&conn->httpPostFirst,
1694 &conn->httpPostLast,
1695 CURLFORM_COPYNAME,
1696 String_val(Field(formItem, 0)),
1697 CURLFORM_NAMELENGTH,
1698 caml_string_length(Field(formItem, 0)),
1699 CURLFORM_COPYCONTENTS,
1700 String_val(Field(formItem, 1)),
1701 CURLFORM_CONTENTSLENGTH,
1702 caml_string_length(Field(formItem, 1)),
1703 CURLFORM_CONTENTTYPE,
1704 String_val(Field(contentType, 0)),
1705 CURLFORM_END);
1707 else
1709 caml_failwith("Incorrect CURLFORM_CONTENT parameters");
1711 break;
1713 case 1: /* CURLFORM_FILECONTENT */
1714 if (Wosize_val(formItem) < 3)
1716 caml_failwith("Incorrect CURLFORM_FILECONTENT parameters");
1719 if (Is_long(Field(formItem, 2)) &&
1720 Long_val(Field(formItem, 2)) == 0)
1722 curl_formadd(&conn->httpPostFirst,
1723 &conn->httpPostLast,
1724 CURLFORM_COPYNAME,
1725 String_val(Field(formItem, 0)),
1726 CURLFORM_NAMELENGTH,
1727 caml_string_length(Field(formItem, 0)),
1728 CURLFORM_FILECONTENT,
1729 String_val(Field(formItem, 1)),
1730 CURLFORM_END);
1732 else if (Is_block(Field(formItem, 2)))
1734 contentType = Field(formItem, 2);
1736 curl_formadd(&conn->httpPostFirst,
1737 &conn->httpPostLast,
1738 CURLFORM_COPYNAME,
1739 String_val(Field(formItem, 0)),
1740 CURLFORM_NAMELENGTH,
1741 caml_string_length(Field(formItem, 0)),
1742 CURLFORM_FILECONTENT,
1743 String_val(Field(formItem, 1)),
1744 CURLFORM_CONTENTTYPE,
1745 String_val(Field(contentType, 0)),
1746 CURLFORM_END);
1748 else
1750 caml_failwith("Incorrect CURLFORM_FILECONTENT parameters");
1752 break;
1754 case 2: /* CURLFORM_FILE */
1755 if (Wosize_val(formItem) < 3)
1757 caml_failwith("Incorrect CURLFORM_FILE parameters");
1760 if (Is_long(Field(formItem, 2)) &&
1761 Long_val(Field(formItem, 2)) == 0)
1763 curl_formadd(&conn->httpPostFirst,
1764 &conn->httpPostLast,
1765 CURLFORM_COPYNAME,
1766 String_val(Field(formItem, 0)),
1767 CURLFORM_NAMELENGTH,
1768 caml_string_length(Field(formItem, 0)),
1769 CURLFORM_FILE,
1770 String_val(Field(formItem, 1)),
1771 CURLFORM_END);
1773 else if (Is_block(Field(formItem, 2)))
1775 contentType = Field(formItem, 2);
1777 curl_formadd(&conn->httpPostFirst,
1778 &conn->httpPostLast,
1779 CURLFORM_COPYNAME,
1780 String_val(Field(formItem, 0)),
1781 CURLFORM_NAMELENGTH,
1782 caml_string_length(Field(formItem, 0)),
1783 CURLFORM_FILE,
1784 String_val(Field(formItem, 1)),
1785 CURLFORM_CONTENTTYPE,
1786 String_val(Field(contentType, 0)),
1787 CURLFORM_END);
1789 else
1791 caml_failwith("Incorrect CURLFORM_FILE parameters");
1793 break;
1795 case 3: /* CURLFORM_BUFFER */
1796 if (Wosize_val(formItem) < 4)
1798 caml_failwith("Incorrect CURLFORM_BUFFER parameters");
1801 if (Is_long(Field(formItem, 3)) &&
1802 Long_val(Field(formItem, 3)) == 0)
1804 conn->httpPostBuffers = curl_slist_prepend_ml(conn->httpPostBuffers, Field(formItem, 2));
1806 curl_formadd(&conn->httpPostFirst,
1807 &conn->httpPostLast,
1808 CURLFORM_COPYNAME,
1809 String_val(Field(formItem, 0)),
1810 CURLFORM_NAMELENGTH,
1811 caml_string_length(Field(formItem, 0)),
1812 CURLFORM_BUFFER,
1813 String_val(Field(formItem, 1)),
1814 CURLFORM_BUFFERPTR,
1815 conn->httpPostBuffers->data,
1816 CURLFORM_BUFFERLENGTH,
1817 caml_string_length(Field(formItem, 2)),
1818 CURLFORM_END);
1820 else if (Is_block(Field(formItem, 3)))
1822 conn->httpPostBuffers = curl_slist_prepend_ml(conn->httpPostBuffers, Field(formItem, 2));
1824 contentType = Field(formItem, 3);
1826 curl_formadd(&conn->httpPostFirst,
1827 &conn->httpPostLast,
1828 CURLFORM_COPYNAME,
1829 String_val(Field(formItem, 0)),
1830 CURLFORM_NAMELENGTH,
1831 caml_string_length(Field(formItem, 0)),
1832 CURLFORM_BUFFER,
1833 String_val(Field(formItem, 1)),
1834 CURLFORM_BUFFERPTR,
1835 conn->httpPostBuffers->data,
1836 CURLFORM_BUFFERLENGTH,
1837 caml_string_length(Field(formItem, 2)),
1838 CURLFORM_CONTENTTYPE,
1839 String_val(Field(contentType, 0)),
1840 CURLFORM_END);
1842 else
1844 caml_failwith("Incorrect CURLFORM_BUFFER parameters");
1846 break;
1849 listIter = Field(listIter, 1);
1852 result = curl_easy_setopt(conn->handle,
1853 CURLOPT_HTTPPOST,
1854 conn->httpPostFirst);
1856 if (result != CURLE_OK)
1857 raiseError(conn, result);
1859 CAMLreturn0;
1862 SETOPT_STRING( SSLCERT)
1863 SETOPT_STRING( SSLCERTTYPE)
1864 SETOPT_STRING( SSLCERTPASSWD)
1865 SETOPT_STRING( SSLKEY)
1866 SETOPT_STRING( SSLKEYTYPE)
1867 SETOPT_STRING( SSLKEYPASSWD)
1868 SETOPT_STRING( SSLENGINE)
1869 SETOPT_BOOL( SSLENGINE_DEFAULT)
1870 SETOPT_BOOL( CRLF)
1872 SETOPT_SLIST( QUOTE)
1873 SETOPT_SLIST( POSTQUOTE)
1875 SETOPT_STRING( COOKIEFILE)
1876 #if HAVE_DECL_CURLOPT_CERTINFO
1877 SETOPT_BOOL( CERTINFO)
1878 #endif
1880 #if !defined(CURL_SSLVERSION_TLSv1_0)
1881 #define CURL_SSLVERSION_TLSv1_0 CURL_SSLVERSION_TLSv1
1882 #endif
1884 #if !defined(CURL_SSLVERSION_TLSv1_1)
1885 #define CURL_SSLVERSION_TLSv1_1 CURL_SSLVERSION_TLSv1
1886 #endif
1888 #if !defined(CURL_SSLVERSION_TLSv1_2)
1889 #define CURL_SSLVERSION_TLSv1_2 CURL_SSLVERSION_TLSv1
1890 #endif
1892 #if !defined(CURL_SSLVERSION_TLSv1_3)
1893 #define CURL_SSLVERSION_TLSv1_3 CURL_SSLVERSION_TLSv1
1894 #endif
1896 static void handle_SSLVERSION(Connection *conn, value option)
1898 CAMLparam1(option);
1899 CURLcode result = CURLE_OK;
1900 int v = CURL_SSLVERSION_DEFAULT;
1902 switch (Long_val(option))
1904 case 0: v = CURL_SSLVERSION_DEFAULT; break;
1905 case 1: v = CURL_SSLVERSION_TLSv1; break;
1906 case 2: v = CURL_SSLVERSION_SSLv2; break;
1907 case 3: v = CURL_SSLVERSION_SSLv3; break;
1908 case 4: v = CURL_SSLVERSION_TLSv1_0; break;
1909 case 5: v = CURL_SSLVERSION_TLSv1_1; break;
1910 case 6: v = CURL_SSLVERSION_TLSv1_2; break;
1911 case 7: v = CURL_SSLVERSION_TLSv1_3; break;
1912 default:
1913 caml_failwith("Invalid SSLVERSION Option");
1914 break;
1917 result = curl_easy_setopt(conn->handle, CURLOPT_SSLVERSION, v);
1919 if (result != CURLE_OK)
1920 raiseError(conn, result);
1922 CAMLreturn0;
1925 static void handle_TIMECONDITION(Connection *conn, value option)
1927 CAMLparam1(option);
1928 CURLcode result = CURLE_OK;
1929 int timecond = CURL_TIMECOND_NONE;
1931 switch (Long_val(option))
1933 case 0: timecond = CURL_TIMECOND_NONE; break;
1934 case 1: timecond = CURL_TIMECOND_IFMODSINCE; break;
1935 case 2: timecond = CURL_TIMECOND_IFUNMODSINCE; break;
1936 case 3: timecond = CURL_TIMECOND_LASTMOD; break;
1937 default:
1938 caml_failwith("Invalid TIMECOND Option");
1939 break;
1942 result = curl_easy_setopt(conn->handle, CURLOPT_TIMECONDITION, timecond);
1944 if (result != CURLE_OK)
1945 raiseError(conn, result);
1947 CAMLreturn0;
1950 SETOPT_VAL( TIMEVALUE, Int32_val)
1951 SETOPT_STRING( CUSTOMREQUEST)
1952 SETOPT_STRING( INTERFACE)
1954 static void handle_KRB4LEVEL(Connection *conn, value option)
1956 CAMLparam1(option);
1957 CURLcode result = CURLE_OK;
1959 switch (Long_val(option))
1961 case 0: /* KRB4_NONE */
1962 result = curl_easy_setopt(conn->handle,
1963 CURLOPT_KRB4LEVEL,
1964 NULL);
1965 break;
1967 case 1: /* KRB4_CLEAR */
1968 result = curl_easy_setopt(conn->handle,
1969 CURLOPT_KRB4LEVEL,
1970 "clear");
1971 break;
1973 case 2: /* KRB4_SAFE */
1974 result = curl_easy_setopt(conn->handle,
1975 CURLOPT_KRB4LEVEL,
1976 "safe");
1977 break;
1979 case 3: /* KRB4_CONFIDENTIAL */
1980 result = curl_easy_setopt(conn->handle,
1981 CURLOPT_KRB4LEVEL,
1982 "confidential");
1983 break;
1985 case 4: /* KRB4_PRIVATE */
1986 result = curl_easy_setopt(conn->handle,
1987 CURLOPT_KRB4LEVEL,
1988 "private");
1989 break;
1991 default:
1992 caml_failwith("Invalid KRB4 Option");
1993 break;
1996 if (result != CURLE_OK)
1997 raiseError(conn, result);
1999 CAMLreturn0;
2002 SETOPT_BOOL( SSL_VERIFYPEER)
2003 SETOPT_STRING( CAINFO)
2004 SETOPT_STRING( CAPATH)
2005 SETOPT_BOOL( FILETIME)
2006 SETOPT_LONG( MAXREDIRS)
2007 SETOPT_LONG( MAXCONNECTS)
2009 static void handle_CLOSEPOLICY(Connection *conn, value option)
2011 CAMLparam1(option);
2012 CURLcode result = CURLE_OK;
2014 switch (Long_val(option))
2016 case 0: /* CLOSEPOLICY_OLDEST */
2017 result = curl_easy_setopt(conn->handle,
2018 CURLOPT_CLOSEPOLICY,
2019 CURLCLOSEPOLICY_OLDEST);
2020 break;
2022 case 1: /* CLOSEPOLICY_LEAST_RECENTLY_USED */
2023 result = curl_easy_setopt(conn->handle,
2024 CURLOPT_CLOSEPOLICY,
2025 CURLCLOSEPOLICY_LEAST_RECENTLY_USED);
2026 break;
2028 default:
2029 caml_failwith("Invalid CLOSEPOLICY Option");
2030 break;
2033 if (result != CURLE_OK)
2034 raiseError(conn, result);
2036 CAMLreturn0;
2039 SETOPT_BOOL( FRESH_CONNECT)
2040 SETOPT_BOOL( FORBID_REUSE)
2041 SETOPT_STRING( RANDOM_FILE)
2042 SETOPT_STRING( EGDSOCKET)
2043 SETOPT_LONG( CONNECTTIMEOUT)
2044 SETOPT_BOOL( HTTPGET)
2046 static void handle_SSL_VERIFYHOST(Connection *conn, value option)
2048 CAMLparam1(option);
2049 CURLcode result = CURLE_OK;
2051 switch (Long_val(option))
2053 case 0: /* SSLVERIFYHOST_NONE */
2054 case 1: /* SSLVERIFYHOST_EXISTENCE */
2055 case 2: /* SSLVERIFYHOST_HOSTNAME */
2056 result = curl_easy_setopt(conn->handle,
2057 CURLOPT_SSL_VERIFYHOST,
2058 /* map EXISTENCE to HOSTNAME */
2059 Long_val(option) == 0 ? 0 : 2);
2060 break;
2062 default:
2063 caml_failwith("Invalid SSLVERIFYHOST Option");
2064 break;
2067 if (result != CURLE_OK)
2068 raiseError(conn, result);
2070 CAMLreturn0;
2073 SETOPT_STRING( COOKIEJAR)
2074 SETOPT_STRING( SSL_CIPHER_LIST)
2076 static void handle_HTTP_VERSION(Connection *conn, value option)
2078 CAMLparam1(option);
2079 CURLcode result = CURLE_OK;
2081 long version = CURL_HTTP_VERSION_NONE;
2083 switch (Long_val(option))
2085 case 0: version = CURL_HTTP_VERSION_NONE; break;
2086 case 1: version = CURL_HTTP_VERSION_1_0; break;
2087 case 2: version = CURL_HTTP_VERSION_1_1; break;
2088 case 3:
2089 #if defined(CURL_HTTP_VERSION_2)
2090 version = CURL_HTTP_VERSION_2;
2091 #elif defined(CURL_HTTP_VERSION_2_0)
2092 version = CURL_HTTP_VERSION_2_0;
2093 #endif
2094 break;
2095 case 4:
2096 #if defined(CURL_HTTP_VERSION_2TLS)
2097 version = CURL_HTTP_VERSION_2TLS;
2098 #endif
2099 break;
2100 default:
2101 caml_invalid_argument("CURLOPT_HTTP_VERSION");
2102 break;
2105 result = curl_easy_setopt(conn->handle, CURLOPT_HTTP_VERSION, version);
2107 if (result != CURLE_OK)
2108 raiseError(conn, result);
2110 CAMLreturn0;
2113 SETOPT_BOOL( FTP_USE_EPSV)
2114 SETOPT_LONG( DNS_CACHE_TIMEOUT)
2115 SETOPT_BOOL( DNS_USE_GLOBAL_CACHE)
2117 static void handle_PRIVATE(Connection *conn, value option)
2119 CAMLparam1(option);
2120 Store_field(conn->ocamlValues, Ocaml_PRIVATE, option);
2121 CAMLreturn0;
2124 #if HAVE_DECL_CURLOPT_HTTP200ALIASES
2125 SETOPT_SLIST( HTTP200ALIASES)
2126 #endif
2128 #if HAVE_DECL_CURLOPT_UNRESTRICTED_AUTH
2129 SETOPT_BOOL( UNRESTRICTED_AUTH)
2130 #endif
2132 #if HAVE_DECL_CURLOPT_FTP_USE_EPRT
2133 SETOPT_BOOL( FTP_USE_EPRT)
2134 #endif
2136 #if HAVE_DECL_CURLOPT_HTTPAUTH
2137 static void handle_HTTPAUTH(Connection *conn, value option)
2139 CAMLparam1(option);
2140 CAMLlocal1(listIter);
2141 CURLcode result = CURLE_OK;
2142 long auth = CURLAUTH_NONE;
2144 listIter = option;
2146 while (!Is_long(listIter))
2148 switch (Long_val(Field(listIter, 0)))
2150 case 0: /* CURLAUTH_BASIC */
2151 auth |= CURLAUTH_BASIC;
2152 break;
2154 case 1: /* CURLAUTH_DIGEST */
2155 auth |= CURLAUTH_DIGEST;
2156 break;
2158 case 2: /* CURLAUTH_GSSNEGOTIATE */
2159 auth |= CURLAUTH_GSSNEGOTIATE;
2160 break;
2162 case 3: /* CURLAUTH_NTLM */
2163 auth |= CURLAUTH_NTLM;
2164 break;
2166 case 4: /* CURLAUTH_ANY */
2167 auth |= CURLAUTH_ANY;
2168 break;
2170 case 5: /* CURLAUTH_ANYSAFE */
2171 auth |= CURLAUTH_ANYSAFE;
2172 break;
2174 default:
2175 caml_failwith("Invalid HTTPAUTH Value");
2176 break;
2179 listIter = Field(listIter, 1);
2182 result = curl_easy_setopt(conn->handle,
2183 CURLOPT_HTTPAUTH,
2184 auth);
2186 if (result != CURLE_OK)
2187 raiseError(conn, result);
2189 CAMLreturn0;
2191 #endif
2193 #if HAVE_DECL_CURLOPT_FTP_CREATE_MISSING_DIRS
2194 SETOPT_BOOL( FTP_CREATE_MISSING_DIRS)
2195 #endif
2197 #if HAVE_DECL_CURLOPT_PROXYAUTH
2198 static void handle_PROXYAUTH(Connection *conn, value option)
2200 CAMLparam1(option);
2201 CAMLlocal1(listIter);
2202 CURLcode result = CURLE_OK;
2203 long auth = CURLAUTH_NONE;
2205 listIter = option;
2207 while (!Is_long(listIter))
2209 switch (Long_val(Field(listIter, 0)))
2211 case 0: /* CURLAUTH_BASIC */
2212 auth |= CURLAUTH_BASIC;
2213 break;
2215 case 1: /* CURLAUTH_DIGEST */
2216 auth |= CURLAUTH_DIGEST;
2217 break;
2219 case 2: /* CURLAUTH_GSSNEGOTIATE */
2220 auth |= CURLAUTH_GSSNEGOTIATE;
2221 break;
2223 case 3: /* CURLAUTH_NTLM */
2224 auth |= CURLAUTH_NTLM;
2225 break;
2227 case 4: /* CURLAUTH_ANY */
2228 auth |= CURLAUTH_ANY;
2229 break;
2231 case 5: /* CURLAUTH_ANYSAFE */
2232 auth |= CURLAUTH_ANYSAFE;
2233 break;
2235 default:
2236 caml_failwith("Invalid HTTPAUTH Value");
2237 break;
2240 listIter = Field(listIter, 1);
2243 result = curl_easy_setopt(conn->handle,
2244 CURLOPT_PROXYAUTH,
2245 auth);
2247 if (result != CURLE_OK)
2248 raiseError(conn, result);
2250 CAMLreturn0;
2252 #endif
2254 #if HAVE_DECL_CURLOPT_FTP_RESPONSE_TIMEOUT
2255 SETOPT_LONG( FTP_RESPONSE_TIMEOUT)
2256 #endif
2258 #if HAVE_DECL_CURLOPT_IPRESOLVE
2259 static void handle_IPRESOLVE(Connection *conn, value option)
2261 CAMLparam1(option);
2262 CURLcode result = CURLE_OK;
2264 switch (Long_val(option))
2266 case 0: /* CURL_IPRESOLVE_WHATEVER */
2267 result = curl_easy_setopt(conn->handle,
2268 CURLOPT_IPRESOLVE,
2269 CURL_IPRESOLVE_WHATEVER);
2270 break;
2272 case 1: /* CURL_IPRESOLVE_V4 */
2273 result = curl_easy_setopt(conn->handle,
2274 CURLOPT_IPRESOLVE,
2275 CURL_IPRESOLVE_V4);
2276 break;
2278 case 2: /* CURL_IPRESOLVE_V6 */
2279 result = curl_easy_setopt(conn->handle,
2280 CURLOPT_IPRESOLVE,
2281 CURL_IPRESOLVE_V6);
2282 break;
2284 default:
2285 caml_failwith("Invalid IPRESOLVE Value");
2286 break;
2289 if (result != CURLE_OK)
2290 raiseError(conn, result);
2292 CAMLreturn0;
2294 #endif
2296 #if HAVE_DECL_CURLOPT_MAXFILESIZE
2297 SETOPT_VAL( MAXFILESIZE, Int32_val)
2298 #endif
2300 #if HAVE_DECL_CURLOPT_INFILESIZE_LARGE
2301 SETOPT_INT64( INFILESIZE_LARGE)
2302 #endif
2304 #if HAVE_DECL_CURLOPT_RESUME_FROM_LARGE
2305 SETOPT_INT64( RESUME_FROM_LARGE)
2306 #endif
2308 #if HAVE_DECL_CURLOPT_MAXFILESIZE_LARGE
2309 SETOPT_INT64( MAXFILESIZE_LARGE)
2310 #endif
2312 #if HAVE_DECL_CURLOPT_NETRC_FILE
2313 SETOPT_STRING( NETRC_FILE)
2314 #endif
2316 #if HAVE_DECL_CURLOPT_FTP_SSL
2317 static void handle_FTP_SSL(Connection *conn, value option)
2319 CAMLparam1(option);
2320 CURLcode result = CURLE_OK;
2322 switch (Long_val(option))
2324 case 0: /* CURLFTPSSL_NONE */
2325 result = curl_easy_setopt(conn->handle,
2326 CURLOPT_FTP_SSL,
2327 CURLFTPSSL_NONE);
2328 break;
2330 case 1: /* CURLFTPSSL_TRY */
2331 result = curl_easy_setopt(conn->handle,
2332 CURLOPT_FTP_SSL,
2333 CURLFTPSSL_TRY);
2334 break;
2336 case 2: /* CURLFTPSSL_CONTROL */
2337 result = curl_easy_setopt(conn->handle,
2338 CURLOPT_FTP_SSL,
2339 CURLFTPSSL_CONTROL);
2340 break;
2342 case 3: /* CURLFTPSSL_ALL */
2343 result = curl_easy_setopt(conn->handle,
2344 CURLOPT_FTP_SSL,
2345 CURLFTPSSL_ALL);
2346 break;
2348 default:
2349 caml_failwith("Invalid FTP_SSL Value");
2350 break;
2353 if (result != CURLE_OK)
2354 raiseError(conn, result);
2356 CAMLreturn0;
2358 #endif
2360 #if HAVE_DECL_CURLOPT_POSTFIELDSIZE_LARGE
2361 SETOPT_INT64( POSTFIELDSIZE_LARGE)
2362 #endif
2364 #if HAVE_DECL_CURLOPT_TCP_NODELAY
2365 /* not using SETOPT_BOOL here because of TCP_NODELAY defined in winsock.h */
2366 SETOPT_VAL_( handle_TCP_NODELAY, CURLOPT_TCP_NODELAY, Bool_val)
2367 #endif
2369 #if HAVE_DECL_CURLOPT_FTPSSLAUTH
2370 static void handle_FTPSSLAUTH(Connection *conn, value option)
2372 CAMLparam1(option);
2373 CURLcode result = CURLE_OK;
2375 switch (Long_val(option))
2377 case 0: /* CURLFTPAUTH_DEFAULT */
2378 result = curl_easy_setopt(conn->handle,
2379 CURLOPT_FTPSSLAUTH,
2380 CURLFTPAUTH_DEFAULT);
2381 break;
2383 case 1: /* CURLFTPAUTH_SSL */
2384 result = curl_easy_setopt(conn->handle,
2385 CURLOPT_FTPSSLAUTH,
2386 CURLFTPAUTH_SSL);
2387 break;
2389 case 2: /* CURLFTPAUTH_TLS */
2390 result = curl_easy_setopt(conn->handle,
2391 CURLOPT_FTPSSLAUTH,
2392 CURLFTPAUTH_TLS);
2393 break;
2395 default:
2396 caml_failwith("Invalid FTPSSLAUTH value");
2397 break;
2400 if (result != CURLE_OK)
2401 raiseError(conn, result);
2403 CAMLreturn0;
2405 #endif
2407 #if HAVE_DECL_CURLOPT_FTP_ACCOUNT
2408 SETOPT_STRING( FTP_ACCOUNT)
2409 #endif
2411 #if HAVE_DECL_CURLOPT_COOKIELIST
2412 SETOPT_STRING( COOKIELIST)
2413 #endif
2415 #if HAVE_DECL_CURLOPT_IGNORE_CONTENT_LENGTH
2416 SETOPT_BOOL( IGNORE_CONTENT_LENGTH)
2417 #endif
2419 #if HAVE_DECL_CURLOPT_FTP_SKIP_PASV_IP
2420 SETOPT_BOOL( FTP_SKIP_PASV_IP)
2421 #endif
2423 #if HAVE_DECL_CURLOPT_FTP_FILEMETHOD
2424 static void handle_FTP_FILEMETHOD(Connection *conn, value option)
2426 CAMLparam1(option);
2427 CURLcode result = CURLE_OK;
2429 switch (Long_val(option))
2431 case 0: /* CURLFTPMETHOD_DEFAULT */
2432 result = curl_easy_setopt(conn->handle,
2433 CURLOPT_FTP_FILEMETHOD,
2434 CURLFTPMETHOD_DEFAULT);
2435 break;
2437 case 1: /* CURLFTMETHOD_MULTICWD */
2438 result = curl_easy_setopt(conn->handle,
2439 CURLOPT_FTP_FILEMETHOD,
2440 CURLFTPMETHOD_MULTICWD);
2441 break;
2443 case 2: /* CURLFTPMETHOD_NOCWD */
2444 result = curl_easy_setopt(conn->handle,
2445 CURLOPT_FTP_FILEMETHOD,
2446 CURLFTPMETHOD_NOCWD);
2447 break;
2449 case 3: /* CURLFTPMETHOD_SINGLECWD */
2450 result = curl_easy_setopt(conn->handle,
2451 CURLOPT_FTP_FILEMETHOD,
2452 CURLFTPMETHOD_SINGLECWD);
2454 default:
2455 caml_failwith("Invalid FTP_FILEMETHOD value");
2456 break;
2459 if (result != CURLE_OK)
2460 raiseError(conn, result);
2462 CAMLreturn0;
2464 #endif
2466 #if HAVE_DECL_CURLOPT_LOCALPORT
2467 SETOPT_LONG( LOCALPORT)
2468 #endif
2470 #if HAVE_DECL_CURLOPT_LOCALPORTRANGE
2471 SETOPT_LONG( LOCALPORTRANGE)
2472 #endif
2474 #if HAVE_DECL_CURLOPT_CONNECT_ONLY
2475 SETOPT_BOOL( CONNECT_ONLY)
2476 #endif
2478 #if HAVE_DECL_CURLOPT_MAX_SEND_SPEED_LARGE
2479 SETOPT_INT64( MAX_SEND_SPEED_LARGE)
2480 #endif
2482 #if HAVE_DECL_CURLOPT_MAX_RECV_SPEED_LARGE
2483 SETOPT_INT64( MAX_RECV_SPEED_LARGE)
2484 #endif
2486 #if HAVE_DECL_CURLOPT_FTP_ALTERNATIVE_TO_USER
2487 SETOPT_STRING( FTP_ALTERNATIVE_TO_USER)
2488 #endif
2490 #if HAVE_DECL_CURLOPT_SSL_SESSIONID_CACHE
2491 SETOPT_BOOL( SSL_SESSIONID_CACHE)
2492 #endif
2494 #if HAVE_DECL_CURLOPT_SSH_AUTH_TYPES
2495 static void handle_SSH_AUTH_TYPES(Connection *conn, value option)
2497 CAMLparam1(option);
2498 CAMLlocal1(listIter);
2499 CURLcode result = CURLE_OK;
2500 long authTypes = CURLSSH_AUTH_NONE;
2502 listIter = option;
2504 while (!Is_long(listIter))
2506 switch (Long_val(Field(listIter, 0)))
2508 case 0: /* CURLSSH_AUTH_ANY */
2509 authTypes |= CURLSSH_AUTH_ANY;
2510 break;
2512 case 1: /* CURLSSH_AUTH_PUBLICKEY */
2513 authTypes |= CURLSSH_AUTH_PUBLICKEY;
2514 break;
2516 case 2: /* CURLSSH_AUTH_PASSWORD */
2517 authTypes |= CURLSSH_AUTH_PASSWORD;
2518 break;
2520 case 3: /* CURLSSH_AUTH_HOST */
2521 authTypes |= CURLSSH_AUTH_HOST;
2522 break;
2524 case 4: /* CURLSSH_AUTH_KEYBOARD */
2525 authTypes |= CURLSSH_AUTH_KEYBOARD;
2526 break;
2528 default:
2529 caml_failwith("Invalid CURLSSH_AUTH_TYPES Value");
2530 break;
2533 listIter = Field(listIter, 1);
2536 result = curl_easy_setopt(conn->handle,
2537 CURLOPT_SSH_AUTH_TYPES,
2538 authTypes);
2540 if (result != CURLE_OK)
2541 raiseError(conn, result);
2543 CAMLreturn0;
2545 #endif
2547 #if HAVE_DECL_CURLOPT_SSH_PUBLIC_KEYFILE
2548 SETOPT_STRING( SSH_PUBLIC_KEYFILE)
2549 #endif
2551 #if HAVE_DECL_CURLOPT_SSH_PRIVATE_KEYFILE
2552 SETOPT_STRING( SSH_PRIVATE_KEYFILE)
2553 #endif
2555 #if HAVE_DECL_CURLOPT_FTP_SSL_CCC
2556 static void handle_FTP_SSL_CCC(Connection *conn, value option)
2558 CAMLparam1(option);
2559 CURLcode result = CURLE_OK;
2561 switch (Long_val(option))
2563 case 0: /* CURLFTPSSL_CCC_NONE */
2564 result = curl_easy_setopt(conn->handle,
2565 CURLOPT_FTP_SSL_CCC,
2566 CURLFTPSSL_CCC_NONE);
2567 break;
2569 case 1: /* CURLFTPSSL_CCC_PASSIVE */
2570 result = curl_easy_setopt(conn->handle,
2571 CURLOPT_FTP_SSL_CCC,
2572 CURLFTPSSL_CCC_PASSIVE);
2573 break;
2575 case 2: /* CURLFTPSSL_CCC_ACTIVE */
2576 result = curl_easy_setopt(conn->handle,
2577 CURLOPT_FTP_SSL_CCC,
2578 CURLFTPSSL_CCC_ACTIVE);
2579 break;
2581 default:
2582 caml_failwith("Invalid FTPSSL_CCC value");
2583 break;
2586 if (result != CURLE_OK)
2587 raiseError(conn, result);
2589 CAMLreturn0;
2591 #endif
2593 #if HAVE_DECL_CURLOPT_TIMEOUT_MS
2594 SETOPT_LONG( TIMEOUT_MS)
2595 #endif
2597 #if HAVE_DECL_CURLOPT_CONNECTTIMEOUT_MS
2598 SETOPT_LONG( CONNECTTIMEOUT_MS)
2599 #endif
2601 #if HAVE_DECL_CURLOPT_HTTP_TRANSFER_DECODING
2602 SETOPT_BOOL( HTTP_TRANSFER_DECODING)
2603 #endif
2605 #if HAVE_DECL_CURLOPT_HTTP_CONTENT_DECODING
2606 SETOPT_BOOL( HTTP_CONTENT_DECODING)
2607 #endif
2609 #if HAVE_DECL_CURLOPT_NEW_FILE_PERMS
2610 SETOPT_LONG( NEW_FILE_PERMS)
2611 #endif
2613 #if HAVE_DECL_CURLOPT_NEW_DIRECTORY_PERMS
2614 SETOPT_LONG( NEW_DIRECTORY_PERMS)
2615 #endif
2617 #if HAVE_DECL_CURLOPT_POST301
2618 SETOPT_BOOL( POST301)
2619 #endif
2621 #if HAVE_DECL_CURLOPT_SSH_HOST_PUBLIC_KEY_MD5
2622 SETOPT_STRING( SSH_HOST_PUBLIC_KEY_MD5)
2623 #endif
2625 #if HAVE_DECL_CURLOPT_COPYPOSTFIELDS
2626 SETOPT_STRING( COPYPOSTFIELDS)
2627 #endif
2629 #if HAVE_DECL_CURLOPT_PROXY_TRANSFER_MODE
2630 SETOPT_BOOL( PROXY_TRANSFER_MODE)
2631 #endif
2633 #if HAVE_DECL_CURLOPT_AUTOREFERER
2634 SETOPT_BOOL( AUTOREFERER)
2635 #endif
2637 #if HAVE_DECL_CURLOPT_PROXYTYPE
2638 static void handle_PROXYTYPE(Connection *conn, value option)
2640 CAMLparam1(option);
2641 CURLcode result = CURLE_OK;
2642 long proxy_type;
2644 switch (Long_val(option))
2646 case 0: proxy_type = CURLPROXY_HTTP; break;
2647 case 1: proxy_type = CURLPROXY_HTTP_1_0; break;
2648 case 2: proxy_type = CURLPROXY_SOCKS4; break;
2649 case 3: proxy_type = CURLPROXY_SOCKS5; break;
2650 case 4: proxy_type = CURLPROXY_SOCKS4A; break;
2651 case 5: proxy_type = CURLPROXY_SOCKS5_HOSTNAME; break;
2652 default:
2653 caml_failwith("Invalid curl proxy type");
2656 result = curl_easy_setopt(conn->handle,
2657 CURLOPT_PROXYTYPE,
2658 proxy_type);
2660 if (result != CURLE_OK)
2661 raiseError(conn, result);
2663 CAMLreturn0;
2665 #endif
2667 #if HAVE_DECL_CURLOPT_PROTOCOLS || HAVE_DECL_CURLOPT_REDIR_PROTOCOLS
2669 long protoMap[] =
2671 CURLPROTO_ALL,
2672 CURLPROTO_HTTP, CURLPROTO_HTTPS, CURLPROTO_FTP, CURLPROTO_FTPS, CURLPROTO_SCP, CURLPROTO_SFTP,
2673 CURLPROTO_TELNET, CURLPROTO_LDAP, CURLPROTO_LDAPS, CURLPROTO_DICT, CURLPROTO_FILE, CURLPROTO_TFTP,
2674 /* factor out with autoconf? */
2675 #if defined(CURLPROTO_IMAP)
2676 CURLPROTO_IMAP,
2677 #else
2679 #endif
2680 #if defined(CURLPROTO_IMAPS)
2681 CURLPROTO_IMAPS,
2682 #else
2684 #endif
2685 #if defined(CURLPROTO_POP3)
2686 CURLPROTO_POP3,
2687 #else
2689 #endif
2690 #if defined(CURLPROTO_POP3S)
2691 CURLPROTO_POP3S,
2692 #else
2694 #endif
2695 #if defined(CURLPROTO_SMTP)
2696 CURLPROTO_SMTP,
2697 #else
2699 #endif
2700 #if defined(CURLPROTO_SMTPS)
2701 CURLPROTO_SMTPS,
2702 #else
2704 #endif
2705 #if defined(CURLPROTO_RTSP)
2706 CURLPROTO_RTSP,
2707 #else
2709 #endif
2710 #if defined(CURLPROTO_RTMP)
2711 CURLPROTO_RTMP,
2712 #else
2714 #endif
2715 #if defined(CURLPROTO_RTMPT)
2716 CURLPROTO_RTMPT,
2717 #else
2719 #endif
2720 #if defined(CURLPROTO_RTMPE)
2721 CURLPROTO_RTMPE,
2722 #else
2724 #endif
2725 #if defined(CURLPROTO_RTMPTE)
2726 CURLPROTO_RTMPTE,
2727 #else
2729 #endif
2730 #if defined(CURLPROTO_RTMPS)
2731 CURLPROTO_RTMPS,
2732 #else
2734 #endif
2735 #if defined(CURLPROTO_RTMPTS)
2736 CURLPROTO_RTMPTS,
2737 #else
2739 #endif
2740 #if defined(CURLPROTO_GOPHER)
2741 CURLPROTO_GOPHER,
2742 #else
2744 #endif
2747 static void handle_PROTOCOLSOPTION(CURLoption curlopt, Connection *conn, value option)
2749 CAMLparam1(option);
2750 CURLcode result = CURLE_OK;
2751 long bits = convert_bit_list(protoMap, sizeof(protoMap) / sizeof(protoMap[0]), option);
2753 result = curl_easy_setopt(conn->handle, curlopt, bits);
2755 if (result != CURLE_OK)
2756 raiseError(conn, result);
2758 CAMLreturn0;
2760 #endif
2762 #if HAVE_DECL_CURLOPT_PROTOCOLS
2763 static void handle_PROTOCOLS(Connection *conn, value option)
2765 handle_PROTOCOLSOPTION(CURLOPT_PROTOCOLS, conn, option);
2767 #endif
2769 #if HAVE_DECL_CURLOPT_REDIR_PROTOCOLS
2770 static void handle_REDIR_PROTOCOLS(Connection *conn, value option)
2772 handle_PROTOCOLSOPTION(CURLOPT_REDIR_PROTOCOLS, conn, option);
2774 #endif
2776 #if HAVE_DECL_CURLOPT_RESOLVE
2777 SETOPT_SLIST( RESOLVE)
2778 #endif
2780 #if HAVE_DECL_CURLOPT_DNS_SERVERS
2781 SETOPT_STRING( DNS_SERVERS)
2782 #endif
2784 #if HAVE_DECL_CURLOPT_MAIL_FROM
2785 SETOPT_STRING( MAIL_FROM)
2786 #endif
2788 #if HAVE_DECL_CURLOPT_MAIL_RCPT
2789 SETOPT_SLIST( MAIL_RCPT)
2790 #endif
2792 #if HAVE_DECL_CURLOPT_PIPEWAIT
2793 SETOPT_BOOL( PIPEWAIT)
2794 #endif
2796 #if HAVE_DECL_CURLOPT_USERNAME
2797 SETOPT_STRING( USERNAME)
2798 #endif
2800 #if HAVE_DECL_CURLOPT_PASSWORD
2801 SETOPT_STRING( PASSWORD)
2802 #endif
2804 #if HAVE_DECL_CURLOPT_LOGIN_OPTIONS
2805 SETOPT_STRING( LOGIN_OPTIONS)
2806 #endif
2808 #if HAVE_DECL_CURLOPT_CONNECT_TO
2809 SETOPT_SLIST( CONNECT_TO)
2810 #endif
2812 #if HAVE_DECL_CURLOPT_POSTREDIR
2814 static int curlPostRedir_table[] = {
2815 CURL_REDIR_POST_ALL,
2816 #if defined(CURL_REDIR_POST_301)
2817 CURL_REDIR_POST_301,
2818 #else
2820 #endif
2821 #if defined(CURL_REDIR_POST_302)
2822 CURL_REDIR_POST_302,
2823 #else
2825 #endif
2826 #if defined(CURL_REDIR_POST_303)
2827 CURL_REDIR_POST_303,
2828 #else
2830 #endif
2833 static void handle_POSTREDIR(Connection *conn, value option)
2835 CAMLparam1(option);
2836 CURLcode result = CURLE_OK;
2837 long bitmask = caml_convert_flag_list(option, curlPostRedir_table);
2839 result = curl_easy_setopt(conn->handle,
2840 CURLOPT_POSTREDIR,
2841 bitmask);
2843 if (result != CURLE_OK)
2844 raiseError(conn, result);
2846 CAMLreturn0;
2848 #endif
2850 SETOPT_VAL( SSH_KNOWNHOSTS, String_val)
2853 ** curl_easy_setopt helper function
2856 #define MAP(name) { handle_ ## name, "CURLOPT_"#name, Ocaml_##name }
2857 #define MAP_NO(name) { NULL, "CURLOPT_"#name , Ocaml_##name }
2858 #define IMM(name) { handle_ ## name, "CURLOPT_"#name, (OcamlValue)(-1) }
2859 #define IMM_NO(name) { NULL, "CURLOPT_"#name , (OcamlValue)(-1) }
2861 CURLOptionMapping implementedOptionMap[] =
2863 MAP(WRITEFUNCTION),
2864 MAP(READFUNCTION),
2865 IMM(INFILESIZE),
2866 MAP(URL),
2867 MAP(PROXY),
2868 IMM(PROXYPORT),
2869 IMM(HTTPPROXYTUNNEL),
2870 IMM(VERBOSE),
2871 IMM(HEADER),
2872 IMM(NOPROGRESS),
2873 #if HAVE_DECL_CURLOPT_NOSIGNAL
2874 IMM(NOSIGNAL),
2875 #else
2876 IMM_NO(NOSIGNAL),
2877 #endif
2878 IMM(NOBODY),
2879 IMM(FAILONERROR),
2880 IMM(UPLOAD),
2881 IMM(POST),
2882 IMM(FTPLISTONLY),
2883 IMM(FTPAPPEND),
2884 IMM(NETRC),
2885 #if HAVE_DECL_CURLOPT_ENCODING
2886 IMM(ENCODING),
2887 #else
2888 IMM_NO(ENCODING),
2889 #endif
2890 IMM(FOLLOWLOCATION),
2891 IMM(TRANSFERTEXT),
2892 IMM(PUT),
2893 MAP(USERPWD),
2894 MAP(PROXYUSERPWD),
2895 MAP(RANGE),
2896 IMM(ERRORBUFFER), /* mutable buffer, as output value, do not duplicate */
2897 IMM(TIMEOUT),
2898 MAP(POSTFIELDS),
2899 IMM(POSTFIELDSIZE),
2900 MAP(REFERER),
2901 MAP(USERAGENT),
2902 MAP(FTPPORT),
2903 IMM(LOW_SPEED_LIMIT),
2904 IMM(LOW_SPEED_TIME),
2905 IMM(RESUME_FROM),
2906 MAP(COOKIE),
2907 MAP(HTTPHEADER),
2908 MAP(HTTPPOST),
2909 MAP(SSLCERT),
2910 MAP(SSLCERTTYPE),
2911 MAP(SSLCERTPASSWD),
2912 MAP(SSLKEY),
2913 MAP(SSLKEYTYPE),
2914 MAP(SSLKEYPASSWD),
2915 MAP(SSLENGINE),
2916 IMM(SSLENGINE_DEFAULT),
2917 IMM(CRLF),
2918 MAP(QUOTE),
2919 MAP(POSTQUOTE),
2920 MAP(HEADERFUNCTION),
2921 MAP(COOKIEFILE),
2922 IMM(SSLVERSION),
2923 IMM(TIMECONDITION),
2924 IMM(TIMEVALUE),
2925 MAP(CUSTOMREQUEST),
2926 MAP(INTERFACE),
2927 IMM(KRB4LEVEL),
2928 MAP(PROGRESSFUNCTION),
2929 IMM(SSL_VERIFYPEER),
2930 MAP(CAINFO),
2931 MAP(CAPATH),
2932 IMM(FILETIME),
2933 IMM(MAXREDIRS),
2934 IMM(MAXCONNECTS),
2935 IMM(CLOSEPOLICY),
2936 IMM(FRESH_CONNECT),
2937 IMM(FORBID_REUSE),
2938 MAP(RANDOM_FILE),
2939 MAP(EGDSOCKET),
2940 IMM(CONNECTTIMEOUT),
2941 IMM(HTTPGET),
2942 IMM(SSL_VERIFYHOST),
2943 MAP(COOKIEJAR),
2944 MAP(SSL_CIPHER_LIST),
2945 IMM(HTTP_VERSION),
2946 IMM(FTP_USE_EPSV),
2947 IMM(DNS_CACHE_TIMEOUT),
2948 IMM(DNS_USE_GLOBAL_CACHE),
2949 MAP(DEBUGFUNCTION),
2950 MAP(PRIVATE),
2951 #if HAVE_DECL_CURLOPT_HTTP200ALIASES
2952 MAP(HTTP200ALIASES),
2953 #else
2954 MAP_NO(HTTP200ALIASES),
2955 #endif
2956 #if HAVE_DECL_CURLOPT_UNRESTRICTED_AUTH
2957 IMM(UNRESTRICTED_AUTH),
2958 #else
2959 IMM_NO(UNRESTRICTED_AUTH),
2960 #endif
2961 #if HAVE_DECL_CURLOPT_FTP_USE_EPRT
2962 IMM(FTP_USE_EPRT),
2963 #else
2964 IMM_NO(FTP_USE_EPRT),
2965 #endif
2966 #if HAVE_DECL_CURLOPT_HTTPAUTH
2967 IMM(HTTPAUTH),
2968 #else
2969 IMM_NO(HTTPAUTH),
2970 #endif
2971 #if HAVE_DECL_CURLOPT_FTP_CREATE_MISSING_DIRS
2972 IMM(FTP_CREATE_MISSING_DIRS),
2973 #else
2974 IMM_NO(FTP_CREATE_MISSING_DIRS),
2975 #endif
2976 #if HAVE_DECL_CURLOPT_PROXYAUTH
2977 IMM(PROXYAUTH),
2978 #else
2979 IMM_NO(PROXYAUTH),
2980 #endif
2981 #if HAVE_DECL_CURLOPT_FTP_RESPONSE_TIMEOUT
2982 IMM(FTP_RESPONSE_TIMEOUT),
2983 #else
2984 IMM_NO(FTP_RESPONSE_TIMEOUT),
2985 #endif
2986 #if HAVE_DECL_CURLOPT_IPRESOLVE
2987 IMM(IPRESOLVE),
2988 #else
2989 IMM_NO(IPRESOLVE),
2990 #endif
2991 #if HAVE_DECL_CURLOPT_MAXFILESIZE
2992 IMM(MAXFILESIZE),
2993 #else
2994 IMM_NO(MAXFILESIZE),
2995 #endif
2996 #if HAVE_DECL_CURLOPT_INFILESIZE_LARGE
2997 IMM(INFILESIZE_LARGE),
2998 #else
2999 IMM_NO(INFILESIZE_LARGE),
3000 #endif
3001 #if HAVE_DECL_CURLOPT_RESUME_FROM_LARGE
3002 IMM(RESUME_FROM_LARGE),
3003 #else
3004 IMM_NO(RESUME_FROM_LARGE),
3005 #endif
3006 #if HAVE_DECL_CURLOPT_MAXFILESIZE_LARGE
3007 IMM(MAXFILESIZE_LARGE),
3008 #else
3009 IMM_NO(MAXFILESIZE_LARGE),
3010 #endif
3011 #if HAVE_DECL_CURLOPT_NETRC_FILE
3012 MAP(NETRC_FILE),
3013 #else
3014 MAP_NO(NETRC_FILE),
3015 #endif
3016 #if HAVE_DECL_CURLOPT_FTP_SSL
3017 IMM(FTP_SSL),
3018 #else
3019 IMM_NO(FTP_SSL),
3020 #endif
3021 #if HAVE_DECL_CURLOPT_POSTFIELDSIZE_LARGE
3022 IMM(POSTFIELDSIZE_LARGE),
3023 #else
3024 IMM_NO(POSTFIELDSIZE_LARGE),
3025 #endif
3026 #if HAVE_DECL_CURLOPT_TCP_NODELAY
3027 IMM(TCP_NODELAY),
3028 #else
3029 IMM_NO(TCP_NODELAY),
3030 #endif
3031 #if HAVE_DECL_CURLOPT_FTPSSLAUTH
3032 IMM(FTPSSLAUTH),
3033 #else
3034 IMM_NO(FTPSSLAUTH),
3035 #endif
3036 #if HAVE_DECL_CURLOPT_IOCTLFUNCTION
3037 MAP(IOCTLFUNCTION),
3038 #else
3039 MAP_NO(IOCTLFUNCTION),
3040 #endif
3041 #if HAVE_DECL_CURLOPT_FTP_ACCOUNT
3042 MAP(FTP_ACCOUNT),
3043 #else
3044 MAP_NO(FTP_ACCOUNT),
3045 #endif
3046 #if HAVE_DECL_CURLOPT_COOKIELIST
3047 MAP(COOKIELIST),
3048 #else
3049 MAP_NO(COOKIELIST),
3050 #endif
3051 #if HAVE_DECL_CURLOPT_IGNORE_CONTENT_LENGTH
3052 IMM(IGNORE_CONTENT_LENGTH),
3053 #else
3054 IMM_NO(IGNORE_CONTENT_LENGTH),
3055 #endif
3056 #if HAVE_DECL_CURLOPT_FTP_SKIP_PASV_IP
3057 IMM(FTP_SKIP_PASV_IP),
3058 #else
3059 IMM_NO(FTP_SKIP_PASV_IP),
3060 #endif
3061 #if HAVE_DECL_CURLOPT_FTP_FILEMETHOD
3062 IMM(FTP_FILEMETHOD),
3063 #else
3064 IMM_NO(FTP_FILEMETHOD),
3065 #endif
3066 #if HAVE_DECL_CURLOPT_LOCALPORT
3067 IMM(LOCALPORT),
3068 #else
3069 IMM_NO(LOCALPORT),
3070 #endif
3071 #if HAVE_DECL_CURLOPT_LOCALPORTRANGE
3072 IMM(LOCALPORTRANGE),
3073 #else
3074 IMM_NO(LOCALPORTRANGE),
3075 #endif
3076 #if HAVE_DECL_CURLOPT_CONNECT_ONLY
3077 IMM(CONNECT_ONLY),
3078 #else
3079 IMM_NO(CONNECT_ONLY),
3080 #endif
3081 #if HAVE_DECL_CURLOPT_MAX_SEND_SPEED_LARGE
3082 IMM(MAX_SEND_SPEED_LARGE),
3083 #else
3084 IMM_NO(MAX_SEND_SPEED_LARGE),
3085 #endif
3086 #if HAVE_DECL_CURLOPT_MAX_RECV_SPEED_LARGE
3087 IMM(MAX_RECV_SPEED_LARGE),
3088 #else
3089 IMM_NO(MAX_RECV_SPEED_LARGE),
3090 #endif
3091 #if HAVE_DECL_CURLOPT_FTP_ALTERNATIVE_TO_USER
3092 MAP(FTP_ALTERNATIVE_TO_USER),
3093 #else
3094 MAP_NO(FTP_ALTERNATIVE_TO_USER),
3095 #endif
3096 #if HAVE_DECL_CURLOPT_SSL_SESSIONID_CACHE
3097 IMM(SSL_SESSIONID_CACHE),
3098 #else
3099 IMM_NO(SSL_SESSIONID_CACHE),
3100 #endif
3101 #if HAVE_DECL_CURLOPT_SSH_AUTH_TYPES
3102 IMM(SSH_AUTH_TYPES),
3103 #else
3104 IMM_NO(SSH_AUTH_TYPES),
3105 #endif
3106 #if HAVE_DECL_CURLOPT_SSH_PUBLIC_KEYFILE
3107 MAP(SSH_PUBLIC_KEYFILE),
3108 #else
3109 MAP_NO(SSH_PUBLIC_KEYFILE),
3110 #endif
3111 #if HAVE_DECL_CURLOPT_SSH_PRIVATE_KEYFILE
3112 MAP(SSH_PRIVATE_KEYFILE),
3113 #else
3114 MAP_NO(SSH_PRIVATE_KEYFILE),
3115 #endif
3116 #if HAVE_DECL_CURLOPT_FTP_SSL_CCC
3117 IMM(FTP_SSL_CCC),
3118 #else
3119 IMM_NO(FTP_SSL_CCC),
3120 #endif
3121 #if HAVE_DECL_CURLOPT_TIMEOUT_MS
3122 IMM(TIMEOUT_MS),
3123 #else
3124 IMM_NO(TIMEOUT_MS),
3125 #endif
3126 #if HAVE_DECL_CURLOPT_CONNECTTIMEOUT_MS
3127 IMM(CONNECTTIMEOUT_MS),
3128 #else
3129 IMM_NO(CONNECTTIMEOUT_MS),
3130 #endif
3131 #if HAVE_DECL_CURLOPT_HTTP_TRANSFER_DECODING
3132 IMM(HTTP_TRANSFER_DECODING),
3133 #else
3134 IMM_NO(HTTP_TRANSFER_DECODING),
3135 #endif
3136 #if HAVE_DECL_CURLOPT_HTTP_CONTENT_DECODING
3137 IMM(HTTP_CONTENT_DECODING),
3138 #else
3139 IMM_NO(HTTP_CONTENT_DECODING),
3140 #endif
3141 #if HAVE_DECL_CURLOPT_NEW_FILE_PERMS
3142 IMM(NEW_FILE_PERMS),
3143 #else
3144 IMM_NO(NEW_FILE_PERMS),
3145 #endif
3146 #if HAVE_DECL_CURLOPT_NEW_DIRECTORY_PERMS
3147 IMM(NEW_DIRECTORY_PERMS),
3148 #else
3149 IMM_NO(NEW_DIRECTORY_PERMS),
3150 #endif
3151 #if HAVE_DECL_CURLOPT_POST301
3152 IMM(POST301),
3153 #else
3154 IMM_NO(POST301),
3155 #endif
3156 #if HAVE_DECL_CURLOPT_SSH_HOST_PUBLIC_KEY_MD5
3157 MAP(SSH_HOST_PUBLIC_KEY_MD5),
3158 #else
3159 MAP_NO(SSH_HOST_PUBLIC_KEY_MD5),
3160 #endif
3161 #if HAVE_DECL_CURLOPT_COPYPOSTFIELDS
3162 MAP(COPYPOSTFIELDS),
3163 #else
3164 MAP_NO(COPYPOSTFIELDS),
3165 #endif
3166 #if HAVE_DECL_CURLOPT_PROXY_TRANSFER_MODE
3167 IMM(PROXY_TRANSFER_MODE),
3168 #else
3169 IMM_NO(PROXY_TRANSFER_MODE),
3170 #endif
3171 #if HAVE_DECL_CURLOPT_SEEKFUNCTION
3172 MAP(SEEKFUNCTION),
3173 #else
3174 MAP_NO(SEEKFUNCTION),
3175 #endif
3176 #if HAVE_DECL_CURLOPT_AUTOREFERER
3177 IMM(AUTOREFERER),
3178 #else
3179 IMM_NO(AUTOREFERER),
3180 #endif
3181 #if HAVE_DECL_CURLOPT_OPENSOCKETFUNCTION
3182 MAP(OPENSOCKETFUNCTION),
3183 #else
3184 MAP_NO(OPENSOCKETFUNCTION),
3185 #endif
3186 #if HAVE_DECL_CURLOPT_PROXYTYPE
3187 IMM(PROXYTYPE),
3188 #else
3189 IMM_NO(PROXYTYPE),
3190 #endif
3191 #if HAVE_DECL_CURLOPT_PROTOCOLS
3192 IMM(PROTOCOLS),
3193 #else
3194 IMM_NO(PROTOCOLS),
3195 #endif
3196 #if HAVE_DECL_CURLOPT_REDIR_PROTOCOLS
3197 IMM(REDIR_PROTOCOLS),
3198 #else
3199 IMM_NO(REDIR_PROTOCOLS),
3200 #endif
3201 #if HAVE_DECL_CURLOPT_RESOLVE
3202 MAP(RESOLVE),
3203 #else
3204 MAP_NO(RESOLVE),
3205 #endif
3206 #if HAVE_DECL_CURLOPT_DNS_SERVERS
3207 MAP(DNS_SERVERS),
3208 #else
3209 MAP_NO(DNS_SERVERS),
3210 #endif
3211 #if HAVE_DECL_CURLOPT_MAIL_FROM
3212 MAP(MAIL_FROM),
3213 #else
3214 MAP_NO(MAIL_FROM),
3215 #endif
3216 #if HAVE_DECL_CURLOPT_MAIL_RCPT
3217 MAP(MAIL_RCPT),
3218 #else
3219 MAP_NO(MAIL_RCPT),
3220 #endif
3221 #if HAVE_DECL_CURLOPT_PIPEWAIT
3222 IMM(PIPEWAIT),
3223 #else
3224 IMM_NO(PIPEWAIT),
3225 #endif
3226 #if HAVE_DECL_CURLOPT_CERTINFO
3227 IMM(CERTINFO),
3228 #else
3229 IMM_NO(CERTINFO),
3230 #endif
3231 #if HAVE_DECL_CURLOPT_USERNAME
3232 MAP(USERNAME),
3233 #else
3234 MAP_NO(USERNAME),
3235 #endif
3236 #if HAVE_DECL_CURLOPT_PASSWORD
3237 MAP(PASSWORD),
3238 #else
3239 MAP_NO(PASSWORD),
3240 #endif
3241 #if HAVE_DECL_CURLOPT_LOGIN_OPTIONS
3242 MAP(LOGIN_OPTIONS),
3243 #else
3244 MAP_NO(LOGIN_OPTIONS),
3245 #endif
3246 #if HAVE_DECL_CURLOPT_CONNECT_TO
3247 MAP(CONNECT_TO),
3248 #else
3249 MAP_NO(CONNECT_TO),
3250 #endif
3251 #if HAVE_DECL_CURLOPT_POSTREDIR
3252 MAP(POSTREDIR),
3253 #else
3254 MAP_NO(POSTREDIR),
3255 #endif
3256 #if HAVE_DECL_CURLOPT_MIMEPOST
3257 MAP(MIMEPOST),
3258 #else
3259 MAP_NO(MIMEPOST),
3260 #endif
3261 IMM(SSH_KNOWNHOSTS),
3262 MAP(SSH_KEYFUNCTION),
3265 static Connection *duplicateConnection(Connection *original)
3267 Connection *connection = NULL;
3268 CURL* h = NULL;
3269 size_t i = 0;
3270 CURLOptionMapping* self = NULL;
3272 caml_enter_blocking_section();
3273 h = curl_easy_duphandle(original->handle);
3274 caml_leave_blocking_section();
3276 connection = allocConnection(h);
3278 for (i = 0; i < sizeof(implementedOptionMap)/sizeof(CURLOptionMapping); i++)
3280 self = &implementedOptionMap[i];
3281 if (-1 == self->ocamlValue) continue;
3282 if (self->optionHandler && (Field(original->ocamlValues, self->ocamlValue) != Val_unit))
3284 self->optionHandler(connection, Field(original->ocamlValues, self->ocamlValue));
3288 return connection;
3291 value caml_curl_easy_setopt(value conn, value option)
3293 CAMLparam2(conn, option);
3294 CAMLlocal1(data);
3295 Connection *connection = Connection_val(conn);
3296 CURLOptionMapping* thisOption = NULL;
3297 static value* exception = NULL;
3299 checkConnection(connection);
3301 data = Field(option, 0);
3303 if (Tag_val(option) < sizeof(implementedOptionMap)/sizeof(CURLOptionMapping))
3305 thisOption = &implementedOptionMap[Tag_val(option)];
3306 if (thisOption->optionHandler)
3308 thisOption->optionHandler(connection, data);
3310 else
3312 if (NULL == exception)
3314 exception = caml_named_value("Curl.NotImplemented");
3315 if (NULL == exception) caml_invalid_argument("Curl.NotImplemented");
3318 caml_raise_with_string(*exception, thisOption->name);
3321 else
3323 caml_failwith("Invalid CURLOPT Option");
3326 CAMLreturn(Val_unit);
3330 ** curl_easy_perform helper function
3333 value caml_curl_easy_perform(value conn)
3335 CAMLparam1(conn);
3336 CURLcode result = CURLE_OK;
3337 Connection *connection = Connection_val(conn);
3339 checkConnection(connection);
3341 caml_enter_blocking_section();
3342 result = curl_easy_perform(connection->handle);
3343 caml_leave_blocking_section();
3345 if (result != CURLE_OK)
3346 raiseError(connection, result);
3348 CAMLreturn(Val_unit);
3352 ** curl_easy_cleanup helper function
3355 value caml_curl_easy_cleanup(value conn)
3357 CAMLparam1(conn);
3358 Connection *connection = Connection_val(conn);
3360 checkConnection(connection);
3362 removeConnection(connection, 0);
3364 CAMLreturn(Val_unit);
3368 ** curl_easy_duphandle helper function
3371 value caml_curl_easy_duphandle(value conn)
3373 CAMLparam1(conn);
3374 CAMLlocal1(result);
3375 Connection *connection = Connection_val(conn);
3377 checkConnection(connection);
3379 result = caml_curl_alloc(duplicateConnection(connection));
3381 CAMLreturn(result);
3385 ** curl_easy_getinfo helper function
3388 enum GetInfoResultType {
3389 StringValue, LongValue, DoubleValue, StringListValue, StringListListValue,
3390 OCamlValue, /* keep last - no matching OCaml CURLINFO_ constructor */
3393 value convertStringList(struct curl_slist *p)
3395 CAMLparam0();
3396 CAMLlocal3(result, current, next);
3398 result = Val_emptylist;
3399 current = Val_emptylist;
3400 next = Val_emptylist;
3402 while (p != NULL)
3404 next = caml_alloc_tuple(2);
3405 Store_field(next, 0, caml_copy_string(p->data));
3406 Store_field(next, 1, Val_emptylist);
3408 if (result == Val_emptylist)
3409 result = next;
3411 if (current != Val_emptylist)
3412 Store_field(current, 1, next);
3414 current = next;
3416 p = p->next;
3419 CAMLreturn(result);
3422 value caml_curl_easy_getinfo(value conn, value option)
3424 CAMLparam2(conn, option);
3425 CAMLlocal3(result, current, next);
3426 CURLcode curlResult;
3427 Connection *connection = Connection_val(conn);
3428 enum GetInfoResultType resultType;
3429 char *strValue = NULL;
3430 double doubleValue;
3431 long longValue;
3432 struct curl_slist *stringListValue = NULL;
3433 #if HAVE_DECL_CURLINFO_CERTINFO
3434 int i;
3435 union {
3436 struct curl_slist *to_info;
3437 struct curl_certinfo *to_certinfo;
3438 } ptr;
3439 #endif
3441 checkConnection(connection);
3443 switch(Long_val(option))
3445 #if HAVE_DECL_CURLINFO_EFFECTIVE_URL
3446 case 0: /* CURLINFO_EFFECTIVE_URL */
3447 resultType = StringValue;
3449 curlResult = curl_easy_getinfo(connection->handle,
3450 CURLINFO_EFFECTIVE_URL,
3451 &strValue);
3452 break;
3453 #else
3454 #pragma message("libcurl does not provide CURLINFO_EFFECTIVE_URL")
3455 #endif
3457 #if HAVE_DECL_CURLINFO_RESPONSE_CODE || HAVE_DECL_CURLINFO_HTTP_CODE
3458 case 1: /* CURLINFO_HTTP_CODE */
3459 case 2: /* CURLINFO_RESPONSE_CODE */
3460 #if HAVE_DECL_CURLINFO_RESPONSE_CODE
3461 resultType = LongValue;
3463 curlResult = curl_easy_getinfo(connection->handle,
3464 CURLINFO_RESPONSE_CODE,
3465 &longValue);
3466 #else
3467 resultType = LongValue;
3469 curlResult = curl_easy_getinfo(connection->handle,
3470 CURLINFO_HTTP_CODE,
3471 &longValue);
3472 #endif
3473 break;
3474 #endif
3476 #if HAVE_DECL_CURLINFO_TOTAL_TIME
3477 case 3: /* CURLINFO_TOTAL_TIME */
3478 resultType = DoubleValue;
3480 curlResult = curl_easy_getinfo(connection->handle,
3481 CURLINFO_TOTAL_TIME,
3482 &doubleValue);
3483 break;
3484 #endif
3486 #if HAVE_DECL_CURLINFO_NAMELOOKUP_TIME
3487 case 4: /* CURLINFO_NAMELOOKUP_TIME */
3488 resultType = DoubleValue;
3490 curlResult = curl_easy_getinfo(connection->handle,
3491 CURLINFO_NAMELOOKUP_TIME,
3492 &doubleValue);
3493 break;
3494 #endif
3496 #if HAVE_DECL_CURLINFO_CONNECT_TIME
3497 case 5: /* CURLINFO_CONNECT_TIME */
3498 resultType = DoubleValue;
3500 curlResult = curl_easy_getinfo(connection->handle,
3501 CURLINFO_CONNECT_TIME,
3502 &doubleValue);
3503 break;
3504 #endif
3506 #if HAVE_DECL_CURLINFO_PRETRANSFER_TIME
3507 case 6: /* CURLINFO_PRETRANSFER_TIME */
3508 resultType = DoubleValue;
3510 curlResult = curl_easy_getinfo(connection->handle,
3511 CURLINFO_PRETRANSFER_TIME,
3512 &doubleValue);
3513 break;
3514 #endif
3516 #if HAVE_DECL_CURLINFO_SIZE_UPLOAD
3517 case 7: /* CURLINFO_SIZE_UPLOAD */
3518 resultType = DoubleValue;
3520 curlResult = curl_easy_getinfo(connection->handle,
3521 CURLINFO_SIZE_UPLOAD,
3522 &doubleValue);
3523 break;
3524 #endif
3526 #if HAVE_DECL_CURLINFO_SIZE_DOWNLOAD
3527 case 8: /* CURLINFO_SIZE_DOWNLOAD */
3528 resultType = DoubleValue;
3530 curlResult = curl_easy_getinfo(connection->handle,
3531 CURLINFO_SIZE_DOWNLOAD,
3532 &doubleValue);
3533 break;
3534 #endif
3536 #if HAVE_DECL_CURLINFO_SPEED_DOWNLOAD
3537 case 9: /* CURLINFO_SPEED_DOWNLOAD */
3538 resultType = DoubleValue;
3540 curlResult = curl_easy_getinfo(connection->handle,
3541 CURLINFO_SPEED_DOWNLOAD,
3542 &doubleValue);
3543 break;
3544 #endif
3546 #if HAVE_DECL_CURLINFO_SPEED_UPLOAD
3547 case 10: /* CURLINFO_SPEED_UPLOAD */
3548 resultType = DoubleValue;
3550 curlResult = curl_easy_getinfo(connection->handle,
3551 CURLINFO_SPEED_UPLOAD,
3552 &doubleValue);
3553 break;
3555 #endif
3557 #if HAVE_DECL_CURLINFO_HEADER_SIZE
3558 case 11: /* CURLINFO_HEADER_SIZE */
3559 resultType = LongValue;
3561 curlResult = curl_easy_getinfo(connection->handle,
3562 CURLINFO_HEADER_SIZE,
3563 &longValue);
3564 break;
3565 #endif
3567 #if HAVE_DECL_CURLINFO_REQUEST_SIZE
3568 case 12: /* CURLINFO_REQUEST_SIZE */
3569 resultType = LongValue;
3571 curlResult = curl_easy_getinfo(connection->handle,
3572 CURLINFO_REQUEST_SIZE,
3573 &longValue);
3574 break;
3575 #endif
3577 #if HAVE_DECL_CURLINFO_SSL_VERIFYRESULT
3578 case 13: /* CURLINFO_SSL_VERIFYRESULT */
3579 resultType = LongValue;
3581 curlResult = curl_easy_getinfo(connection->handle,
3582 CURLINFO_SSL_VERIFYRESULT,
3583 &longValue);
3584 break;
3585 #endif
3587 #if HAVE_DECL_CURLINFO_FILETIME
3588 case 14: /* CURLINFO_FILETIME */
3589 resultType = DoubleValue;
3591 curlResult = curl_easy_getinfo(connection->handle,
3592 CURLINFO_FILETIME,
3593 &longValue);
3595 doubleValue = longValue;
3596 break;
3597 #endif
3599 #if HAVE_DECL_CURLINFO_CONTENT_LENGTH_DOWNLOAD
3600 case 15: /* CURLINFO_CONTENT_LENGTH_DOWNLOAD */
3601 resultType = DoubleValue;
3603 curlResult = curl_easy_getinfo(connection->handle,
3604 CURLINFO_CONTENT_LENGTH_DOWNLOAD,
3605 &doubleValue);
3606 break;
3607 #endif
3609 #if HAVE_DECL_CURLINFO_CONTENT_LENGTH_UPLOAD
3610 case 16: /* CURLINFO_CONTENT_LENGTH_UPLOAD */
3611 resultType = DoubleValue;
3613 curlResult = curl_easy_getinfo(connection->handle,
3614 CURLINFO_CONTENT_LENGTH_UPLOAD,
3615 &doubleValue);
3616 break;
3617 #endif
3619 #if HAVE_DECL_CURLINFO_STARTTRANSFER_TIME
3620 case 17: /* CURLINFO_STARTTRANSFER_TIME */
3621 resultType = DoubleValue;
3623 curlResult = curl_easy_getinfo(connection->handle,
3624 CURLINFO_STARTTRANSFER_TIME,
3625 &doubleValue);
3626 break;
3627 #endif
3629 #if HAVE_DECL_CURLINFO_CONTENT_TYPE
3630 case 18: /* CURLINFO_CONTENT_TYPE */
3631 resultType = StringValue;
3633 curlResult = curl_easy_getinfo(connection->handle,
3634 CURLINFO_CONTENT_TYPE,
3635 &strValue);
3636 break;
3637 #endif
3639 #if HAVE_DECL_CURLINFO_REDIRECT_TIME
3640 case 19: /* CURLINFO_REDIRECT_TIME */
3641 resultType = DoubleValue;
3643 curlResult = curl_easy_getinfo(connection->handle,
3644 CURLINFO_REDIRECT_TIME,
3645 &doubleValue);
3646 break;
3647 #endif
3649 #if HAVE_DECL_CURLINFO_REDIRECT_COUNT
3650 case 20: /* CURLINFO_REDIRECT_COUNT */
3651 resultType = LongValue;
3653 curlResult = curl_easy_getinfo(connection->handle,
3654 CURLINFO_REDIRECT_COUNT,
3655 &longValue);
3656 break;
3657 #endif
3659 case 21: /* CURLINFO_PRIVATE */
3660 resultType = OCamlValue;
3661 curlResult = CURLE_OK;
3662 result = caml_alloc(1, StringValue);
3663 Store_field(result, 0, Field(connection->ocamlValues, Ocaml_PRIVATE));
3664 break;
3666 #if HAVE_DECL_CURLINFO_HTTP_CONNECTCODE
3667 case 22: /* CURLINFO_HTTP_CONNECTCODE */
3668 resultType = LongValue;
3670 curlResult = curl_easy_getinfo(connection->handle,
3671 CURLINFO_HTTP_CONNECTCODE,
3672 &longValue);
3673 break;
3674 #endif
3676 #if HAVE_DECL_CURLINFO_HTTPAUTH_AVAIL
3677 case 23: /* CURLINFO_HTTPAUTH_AVAIL */
3678 resultType = LongValue;
3680 curlResult = curl_easy_getinfo(connection->handle,
3681 CURLINFO_HTTPAUTH_AVAIL,
3682 &longValue);
3683 break;
3684 #endif
3686 #if HAVE_DECL_CURLINFO_PROXYAUTH_AVAIL
3687 case 24: /* CURLINFO_PROXYAUTH_AVAIL */
3688 resultType = LongValue;
3690 curlResult = curl_easy_getinfo(connection->handle,
3691 CURLINFO_PROXYAUTH_AVAIL,
3692 &longValue);
3693 break;
3694 #endif
3696 #if HAVE_DECL_CURLINFO_OS_ERRNO
3697 case 25: /* CURLINFO_OS_ERRNO */
3698 resultType = LongValue;
3700 curlResult = curl_easy_getinfo(connection->handle,
3701 CURLINFO_OS_ERRNO,
3702 &longValue);
3703 break;
3704 #endif
3706 #if HAVE_DECL_CURLINFO_NUM_CONNECTS
3707 case 26: /* CURLINFO_NUM_CONNECTS */
3708 resultType = LongValue;
3710 curlResult = curl_easy_getinfo(connection->handle,
3711 CURLINFO_NUM_CONNECTS,
3712 &longValue);
3713 break;
3714 #endif
3716 #if HAVE_DECL_CURLINFO_SSL_ENGINES
3717 case 27: /* CURLINFO_SSL_ENGINES */
3718 resultType = StringListValue;
3720 curlResult = curl_easy_getinfo(connection->handle,
3721 CURLINFO_SSL_ENGINES,
3722 &stringListValue);
3723 break;
3724 #endif
3726 #if HAVE_DECL_CURLINFO_COOKIELIST
3727 case 28: /* CURLINFO_COOKIELIST */
3728 resultType = StringListValue;
3730 curlResult = curl_easy_getinfo(connection->handle,
3731 CURLINFO_COOKIELIST,
3732 &stringListValue);
3733 break;
3734 #endif
3736 #if HAVE_DECL_CURLINFO_LASTSOCKET
3737 case 29: /* CURLINFO_LASTSOCKET */
3738 resultType = LongValue;
3740 curlResult = curl_easy_getinfo(connection->handle,
3741 CURLINFO_LASTSOCKET,
3742 &longValue);
3743 break;
3744 #endif
3746 #if HAVE_DECL_CURLINFO_FTP_ENTRY_PATH
3747 case 30: /* CURLINFO_FTP_ENTRY_PATH */
3748 resultType = StringValue;
3750 curlResult = curl_easy_getinfo(connection->handle,
3751 CURLINFO_FTP_ENTRY_PATH,
3752 &strValue);
3753 break;
3754 #endif
3756 #if HAVE_DECL_CURLINFO_REDIRECT_URL
3757 case 31: /* CURLINFO_REDIRECT_URL */
3758 resultType = StringValue;
3760 curlResult = curl_easy_getinfo(connection->handle,
3761 CURLINFO_REDIRECT_URL,
3762 &strValue);
3763 break;
3764 #else
3765 #pragma message("libcurl does not provide CURLINFO_REDIRECT_URL")
3766 #endif
3768 #if HAVE_DECL_CURLINFO_PRIMARY_IP
3769 case 32: /* CURLINFO_PRIMARY_IP */
3770 resultType = StringValue;
3772 curlResult = curl_easy_getinfo(connection->handle,
3773 CURLINFO_PRIMARY_IP,
3774 &strValue);
3775 break;
3776 #else
3777 #pragma message("libcurl does not provide CURLINFO_PRIMARY_IP")
3778 #endif
3780 #if HAVE_DECL_CURLINFO_LOCAL_IP
3781 case 33: /* CURLINFO_LOCAL_IP */
3782 resultType = StringValue;
3784 curlResult = curl_easy_getinfo(connection->handle,
3785 CURLINFO_LOCAL_IP,
3786 &strValue);
3787 break;
3788 #else
3789 #pragma message("libcurl does not provide CURLINFO_LOCAL_IP")
3790 #endif
3792 #if HAVE_DECL_CURLINFO_LOCAL_PORT
3793 case 34: /* CURLINFO_LOCAL_PORT */
3794 resultType = LongValue;
3796 curlResult = curl_easy_getinfo(connection->handle,
3797 CURLINFO_LOCAL_PORT,
3798 &longValue);
3799 break;
3800 #else
3801 #pragma message("libcurl does not provide CURLINFO_LOCAL_PORT")
3802 #endif
3804 #if HAVE_DECL_CURLINFO_CONDITION_UNMET
3805 case 35: /* CURLINFO_CONDITION_UNMET */
3806 resultType = LongValue;
3808 curlResult = curl_easy_getinfo(connection->handle,
3809 CURLINFO_CONDITION_UNMET,
3810 &longValue);
3811 break;
3812 #else
3813 #pragma message("libcurl does not provide CURLINFO_CONDITION_UNMET")
3814 #endif
3815 #if HAVE_DECL_CURLINFO_CERTINFO
3816 case 36: /* CURLINFO_CERTINFO */
3817 resultType = StringListListValue;
3818 ptr.to_info = NULL;
3819 curlResult = curl_easy_getinfo(connection->handle,
3820 CURLINFO_CERTINFO,
3821 &ptr.to_info);
3823 result = Val_emptylist;
3824 current = Val_emptylist;
3825 next = Val_emptylist;
3827 if (curlResult != CURLE_OK || !ptr.to_info)
3828 break;
3830 for (i = 0; i < ptr.to_certinfo->num_of_certs; i++) {
3831 next = caml_alloc_tuple(2);
3832 Store_field(next, 0, convertStringList(ptr.to_certinfo->certinfo[i]));
3833 Store_field(next, 1, current);
3834 current = next;
3836 break;
3837 #else
3838 #pragma message("libcurl does not provide CURLINFO_CERTINFO")
3839 #endif
3840 default:
3841 caml_failwith("Invalid CURLINFO Option");
3842 break;
3845 if (curlResult != CURLE_OK)
3846 raiseError(connection, curlResult);
3848 switch (resultType)
3850 case StringValue:
3851 result = caml_alloc(1, StringValue);
3852 Store_field(result, 0, caml_copy_string(strValue?strValue:""));
3853 break;
3855 case LongValue:
3856 result = caml_alloc(1, LongValue);
3857 Store_field(result, 0, Val_long(longValue));
3858 break;
3860 case DoubleValue:
3861 result = caml_alloc(1, DoubleValue);
3862 Store_field(result, 0, caml_copy_double(doubleValue));
3863 break;
3865 case StringListValue:
3866 result = caml_alloc(1, StringListValue);
3867 Store_field(result, 0, convertStringList(stringListValue));
3868 curl_slist_free_all(stringListValue);
3869 break;
3871 case StringListListValue:
3872 result = caml_alloc(1, StringListListValue);
3873 Store_field(result, 0, current);
3874 break;
3876 case OCamlValue:
3877 break;
3880 CAMLreturn(result);
3884 ** curl_escape helper function
3887 value caml_curl_escape(value str)
3889 CAMLparam1(str);
3890 CAMLlocal1(result);
3891 char *curlResult;
3893 curlResult = curl_escape(String_val(str), caml_string_length(str));
3894 result = caml_copy_string(curlResult);
3895 free(curlResult);
3897 CAMLreturn(result);
3901 ** curl_unescape helper function
3904 value caml_curl_unescape(value str)
3906 CAMLparam1(str);
3907 CAMLlocal1(result);
3908 char *curlResult;
3910 curlResult = curl_unescape(String_val(str), caml_string_length(str));
3911 result = caml_copy_string(curlResult);
3912 free(curlResult);
3914 CAMLreturn(result);
3918 ** curl_getdate helper function
3921 value caml_curl_getdate(value str, value now)
3923 CAMLparam2(str, now);
3924 CAMLlocal1(result);
3925 time_t curlResult;
3926 time_t curlNow;
3928 curlNow = (time_t)Double_val(now);
3929 curlResult = curl_getdate(String_val(str), &curlNow);
3930 result = caml_copy_double((double)curlResult);
3932 CAMLreturn(result);
3936 ** curl_version helper function
3939 value caml_curl_version(void)
3941 CAMLparam0();
3942 CAMLlocal1(result);
3943 char *str;
3945 str = curl_version();
3946 result = caml_copy_string(str);
3948 CAMLreturn(result);
3951 struct CURLVersionBitsMapping
3953 int code;
3954 char *name;
3957 struct CURLVersionBitsMapping versionBitsMap[] =
3959 {CURL_VERSION_IPV6, "ipv6"},
3960 {CURL_VERSION_KERBEROS4, "kerberos4"},
3961 {CURL_VERSION_SSL, "ssl"},
3962 {CURL_VERSION_LIBZ, "libz"},
3963 {CURL_VERSION_NTLM, "ntlm"},
3964 {CURL_VERSION_GSSNEGOTIATE, "gssnegotiate"},
3965 {CURL_VERSION_DEBUG, "debug"},
3966 {CURL_VERSION_CURLDEBUG, "curldebug"},
3967 {CURL_VERSION_ASYNCHDNS, "asynchdns"},
3968 {CURL_VERSION_SPNEGO, "spnego"},
3969 {CURL_VERSION_LARGEFILE, "largefile"},
3970 {CURL_VERSION_IDN, "idn"},
3971 {CURL_VERSION_SSPI, "sspi"},
3972 {CURL_VERSION_CONV, "conv"},
3973 #if HAVE_DECL_CURL_VERSION_TLSAUTH_SRP
3974 {CURL_VERSION_TLSAUTH_SRP, "srp"},
3975 #endif
3976 #if HAVE_DECL_CURL_VERSION_NTLM_WB
3977 {CURL_VERSION_NTLM_WB, "wb"},
3978 #endif
3981 value caml_curl_version_info(value unit)
3983 CAMLparam1(unit);
3984 CAMLlocal4(v, vlist, vnum, vfeatures);
3985 const char* const* p = NULL;
3986 size_t i = 0;
3988 curl_version_info_data* data = curl_version_info(CURLVERSION_NOW);
3989 if (NULL == data) caml_failwith("curl_version_info");
3991 vlist = Val_emptylist;
3992 for (p = data->protocols; NULL != *p; p++)
3994 vlist = Val_cons(vlist, caml_copy_string(*p));
3997 vfeatures = Val_emptylist;
3998 for (i = 0; i < sizeof(versionBitsMap)/sizeof(versionBitsMap[0]); i++)
4000 if (0 != (versionBitsMap[i].code & data->features))
4001 vfeatures = Val_cons(vfeatures, caml_copy_string(versionBitsMap[i].name));
4004 vnum = caml_alloc_tuple(3);
4005 Store_field(vnum,0,Val_int(0xFF & (data->version_num >> 16)));
4006 Store_field(vnum,1,Val_int(0xFF & (data->version_num >> 8)));
4007 Store_field(vnum,2,Val_int(0xFF & (data->version_num)));
4009 v = caml_alloc_tuple(12);
4010 Store_field(v,0,caml_copy_string(data->version));
4011 Store_field(v,1,vnum);
4012 Store_field(v,2,caml_copy_string(data->host));
4013 Store_field(v,3,vfeatures);
4014 Store_field(v,4,data->ssl_version ? Val_some(caml_copy_string(data->ssl_version)) : Val_none);
4015 Store_field(v,5,data->libz_version ? Val_some(caml_copy_string(data->libz_version)) : Val_none);
4016 Store_field(v,6,vlist);
4017 Store_field(v,7,caml_copy_string((data->age >= 1 && data->ares) ? data->ares : ""));
4018 Store_field(v,8,Val_int((data->age >= 1) ? data->ares_num : 0));
4019 Store_field(v,9,caml_copy_string((data->age >= 2 && data->libidn) ? data->libidn : ""));
4020 Store_field(v,10,Val_int((data->age >= 3) ? data->iconv_ver_num : 0));
4021 Store_field(v,11,caml_copy_string((data->age >= 3 && data->libssh_version) ? data->libssh_version : ""));
4023 CAMLreturn(v);
4026 value caml_curl_pause(value conn, value opts)
4028 CAMLparam2(conn, opts);
4029 CAMLlocal4(v, vlist, vnum, vfeatures);
4030 Connection *connection = Connection_val(conn);
4031 int bitmask = 0;
4032 CURLcode result;
4034 while (Val_emptylist != opts)
4036 switch (Int_val(Field(opts,0)))
4038 case 0: bitmask |= CURLPAUSE_SEND; break;
4039 case 1: bitmask |= CURLPAUSE_RECV; break;
4040 case 2: bitmask |= CURLPAUSE_ALL; break;
4041 default: caml_failwith("wrong pauseOption");
4043 opts = Field(opts,1);
4046 result = curl_easy_pause(connection->handle,bitmask);
4047 if (result != CURLE_OK)
4048 raiseError(connection, result);
4050 CAMLreturn(Val_unit);
4054 * Curl multi stack support
4056 * Exported thin wrappers for libcurl are prefixed with caml_curl_multi_.
4057 * Other exported functions are prefixed with caml_curlm_, some of them
4058 * can/should be decomposed into smaller parts.
4061 struct ml_multi_handle
4063 CURLM* handle;
4064 value values; /* callbacks */
4067 enum
4069 curlmopt_socket_function,
4070 curlmopt_timer_function,
4072 /* last, not used */
4073 multi_values_total
4076 typedef struct ml_multi_handle ml_multi_handle;
4078 #define Multi_val(v) (*(ml_multi_handle**)Data_custom_val(v))
4079 #define CURLM_val(v) (Multi_val(v)->handle)
4081 static struct custom_operations curl_multi_ops = {
4082 "ygrek.curl_multi",
4083 custom_finalize_default,
4084 custom_compare_default,
4085 custom_hash_default,
4086 custom_serialize_default,
4087 custom_deserialize_default,
4088 #if defined(custom_compare_ext_default)
4089 custom_compare_ext_default,
4090 #endif
4093 value caml_curl_multi_init(value unit)
4095 CAMLparam1(unit);
4096 CAMLlocal1(v);
4097 ml_multi_handle* multi = (ml_multi_handle*)caml_stat_alloc(sizeof(ml_multi_handle));
4098 CURLM* h = curl_multi_init();
4100 if (!h)
4102 caml_stat_free(multi);
4103 caml_failwith("caml_curl_multi_init");
4106 multi->handle = h;
4107 multi->values = caml_alloc(multi_values_total, 0);
4108 caml_register_generational_global_root(&multi->values);
4110 v = caml_alloc_custom(&curl_multi_ops, sizeof(ml_multi_handle*), 0, 1);
4111 Multi_val(v) = multi;
4113 CAMLreturn(v);
4116 value caml_curl_multi_cleanup(value handle)
4118 CAMLparam1(handle);
4119 ml_multi_handle* h = Multi_val(handle);
4121 if (NULL == h)
4122 CAMLreturn(Val_unit);
4124 caml_remove_generational_global_root(&h->values);
4126 if (CURLM_OK != curl_multi_cleanup(h->handle))
4127 caml_failwith("caml_curl_multi_cleanup");
4129 caml_stat_free(h);
4130 Multi_val(handle) = (ml_multi_handle*)NULL;
4132 CAMLreturn(Val_unit);
4135 static CURL* curlm_remove_finished(CURLM* multi_handle, CURLcode* result)
4137 int msgs_in_queue = 0;
4139 while (1)
4141 CURLMsg* msg = curl_multi_info_read(multi_handle, &msgs_in_queue);
4142 if (NULL == msg) return NULL;
4143 if (CURLMSG_DONE == msg->msg)
4145 CURL* easy_handle = msg->easy_handle;
4146 if (result) *result = msg->data.result;
4147 if (CURLM_OK != curl_multi_remove_handle(multi_handle, easy_handle))
4149 /*caml_failwith("curlm_remove_finished");*/
4151 return easy_handle;
4156 value caml_curlm_remove_finished(value v_multi)
4158 CAMLparam1(v_multi);
4159 CAMLlocal2(v_easy, v_tuple);
4160 CURL* handle;
4161 CURLM* multi_handle;
4162 CURLcode result;
4163 Connection* conn = NULL;
4165 multi_handle = CURLM_val(v_multi);
4167 caml_enter_blocking_section();
4168 handle = curlm_remove_finished(multi_handle,&result);
4169 caml_leave_blocking_section();
4171 if (NULL == handle)
4173 CAMLreturn(Val_none);
4175 else
4177 conn = getConnection(handle);
4178 if (conn->curl_ERRORBUFFER != NULL)
4180 Store_field(Field(conn->ocamlValues, Ocaml_ERRORBUFFER), 0, caml_copy_string(conn->curl_ERRORBUFFER));
4182 conn->refcount--;
4183 /* NB: same handle, but different block */
4184 v_easy = caml_curl_alloc(conn);
4185 v_tuple = caml_alloc(2, 0);
4186 Store_field(v_tuple,0,v_easy);
4187 Store_field(v_tuple,1,Val_int(result)); /* CURLcode */
4188 CAMLreturn(Val_some(v_tuple));
4192 value caml_curl_multi_wait(value v_timeout_ms, value v_multi)
4194 CAMLparam2(v_timeout_ms,v_multi);
4195 CURLM *multi_handle = CURLM_val(v_multi);
4196 int timeout_ms = Int_val(v_timeout_ms);
4197 int numfds = -1;
4198 CURLMcode ret;
4200 caml_enter_blocking_section();
4201 ret = curl_multi_wait(multi_handle, NULL, 0, timeout_ms, &numfds);
4202 caml_leave_blocking_section();
4204 if (ret != CURLM_OK) caml_failwith("caml_curl_multi_wait");
4206 CAMLreturn(Val_bool(numfds != 0));
4209 value caml_curl_multi_add_handle(value v_multi, value v_easy)
4211 CAMLparam2(v_multi,v_easy);
4212 CURLM* multi = CURLM_val(v_multi);
4213 Connection* conn = Connection_val(v_easy);
4215 /* prevent collection of OCaml value while the easy handle is used
4216 and may invoke callbacks registered on OCaml side */
4217 conn->refcount++;
4219 /* may invoke callbacks so need to be consistent with locks */
4220 caml_enter_blocking_section();
4221 if (CURLM_OK != curl_multi_add_handle(multi, conn->handle))
4223 conn->refcount--; /* not added, revert */
4224 caml_leave_blocking_section();
4225 caml_failwith("caml_curl_multi_add_handle");
4227 caml_leave_blocking_section();
4229 CAMLreturn(Val_unit);
4232 value caml_curl_multi_remove_handle(value v_multi, value v_easy)
4234 CAMLparam2(v_multi,v_easy);
4235 CURLM* multi = CURLM_val(v_multi);
4236 Connection* conn = Connection_val(v_easy);
4238 /* may invoke callbacks so need to be consistent with locks */
4239 caml_enter_blocking_section();
4240 if (CURLM_OK != curl_multi_remove_handle(multi, conn->handle))
4242 caml_leave_blocking_section();
4243 caml_failwith("caml_curl_multi_remove_handle");
4245 conn->refcount--;
4246 caml_leave_blocking_section();
4248 CAMLreturn(Val_unit);
4251 value caml_curl_multi_perform_all(value v_multi)
4253 CAMLparam1(v_multi);
4254 int still_running = 0;
4255 CURLM* h = CURLM_val(v_multi);
4257 caml_enter_blocking_section();
4258 while (CURLM_CALL_MULTI_PERFORM == curl_multi_perform(h, &still_running));
4259 caml_leave_blocking_section();
4261 CAMLreturn(Val_int(still_running));
4264 /* currently curlCode repr tag matches CURLE_ value
4265 * this is exploited below, but generally should use errorMap */
4267 value caml_curl_strerror(value v_code)
4269 CAMLparam1(v_code);
4270 CAMLreturn(caml_copy_string(curl_easy_strerror((CURLcode)Int_val(v_code))));
4273 value caml_curl_int_of_curlCode(value v_code)
4275 return v_code;
4278 value caml_curl_curlCode_of_int(value v)
4280 return (Int_val(v) < sizeof(errorMap) / sizeof(errorMap[0]) ? Val_some(v) : Val_none);
4284 * Wrappers for the curl_multi_socket_action infrastructure
4285 * Based on curl hiperfifo.c example
4288 #ifdef _WIN32
4289 #ifndef Val_socket
4290 #define Val_socket(v) win_alloc_socket(v)
4291 #endif
4292 #ifndef Socket_val
4293 #error Socket_val not defined in unixsupport.h
4294 #endif
4295 #else /* _WIN32 */
4296 #ifndef Socket_val
4297 #define Socket_val(v) Long_val(v)
4298 #endif
4299 #ifndef Val_socket
4300 #define Val_socket(v) Val_int(v)
4301 #endif
4302 #endif /* _WIN32 */
4304 static void raise_multi_error(char const* msg)
4306 static value* exception = NULL;
4308 if (NULL == exception)
4310 exception = caml_named_value("Curl.Multi.Error");
4311 if (NULL == exception) caml_invalid_argument("Curl.Multi.Error");
4314 caml_raise_with_string(*exception, msg);
4317 static void check_mcode(CURLMcode code)
4319 char const *s = NULL;
4320 switch (code)
4322 case CURLM_OK : return;
4323 case CURLM_CALL_MULTI_PERFORM : s="CURLM_CALL_MULTI_PERFORM"; break;
4324 case CURLM_BAD_HANDLE : s="CURLM_BAD_HANDLE"; break;
4325 case CURLM_BAD_EASY_HANDLE : s="CURLM_BAD_EASY_HANDLE"; break;
4326 case CURLM_OUT_OF_MEMORY : s="CURLM_OUT_OF_MEMORY"; break;
4327 case CURLM_INTERNAL_ERROR : s="CURLM_INTERNAL_ERROR"; break;
4328 case CURLM_UNKNOWN_OPTION : s="CURLM_UNKNOWN_OPTION"; break;
4329 case CURLM_LAST : s="CURLM_LAST"; break;
4330 case CURLM_BAD_SOCKET : s="CURLM_BAD_SOCKET"; break;
4331 default : s="CURLM_unknown"; break;
4333 raise_multi_error(s);
4336 value caml_curl_multi_socket_action(value v_multi, value v_fd, value v_kind)
4338 CAMLparam3(v_multi, v_fd, v_kind);
4339 CURLM* h = CURLM_val(v_multi);
4340 int still_running = 0;
4341 CURLMcode rc = CURLM_OK;
4342 curl_socket_t socket;
4343 int kind = 0;
4345 if (Val_none == v_fd)
4347 socket = CURL_SOCKET_TIMEOUT;
4349 else
4351 socket = Socket_val(Field(v_fd, 0));
4354 switch (Int_val(v_kind))
4356 case 0 : break;
4357 case 1 : kind |= CURL_CSELECT_IN; break;
4358 case 2 : kind |= CURL_CSELECT_OUT; break;
4359 case 3 : kind |= CURL_CSELECT_IN | CURL_CSELECT_OUT; break;
4360 default:
4361 raise_multi_error("caml_curl_multi_socket_action");
4364 /* fprintf(stdout,"fd %u kind %u\n",socket, kind); fflush(stdout); */
4366 caml_enter_blocking_section();
4367 do {
4368 rc = curl_multi_socket_action(h, socket, kind, &still_running);
4369 } while (rc == CURLM_CALL_MULTI_PERFORM);
4370 caml_leave_blocking_section();
4372 check_mcode(rc);
4374 CAMLreturn(Val_int(still_running));
4377 value caml_curl_multi_socket_all(value v_multi)
4379 CAMLparam1(v_multi);
4380 int still_running = 0;
4381 CURLMcode rc = CURLM_OK;
4382 CURLM* h = CURLM_val(v_multi);
4384 caml_enter_blocking_section();
4385 do {
4386 rc = curl_multi_socket_all(h, &still_running);
4387 } while (rc == CURLM_CALL_MULTI_PERFORM);
4388 caml_leave_blocking_section();
4390 check_mcode(rc);
4392 CAMLreturn(Val_int(still_running));
4395 static int curlm_sock_cb(CURL *e, curl_socket_t sock, int what, void *cbp, void *sockp)
4397 caml_leave_blocking_section();
4399 CAMLparam0();
4400 CAMLlocal2(v_what,csock);
4401 (void)e;
4402 (void)sockp; /* not used */
4404 /* v_what = Val_int(what); */
4405 switch (what)
4407 case CURL_POLL_NONE : v_what = Val_int(0); break;
4408 case CURL_POLL_IN : v_what = Val_int(1); break;
4409 case CURL_POLL_OUT : v_what = Val_int(2); break;
4410 case CURL_POLL_INOUT : v_what = Val_int(3); break;
4411 case CURL_POLL_REMOVE : v_what = Val_int(4); break;
4412 default:
4413 fprintf(stderr, "curlm_sock_cb sock=%lld what=%d\n", (long long)sock, what);
4414 fflush(stderr);
4415 raise_multi_error("curlm_sock_cb"); /* FIXME exception from callback */
4417 csock=Val_socket(sock);
4418 caml_callback2(Field(((ml_multi_handle*)cbp)->values,curlmopt_socket_function),
4419 csock, v_what);
4420 CAMLdrop;
4422 caml_enter_blocking_section();
4423 return 0;
4426 value caml_curl_multi_socketfunction(value v_multi, value v_cb)
4428 CAMLparam2(v_multi, v_cb);
4429 ml_multi_handle* multi = Multi_val(v_multi);
4431 Store_field(multi->values, curlmopt_socket_function, v_cb);
4433 curl_multi_setopt(multi->handle, CURLMOPT_SOCKETFUNCTION, curlm_sock_cb);
4434 curl_multi_setopt(multi->handle, CURLMOPT_SOCKETDATA, multi);
4436 CAMLreturn(Val_unit);
4439 static int curlm_timer_cb(CURLM *multi, long timeout_ms, void *userp)
4441 caml_leave_blocking_section();
4443 CAMLparam0();
4444 (void)multi;
4445 caml_callback(Field(((ml_multi_handle*)userp)->values,curlmopt_timer_function), Val_long(timeout_ms));
4446 CAMLdrop;
4448 caml_enter_blocking_section();
4449 return 0;
4452 value caml_curl_multi_timerfunction(value v_multi, value v_cb)
4454 CAMLparam2(v_multi, v_cb);
4455 ml_multi_handle* multi = Multi_val(v_multi);
4457 Store_field(multi->values, curlmopt_timer_function, v_cb);
4459 curl_multi_setopt(multi->handle, CURLMOPT_TIMERFUNCTION, curlm_timer_cb);
4460 curl_multi_setopt(multi->handle, CURLMOPT_TIMERDATA, multi);
4462 CAMLreturn(Val_unit);
4465 value caml_curl_multi_timeout(value v_multi)
4467 CAMLparam1(v_multi);
4468 long ms = 0;
4469 CURLMcode rc = CURLM_OK;
4470 ml_multi_handle* multi = Multi_val(v_multi);
4472 rc = curl_multi_timeout(multi->handle, &ms);
4474 check_mcode(rc);
4476 CAMLreturn(Val_long(ms));
4479 #define SETMOPT_VAL_(func_name, curl_option, conv_val) \
4480 static void func_name(CURLM *handle, value option) \
4482 CAMLparam1(option); \
4483 CURLMcode result = CURLM_OK; \
4485 result = curl_multi_setopt(handle, curl_option, conv_val(option)); \
4487 check_mcode(result); \
4489 CAMLreturn0; \
4492 #define SETMOPT_VAL(name, conv) SETMOPT_VAL_(handle_multi_##name, CURLMOPT_##name, conv)
4493 #define SETMOPT_BOOL(name) SETMOPT_VAL(name, Bool_val)
4494 #define SETMOPT_LONG(name) SETMOPT_VAL(name, Long_val)
4495 #define SETMOPT_INT64(name) SETMOPT_VAL(name, Int64_val)
4497 long pipeliningMap[] =
4499 0, /* CURLPIPE_NOTHING */
4500 1, /* CURLPIPE_HTTP1 */
4501 2, /* CURLPIPE_MULTIPLEX */
4504 static void handle_multi_PIPELINING(CURLM* handle, value option)
4506 CAMLparam1(option);
4507 CURLMcode result = CURLM_OK;
4509 long bits = convert_bit_list(pipeliningMap, sizeof(pipeliningMap) / sizeof(pipeliningMap[0]), option);
4511 result = curl_multi_setopt(handle, CURLMOPT_PIPELINING, bits);
4513 check_mcode(result);
4515 CAMLreturn0;
4518 #if HAVE_DECL_CURLMOPT_MAXCONNECTS
4519 SETMOPT_LONG( MAXCONNECTS)
4520 #endif
4522 #if HAVE_DECL_CURLMOPT_MAX_PIPELINE_LENGTH
4523 SETMOPT_LONG( MAX_PIPELINE_LENGTH)
4524 #endif
4526 #if HAVE_DECL_CURLMOPT_MAX_HOST_CONNECTIONS
4527 SETMOPT_LONG( MAX_HOST_CONNECTIONS)
4528 #endif
4530 typedef struct CURLMOptionMapping CURLMOptionMapping;
4531 #define OPT(name) { handle_multi_## name, "CURLMOPT_"#name}
4532 #define NO_OPT(name) { NULL, "CURLMOPT_"#name}
4534 struct CURLMOptionMapping
4536 void (*optionHandler)(CURLM *, value);
4537 char *name;
4540 CURLMOptionMapping implementedMOptionMap[] = {
4541 OPT( PIPELINING),
4542 #if HAVE_DECL_CURLMOPT_MAXCONNECTS
4543 OPT( MAXCONNECTS),
4544 #else
4545 NO_OPT( MAXCONNECTS),
4546 #endif
4547 #if HAVE_DECL_CURLMOPT_MAX_PIPELINE_LENGTH
4548 OPT( MAX_PIPELINE_LENGTH),
4549 #else
4550 NO_OPT( MAX_PIPELINE_LENGTH),
4551 #endif
4552 #if HAVE_DECL_CURLMOPT_MAX_HOST_CONNECTIONS
4553 OPT( MAX_HOST_CONNECTIONS),
4554 #else
4555 NO_OPT( MAX_HOST_CONNECTIONS),
4556 #endif
4559 value caml_curl_multi_setopt(value v_multi, value option)
4561 CAMLparam2(v_multi, option);
4562 CAMLlocal1(data);
4563 CURLM *handle = Multi_val(v_multi)->handle;
4564 CURLMOptionMapping* thisOption = NULL;
4565 static value* exception = NULL;
4567 data = Field(option, 0);
4569 if (Tag_val(option) < sizeof(implementedMOptionMap)/sizeof(CURLMOptionMapping))
4571 thisOption = &implementedMOptionMap[Tag_val(option)];
4572 if (thisOption->optionHandler)
4574 thisOption->optionHandler(handle, data);
4576 else
4578 if (NULL == exception)
4580 exception = caml_named_value("Curl.NotImplemented");
4581 if (NULL == exception) caml_invalid_argument("Curl.NotImplemented");
4584 caml_raise_with_string(*exception, thisOption->name);
4587 else
4589 caml_failwith("Invalid CURLMOPT Option");
4592 CAMLreturn(Val_unit);
4595 struct used_enum
4597 int last_used;
4598 int last;
4599 char const* name;
4602 #define CURL_ENUM(name,last_used) { CURL_ ## name ## _ ## last_used, CURL_ ## name ## _LAST, #name }
4604 struct used_enum check_enums[] = {
4605 { CURLINFO_SSL_DATA_OUT, CURLINFO_END, "CURLINFO" },
4606 #if defined(CURL_HTTP_VERSION_2TLS) /* FIXME */
4607 CURL_ENUM(HTTP_VERSION, 2TLS),
4608 #endif
4611 value caml_curl_outdated_enums(value v_unit)
4613 CAMLparam0();
4614 CAMLlocal1(v);
4615 size_t i;
4617 v = Val_emptylist;
4619 for (i = 0; i < sizeof(check_enums) / sizeof(struct used_enum); i++)
4621 if (check_enums[i].last_used + 1 != check_enums[i].last)
4623 v = Val_cons(v, caml_copy_string(check_enums[i].name));
4627 CAMLreturn(v);
4630 #ifdef __cplusplus
4632 #endif