123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) 1998 - 2011, Daniel Stenberg, <daniel@haxx.se>, et al.
- *
- * This software is licensed as described in the file COPYING, which
- * you should have received as part of this distribution. The terms
- * are also available at http://curl.haxx.se/docs/copyright.html.
- *
- * You may opt to use, copy, modify, merge, publish, distribute and/or sell
- * copies of the Software, and permit persons to whom the Software is
- * furnished to do so, under the terms of the COPYING file.
- *
- * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
- * KIND, either express or implied.
- *
- * RFC3501 IMAPv4 protocol
- * RFC5092 IMAP URL Scheme
- *
- ***************************************************************************/
- #include "setup.h"
- #ifndef CURL_DISABLE_IMAP
- #ifdef HAVE_UNISTD_H
- #include <unistd.h>
- #endif
- #ifdef HAVE_SYS_SOCKET_H
- #include <sys/socket.h>
- #endif
- #ifdef HAVE_NETINET_IN_H
- #include <netinet/in.h>
- #endif
- #ifdef HAVE_ARPA_INET_H
- #include <arpa/inet.h>
- #endif
- #ifdef HAVE_UTSNAME_H
- #include <sys/utsname.h>
- #endif
- #ifdef HAVE_NETDB_H
- #include <netdb.h>
- #endif
- #ifdef __VMS
- #include <in.h>
- #include <inet.h>
- #endif
- #if (defined(NETWARE) && defined(__NOVELL_LIBC__))
- #undef in_addr_t
- #define in_addr_t unsigned long
- #endif
- #include <curl/curl.h>
- #include "urldata.h"
- #include "sendf.h"
- #include "if2ip.h"
- #include "hostip.h"
- #include "progress.h"
- #include "transfer.h"
- #include "escape.h"
- #include "http.h" /* for HTTP proxy tunnel stuff */
- #include "socks.h"
- #include "imap.h"
- #include "strtoofft.h"
- #include "strequal.h"
- #include "sslgen.h"
- #include "connect.h"
- #include "strerror.h"
- #include "select.h"
- #include "multiif.h"
- #include "url.h"
- #include "rawstr.h"
- #include "strtoofft.h"
- #include "http_proxy.h"
- #define _MPRINTF_REPLACE /* use our functions only */
- #include <curl/mprintf.h>
- #include "curl_memory.h"
- /* The last #include file should be: */
- #include "memdebug.h"
- /* Local API functions */
- static CURLcode imap_parse_url_path(struct connectdata *conn);
- static CURLcode imap_regular_transfer(struct connectdata *conn, bool *done);
- static CURLcode imap_do(struct connectdata *conn, bool *done);
- static CURLcode imap_done(struct connectdata *conn,
- CURLcode, bool premature);
- static CURLcode imap_connect(struct connectdata *conn, bool *done);
- static CURLcode imap_disconnect(struct connectdata *conn, bool dead);
- static CURLcode imap_multi_statemach(struct connectdata *conn, bool *done);
- static int imap_getsock(struct connectdata *conn,
- curl_socket_t *socks,
- int numsocks);
- static CURLcode imap_doing(struct connectdata *conn,
- bool *dophase_done);
- static CURLcode imap_setup_connection(struct connectdata * conn);
- static CURLcode imap_state_upgrade_tls(struct connectdata *conn);
- /*
- * IMAP protocol handler.
- */
- const struct Curl_handler Curl_handler_imap = {
- "IMAP", /* scheme */
- imap_setup_connection, /* setup_connection */
- imap_do, /* do_it */
- imap_done, /* done */
- ZERO_NULL, /* do_more */
- imap_connect, /* connect_it */
- imap_multi_statemach, /* connecting */
- imap_doing, /* doing */
- imap_getsock, /* proto_getsock */
- imap_getsock, /* doing_getsock */
- ZERO_NULL, /* perform_getsock */
- imap_disconnect, /* disconnect */
- ZERO_NULL, /* readwrite */
- PORT_IMAP, /* defport */
- CURLPROTO_IMAP, /* protocol */
- PROTOPT_CLOSEACTION | PROTOPT_NEEDSPWD /* flags */
- };
- #ifdef USE_SSL
- /*
- * IMAPS protocol handler.
- */
- const struct Curl_handler Curl_handler_imaps = {
- "IMAPS", /* scheme */
- imap_setup_connection, /* setup_connection */
- imap_do, /* do_it */
- imap_done, /* done */
- ZERO_NULL, /* do_more */
- imap_connect, /* connect_it */
- imap_multi_statemach, /* connecting */
- imap_doing, /* doing */
- imap_getsock, /* proto_getsock */
- imap_getsock, /* doing_getsock */
- ZERO_NULL, /* perform_getsock */
- imap_disconnect, /* disconnect */
- ZERO_NULL, /* readwrite */
- PORT_IMAPS, /* defport */
- CURLPROTO_IMAP | CURLPROTO_IMAPS, /* protocol */
- PROTOPT_CLOSEACTION | PROTOPT_SSL | PROTOPT_NEEDSPWD /* flags */
- };
- #endif
- #ifndef CURL_DISABLE_HTTP
- /*
- * HTTP-proxyed IMAP protocol handler.
- */
- static const struct Curl_handler Curl_handler_imap_proxy = {
- "IMAP", /* scheme */
- ZERO_NULL, /* setup_connection */
- Curl_http, /* do_it */
- Curl_http_done, /* done */
- ZERO_NULL, /* do_more */
- ZERO_NULL, /* connect_it */
- ZERO_NULL, /* connecting */
- ZERO_NULL, /* doing */
- ZERO_NULL, /* proto_getsock */
- ZERO_NULL, /* doing_getsock */
- ZERO_NULL, /* perform_getsock */
- ZERO_NULL, /* disconnect */
- ZERO_NULL, /* readwrite */
- PORT_IMAP, /* defport */
- CURLPROTO_HTTP, /* protocol */
- PROTOPT_NONE /* flags */
- };
- #ifdef USE_SSL
- /*
- * HTTP-proxyed IMAPS protocol handler.
- */
- static const struct Curl_handler Curl_handler_imaps_proxy = {
- "IMAPS", /* scheme */
- ZERO_NULL, /* setup_connection */
- Curl_http, /* do_it */
- Curl_http_done, /* done */
- ZERO_NULL, /* do_more */
- ZERO_NULL, /* connect_it */
- ZERO_NULL, /* connecting */
- ZERO_NULL, /* doing */
- ZERO_NULL, /* proto_getsock */
- ZERO_NULL, /* doing_getsock */
- ZERO_NULL, /* perform_getsock */
- ZERO_NULL, /* disconnect */
- ZERO_NULL, /* readwrite */
- PORT_IMAPS, /* defport */
- CURLPROTO_HTTP, /* protocol */
- PROTOPT_NONE /* flags */
- };
- #endif
- #endif
- /***********************************************************************
- *
- * imapsendf()
- *
- * Sends the formated string as an IMAP command to a server
- *
- * Designed to never block.
- */
- static CURLcode imapsendf(struct connectdata *conn,
- const char *idstr, /* id to wait for at the
- completion of this command */
- const char *fmt, ...)
- {
- CURLcode res;
- struct imap_conn *imapc = &conn->proto.imapc;
- va_list ap;
- va_start(ap, fmt);
- imapc->idstr = idstr; /* this is the thing */
- res = Curl_pp_vsendf(&imapc->pp, fmt, ap);
- va_end(ap);
- return res;
- }
- static const char *getcmdid(struct connectdata *conn)
- {
- static const char * const ids[]= {
- "A",
- "B",
- "C",
- "D"
- };
- struct imap_conn *imapc = &conn->proto.imapc;
- /* get the next id, but wrap at end of table */
- imapc->cmdid = (int)((imapc->cmdid+1) % (sizeof(ids)/sizeof(ids[0])));
- return ids[imapc->cmdid];
- }
- /* For the IMAP "protocol connect" and "doing" phases only */
- static int imap_getsock(struct connectdata *conn,
- curl_socket_t *socks,
- int numsocks)
- {
- return Curl_pp_getsock(&conn->proto.imapc.pp, socks, numsocks);
- }
- /* function that checks for an imap status code at the start of the
- given string */
- static int imap_endofresp(struct pingpong *pp, int *resp)
- {
- char *line = pp->linestart_resp;
- size_t len = pp->nread_resp;
- struct imap_conn *imapc = &pp->conn->proto.imapc;
- const char *id = imapc->idstr;
- size_t id_len = strlen(id);
- if(len >= id_len + 3) {
- if(!memcmp(id, line, id_len) && (line[id_len] == ' ') ) {
- /* end of response */
- *resp = line[id_len+1]; /* O, N or B */
- return TRUE;
- }
- else if((imapc->state == IMAP_FETCH) &&
- !memcmp("* ", line, 2) ) {
- /* FETCH response we're interested in */
- *resp = '*';
- return TRUE;
- }
- }
- return FALSE; /* nothing for us */
- }
- /* This is the ONLY way to change IMAP state! */
- static void state(struct connectdata *conn,
- imapstate newstate)
- {
- #if defined(DEBUGBUILD) && !defined(CURL_DISABLE_VERBOSE_STRINGS)
- /* for debug purposes */
- static const char * const names[]={
- "STOP",
- "SERVERGREET",
- "LOGIN",
- "STARTTLS",
- "UPGRADETLS",
- "SELECT",
- "FETCH",
- "LOGOUT",
- /* LAST */
- };
- #endif
- struct imap_conn *imapc = &conn->proto.imapc;
- #if defined(DEBUGBUILD) && !defined(CURL_DISABLE_VERBOSE_STRINGS)
- if(imapc->state != newstate)
- infof(conn->data, "IMAP %p state change from %s to %s\n",
- imapc, names[imapc->state], names[newstate]);
- #endif
- imapc->state = newstate;
- }
- static CURLcode imap_state_login(struct connectdata *conn)
- {
- CURLcode result;
- struct FTP *imap = conn->data->state.proto.imap;
- const char *str;
- str = getcmdid(conn);
- /* send USER and password */
- result = imapsendf(conn, str, "%s LOGIN %s %s", str,
- imap->user?imap->user:"",
- imap->passwd?imap->passwd:"");
- if(result)
- return result;
- state(conn, IMAP_LOGIN);
- return CURLE_OK;
- }
- #ifdef USE_SSL
- static void imap_to_imaps(struct connectdata *conn)
- {
- conn->handler = &Curl_handler_imaps;
- }
- #else
- #define imap_to_imaps(x) Curl_nop_stmt
- #endif
- /* for STARTTLS responses */
- static CURLcode imap_state_starttls_resp(struct connectdata *conn,
- int imapcode,
- imapstate instate)
- {
- CURLcode result = CURLE_OK;
- struct SessionHandle *data = conn->data;
- (void)instate; /* no use for this yet */
- if(imapcode != 'O') {
- failf(data, "STARTTLS denied. %c", imapcode);
- result = CURLE_LOGIN_DENIED;
- }
- else {
- if(data->state.used_interface == Curl_if_multi) {
- state(conn, IMAP_UPGRADETLS);
- return imap_state_upgrade_tls(conn);
- }
- else {
- result = Curl_ssl_connect(conn, FIRSTSOCKET);
- if(CURLE_OK == result) {
- imap_to_imaps(conn);
- result = imap_state_login(conn);
- }
- }
- }
- state(conn, IMAP_STOP);
- return result;
- }
- static CURLcode imap_state_upgrade_tls(struct connectdata *conn)
- {
- struct imap_conn *imapc = &conn->proto.imapc;
- CURLcode result;
- result = Curl_ssl_connect_nonblocking(conn, FIRSTSOCKET, &imapc->ssldone);
- if(imapc->ssldone) {
- imap_to_imaps(conn);
- result = imap_state_login(conn);
- state(conn, IMAP_STOP);
- }
- return result;
- }
- /* for LOGIN responses */
- static CURLcode imap_state_login_resp(struct connectdata *conn,
- int imapcode,
- imapstate instate)
- {
- CURLcode result = CURLE_OK;
- struct SessionHandle *data = conn->data;
- (void)instate; /* no use for this yet */
- if(imapcode != 'O') {
- failf(data, "Access denied. %c", imapcode);
- result = CURLE_LOGIN_DENIED;
- }
- state(conn, IMAP_STOP);
- return result;
- }
- /* for the (first line of) FETCH BODY[TEXT] response */
- static CURLcode imap_state_fetch_resp(struct connectdata *conn,
- int imapcode,
- imapstate instate)
- {
- CURLcode result = CURLE_OK;
- struct SessionHandle *data = conn->data;
- struct imap_conn *imapc = &conn->proto.imapc;
- struct FTP *imap = data->state.proto.imap;
- struct pingpong *pp = &imapc->pp;
- const char *ptr = data->state.buffer;
- (void)instate; /* no use for this yet */
- if('*' != imapcode) {
- Curl_pgrsSetDownloadSize(data, 0);
- state(conn, IMAP_STOP);
- return CURLE_OK;
- }
- /* Something like this comes "* 1 FETCH (BODY[TEXT] {2021}\r" */
- while(*ptr && (*ptr != '{'))
- ptr++;
- if(*ptr == '{') {
- curl_off_t filesize = curlx_strtoofft(ptr+1, NULL, 10);
- if(filesize)
- Curl_pgrsSetDownloadSize(data, filesize);
- infof(data, "Found %" FORMAT_OFF_TU " bytes to download\n", filesize);
- if(pp->cache) {
- /* At this point there is a bunch of data in the header "cache" that is
- actually body content, send it as body and then skip it. Do note
- that there may even be additional "headers" after the body. */
- size_t chunk = pp->cache_size;
- if(chunk > (size_t)filesize)
- /* the conversion from curl_off_t to size_t is always fine here */
- chunk = (size_t)filesize;
- result = Curl_client_write(conn, CLIENTWRITE_BODY, pp->cache, chunk);
- if(result)
- return result;
- filesize -= chunk;
- /* we've now used parts of or the entire cache */
- if(pp->cache_size > chunk) {
- /* part of, move the trailing data to the start and reduce the size */
- memmove(pp->cache, pp->cache+chunk,
- pp->cache_size - chunk);
- pp->cache_size -= chunk;
- }
- else {
- /* cache is drained */
- free(pp->cache);
- pp->cache = NULL;
- pp->cache_size = 0;
- }
- }
- infof(data, "Filesize left: %" FORMAT_OFF_T "\n", filesize);
- if(!filesize)
- /* the entire data is already transferred! */
- Curl_setup_transfer(conn, -1, -1, FALSE, NULL, -1, NULL);
- else
- /* IMAP download */
- Curl_setup_transfer(conn, FIRSTSOCKET, filesize, FALSE,
- imap->bytecountp, -1, NULL); /* no upload here */
- data->req.maxdownload = filesize;
- }
- else
- /* We don't know how to parse this line */
- result = CURLE_FTP_WEIRD_SERVER_REPLY; /* TODO: fix this code */
- state(conn, IMAP_STOP);
- return result;
- }
- /* start the DO phase */
- static CURLcode imap_select(struct connectdata *conn)
- {
- CURLcode result = CURLE_OK;
- struct imap_conn *imapc = &conn->proto.imapc;
- const char *str;
- str = getcmdid(conn);
- result = imapsendf(conn, str, "%s SELECT %s", str,
- imapc->mailbox?imapc->mailbox:"");
- if(result)
- return result;
- state(conn, IMAP_SELECT);
- return result;
- }
- static CURLcode imap_fetch(struct connectdata *conn)
- {
- CURLcode result = CURLE_OK;
- const char *str;
- str = getcmdid(conn);
- /* TODO: make this select the correct mail
- * Use "1 body[text]" to get the full mail body of mail 1
- */
- result = imapsendf(conn, str, "%s FETCH 1 BODY[TEXT]", str);
- if(result)
- return result;
- /*
- * When issued, the server will respond with a single line similar to
- * '* 1 FETCH (BODY[TEXT] {2021}'
- *
- * Identifying the fetch and how many bytes of contents we can expect. We
- * must extract that number before continuing to "download as usual".
- */
- state(conn, IMAP_FETCH);
- return result;
- }
- /* for SELECT responses */
- static CURLcode imap_state_select_resp(struct connectdata *conn,
- int imapcode,
- imapstate instate)
- {
- CURLcode result = CURLE_OK;
- struct SessionHandle *data = conn->data;
- (void)instate; /* no use for this yet */
- if(imapcode != 'O') {
- failf(data, "Select failed");
- result = CURLE_LOGIN_DENIED;
- }
- else
- result = imap_fetch(conn);
- return result;
- }
- static CURLcode imap_statemach_act(struct connectdata *conn)
- {
- CURLcode result;
- curl_socket_t sock = conn->sock[FIRSTSOCKET];
- struct SessionHandle *data=conn->data;
- int imapcode;
- struct imap_conn *imapc = &conn->proto.imapc;
- struct pingpong *pp = &imapc->pp;
- size_t nread = 0;
- /* busy upgrading the connection; right now all I/O is SSL/TLS, not IMAP */
- if(imapc->state == IMAP_UPGRADETLS)
- return imap_state_upgrade_tls(conn);
- if(pp->sendleft)
- return Curl_pp_flushsend(pp);
- /* we read a piece of response */
- result = Curl_pp_readresp(sock, pp, &imapcode, &nread);
- if(result)
- return result;
- if(imapcode)
- /* we have now received a full IMAP server response */
- switch(imapc->state) {
- case IMAP_SERVERGREET:
- if(imapcode != 'O') {
- failf(data, "Got unexpected imap-server response");
- return CURLE_FTP_WEIRD_SERVER_REPLY;
- }
- if(data->set.ftp_ssl && !conn->ssl[FIRSTSOCKET].use) {
- /* We don't have a SSL/TLS connection yet, but SSL is requested. Switch
- to TLS connection now */
- const char *str;
- str = getcmdid(conn);
- result = imapsendf(conn, str, "%s STARTTLS", str);
- state(conn, IMAP_STARTTLS);
- }
- else
- result = imap_state_login(conn);
- if(result)
- return result;
- break;
- case IMAP_LOGIN:
- result = imap_state_login_resp(conn, imapcode, imapc->state);
- break;
- case IMAP_STARTTLS:
- result = imap_state_starttls_resp(conn, imapcode, imapc->state);
- break;
- case IMAP_FETCH:
- result = imap_state_fetch_resp(conn, imapcode, imapc->state);
- break;
- case IMAP_SELECT:
- result = imap_state_select_resp(conn, imapcode, imapc->state);
- break;
- case IMAP_LOGOUT:
- /* fallthrough, just stop! */
- default:
- /* internal error */
- state(conn, IMAP_STOP);
- break;
- }
- return result;
- }
- /* called repeatedly until done from multi.c */
- static CURLcode imap_multi_statemach(struct connectdata *conn,
- bool *done)
- {
- struct imap_conn *imapc = &conn->proto.imapc;
- CURLcode result;
- if((conn->handler->flags & PROTOPT_SSL) && !imapc->ssldone)
- result = Curl_ssl_connect_nonblocking(conn, FIRSTSOCKET, &imapc->ssldone);
- else
- result = Curl_pp_multi_statemach(&imapc->pp);
- *done = (imapc->state == IMAP_STOP) ? TRUE : FALSE;
- return result;
- }
- static CURLcode imap_easy_statemach(struct connectdata *conn)
- {
- struct imap_conn *imapc = &conn->proto.imapc;
- struct pingpong *pp = &imapc->pp;
- CURLcode result = CURLE_OK;
- while(imapc->state != IMAP_STOP) {
- result = Curl_pp_easy_statemach(pp);
- if(result)
- break;
- }
- return result;
- }
- /*
- * Allocate and initialize the struct IMAP for the current SessionHandle. If
- * need be.
- */
- static CURLcode imap_init(struct connectdata *conn)
- {
- struct SessionHandle *data = conn->data;
- struct FTP *imap = data->state.proto.imap;
- if(!imap) {
- imap = data->state.proto.imap = calloc(sizeof(struct FTP), 1);
- if(!imap)
- return CURLE_OUT_OF_MEMORY;
- }
- /* get some initial data into the imap struct */
- imap->bytecountp = &data->req.bytecount;
- /* No need to duplicate user+password, the connectdata struct won't change
- during a session, but we re-init them here since on subsequent inits
- since the conn struct may have changed or been replaced.
- */
- imap->user = conn->user;
- imap->passwd = conn->passwd;
- return CURLE_OK;
- }
- /*
- * imap_connect() should do everything that is to be considered a part of
- * the connection phase.
- *
- * The variable 'done' points to will be TRUE if the protocol-layer connect
- * phase is done when this function returns, or FALSE is not. When called as
- * a part of the easy interface, it will always be TRUE.
- */
- static CURLcode imap_connect(struct connectdata *conn,
- bool *done) /* see description above */
- {
- CURLcode result;
- struct imap_conn *imapc = &conn->proto.imapc;
- struct SessionHandle *data=conn->data;
- struct pingpong *pp = &imapc->pp;
- *done = FALSE; /* default to not done yet */
- /* If there already is a protocol-specific struct allocated for this
- sessionhandle, deal with it */
- Curl_reset_reqproto(conn);
- result = imap_init(conn);
- if(CURLE_OK != result)
- return result;
- /* We always support persistent connections on imap */
- conn->bits.close = FALSE;
- pp->response_time = RESP_TIMEOUT; /* set default response time-out */
- pp->statemach_act = imap_statemach_act;
- pp->endofresp = imap_endofresp;
- pp->conn = conn;
- if(conn->bits.tunnel_proxy && conn->bits.httpproxy) {
- /* for IMAP over HTTP proxy */
- struct HTTP http_proxy;
- struct FTP *imap_save;
- /* BLOCKING */
- /* We want "seamless" IMAP operations through HTTP proxy tunnel */
- /* Curl_proxyCONNECT is based on a pointer to a struct HTTP at the member
- * conn->proto.http; we want IMAP through HTTP and we have to change the
- * member temporarily for connecting to the HTTP proxy. After
- * Curl_proxyCONNECT we have to set back the member to the original struct
- * IMAP pointer
- */
- imap_save = data->state.proto.imap;
- memset(&http_proxy, 0, sizeof(http_proxy));
- data->state.proto.http = &http_proxy;
- result = Curl_proxyCONNECT(conn, FIRSTSOCKET,
- conn->host.name, conn->remote_port);
- data->state.proto.imap = imap_save;
- if(CURLE_OK != result)
- return result;
- }
- if((conn->handler->flags & PROTOPT_SSL) &&
- data->state.used_interface != Curl_if_multi) {
- /* BLOCKING */
- result = Curl_ssl_connect(conn, FIRSTSOCKET);
- if(result)
- return result;
- }
- Curl_pp_init(pp); /* init generic pingpong data */
- /* When we connect, we start in the state where we await the server greeting
- response */
- state(conn, IMAP_SERVERGREET);
- imapc->idstr = "*"; /* we start off waiting for a '*' response */
- if(data->state.used_interface == Curl_if_multi)
- result = imap_multi_statemach(conn, done);
- else {
- result = imap_easy_statemach(conn);
- if(!result)
- *done = TRUE;
- }
- return result;
- }
- /***********************************************************************
- *
- * imap_done()
- *
- * The DONE function. This does what needs to be done after a single DO has
- * performed.
- *
- * Input argument is already checked for validity.
- */
- static CURLcode imap_done(struct connectdata *conn, CURLcode status,
- bool premature)
- {
- struct SessionHandle *data = conn->data;
- struct FTP *imap = data->state.proto.imap;
- CURLcode result=CURLE_OK;
- (void)premature;
- if(!imap)
- /* When the easy handle is removed from the multi while libcurl is still
- * trying to resolve the host name, it seems that the imap struct is not
- * yet initialized, but the removal action calls Curl_done() which calls
- * this function. So we simply return success if no imap pointer is set.
- */
- return CURLE_OK;
- if(status) {
- conn->bits.close = TRUE; /* marked for closure */
- result = status; /* use the already set error code */
- }
- /* clear these for next connection */
- imap->transfer = FTPTRANSFER_BODY;
- return result;
- }
- /***********************************************************************
- *
- * imap_perform()
- *
- * This is the actual DO function for IMAP. Get a file/directory according to
- * the options previously setup.
- */
- static
- CURLcode imap_perform(struct connectdata *conn,
- bool *connected, /* connect status after PASV / PORT */
- bool *dophase_done)
- {
- /* this is IMAP and no proxy */
- CURLcode result=CURLE_OK;
- DEBUGF(infof(conn->data, "DO phase starts\n"));
- if(conn->data->set.opt_no_body) {
- /* requested no body means no transfer... */
- struct FTP *imap = conn->data->state.proto.imap;
- imap->transfer = FTPTRANSFER_INFO;
- }
- *dophase_done = FALSE; /* not done yet */
- /* start the first command in the DO phase */
- result = imap_select(conn);
- if(result)
- return result;
- /* run the state-machine */
- if(conn->data->state.used_interface == Curl_if_multi)
- result = imap_multi_statemach(conn, dophase_done);
- else {
- result = imap_easy_statemach(conn);
- *dophase_done = TRUE; /* with the easy interface we are done here */
- }
- *connected = conn->bits.tcpconnect[FIRSTSOCKET];
- if(*dophase_done)
- DEBUGF(infof(conn->data, "DO phase is complete\n"));
- return result;
- }
- /***********************************************************************
- *
- * imap_do()
- *
- * This function is registered as 'curl_do' function. It decodes the path
- * parts etc as a wrapper to the actual DO function (imap_perform).
- *
- * The input argument is already checked for validity.
- */
- static CURLcode imap_do(struct connectdata *conn, bool *done)
- {
- CURLcode retcode = CURLE_OK;
- *done = FALSE; /* default to false */
- /*
- Since connections can be re-used between SessionHandles, this might be a
- connection already existing but on a fresh SessionHandle struct so we must
- make sure we have a good 'struct IMAP' to play with. For new connections,
- the struct IMAP is allocated and setup in the imap_connect() function.
- */
- Curl_reset_reqproto(conn);
- retcode = imap_init(conn);
- if(retcode)
- return retcode;
- retcode = imap_parse_url_path(conn);
- if(retcode)
- return retcode;
- retcode = imap_regular_transfer(conn, done);
- return retcode;
- }
- /***********************************************************************
- *
- * imap_logout()
- *
- * This should be called before calling sclose(). We should then wait for the
- * response from the server before returning. The calling code should then try
- * to close the connection.
- *
- */
- static CURLcode imap_logout(struct connectdata *conn)
- {
- CURLcode result = CURLE_OK;
- const char *str;
- str = getcmdid(conn);
- result = imapsendf(conn, str, "%s LOGOUT", str, NULL);
- if(result)
- return result;
- state(conn, IMAP_LOGOUT);
- result = imap_easy_statemach(conn);
- return result;
- }
- /***********************************************************************
- *
- * imap_disconnect()
- *
- * Disconnect from an IMAP server. Cleanup protocol-specific per-connection
- * resources. BLOCKING.
- */
- static CURLcode imap_disconnect(struct connectdata *conn, bool dead_connection)
- {
- struct imap_conn *imapc= &conn->proto.imapc;
- /* The IMAP session may or may not have been allocated/setup at this
- point! */
- if(!dead_connection && imapc->pp.conn)
- (void)imap_logout(conn); /* ignore errors on the LOGOUT */
- Curl_pp_disconnect(&imapc->pp);
- Curl_safefree(imapc->mailbox);
- return CURLE_OK;
- }
- /***********************************************************************
- *
- * imap_parse_url_path()
- *
- * Parse the URL path into separate path components.
- *
- */
- static CURLcode imap_parse_url_path(struct connectdata *conn)
- {
- /* the imap struct is already inited in imap_connect() */
- struct imap_conn *imapc = &conn->proto.imapc;
- struct SessionHandle *data = conn->data;
- const char *path = data->state.path;
- int len;
- if(!*path)
- path = "INBOX";
- /* url decode the path and use this mailbox */
- imapc->mailbox = curl_easy_unescape(data, path, 0, &len);
- if(!imapc->mailbox)
- return CURLE_OUT_OF_MEMORY;
- return CURLE_OK;
- }
- /* call this when the DO phase has completed */
- static CURLcode imap_dophase_done(struct connectdata *conn,
- bool connected)
- {
- struct FTP *imap = conn->data->state.proto.imap;
- (void)connected;
- if(imap->transfer != FTPTRANSFER_BODY)
- /* no data to transfer */
- Curl_setup_transfer(conn, -1, -1, FALSE, NULL, -1, NULL);
- return CURLE_OK;
- }
- /* called from multi.c while DOing */
- static CURLcode imap_doing(struct connectdata *conn,
- bool *dophase_done)
- {
- CURLcode result;
- result = imap_multi_statemach(conn, dophase_done);
- if(*dophase_done) {
- result = imap_dophase_done(conn, FALSE /* not connected */);
- DEBUGF(infof(conn->data, "DO phase is complete\n"));
- }
- return result;
- }
- /***********************************************************************
- *
- * imap_regular_transfer()
- *
- * The input argument is already checked for validity.
- *
- * Performs all commands done before a regular transfer between a local and a
- * remote host.
- *
- */
- static
- CURLcode imap_regular_transfer(struct connectdata *conn,
- bool *dophase_done)
- {
- CURLcode result=CURLE_OK;
- bool connected=FALSE;
- struct SessionHandle *data = conn->data;
- data->req.size = -1; /* make sure this is unknown at this point */
- Curl_pgrsSetUploadCounter(data, 0);
- Curl_pgrsSetDownloadCounter(data, 0);
- Curl_pgrsSetUploadSize(data, 0);
- Curl_pgrsSetDownloadSize(data, 0);
- result = imap_perform(conn,
- &connected, /* have we connected after PASV/PORT */
- dophase_done); /* all commands in the DO-phase done? */
- if(CURLE_OK == result) {
- if(!*dophase_done)
- /* the DO phase has not completed yet */
- return CURLE_OK;
- result = imap_dophase_done(conn, connected);
- if(result)
- return result;
- }
- return result;
- }
- static CURLcode imap_setup_connection(struct connectdata * conn)
- {
- struct SessionHandle *data = conn->data;
- if(conn->bits.httpproxy && !data->set.tunnel_thru_httpproxy) {
- /* Unless we have asked to tunnel imap operations through the proxy, we
- switch and use HTTP operations only */
- #ifndef CURL_DISABLE_HTTP
- if(conn->handler == &Curl_handler_imap)
- conn->handler = &Curl_handler_imap_proxy;
- else {
- #ifdef USE_SSL
- conn->handler = &Curl_handler_imaps_proxy;
- #else
- failf(data, "IMAPS not supported!");
- return CURLE_UNSUPPORTED_PROTOCOL;
- #endif
- }
- /*
- * We explicitly mark this connection as persistent here as we're doing
- * IMAP over HTTP and thus we accidentally avoid setting this value
- * otherwise.
- */
- conn->bits.close = FALSE;
- #else
- failf(data, "IMAP over http proxy requires HTTP support built-in!");
- return CURLE_UNSUPPORTED_PROTOCOL;
- #endif
- }
- data->state.path++; /* don't include the initial slash */
- return CURLE_OK;
- }
- #endif /* CURL_DISABLE_IMAP */
|