Annotation of embedaddon/curl/docs/examples/http2-download.c, revision 1.1

1.1     ! misho       1: /***************************************************************************
        !             2:  *                                  _   _ ____  _
        !             3:  *  Project                     ___| | | |  _ \| |
        !             4:  *                             / __| | | | |_) | |
        !             5:  *                            | (__| |_| |  _ <| |___
        !             6:  *                             \___|\___/|_| \_\_____|
        !             7:  *
        !             8:  * Copyright (C) 1998 - 2019, Daniel Stenberg, <daniel@haxx.se>, et al.
        !             9:  *
        !            10:  * This software is licensed as described in the file COPYING, which
        !            11:  * you should have received as part of this distribution. The terms
        !            12:  * are also available at https://curl.haxx.se/docs/copyright.html.
        !            13:  *
        !            14:  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
        !            15:  * copies of the Software, and permit persons to whom the Software is
        !            16:  * furnished to do so, under the terms of the COPYING file.
        !            17:  *
        !            18:  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
        !            19:  * KIND, either express or implied.
        !            20:  *
        !            21:  ***************************************************************************/
        !            22: /* <DESC>
        !            23:  * Multiplexed HTTP/2 downloads over a single connection
        !            24:  * </DESC>
        !            25:  */
        !            26: #include <stdio.h>
        !            27: #include <stdlib.h>
        !            28: #include <string.h>
        !            29: 
        !            30: /* somewhat unix-specific */
        !            31: #include <sys/time.h>
        !            32: #include <unistd.h>
        !            33: 
        !            34: /* curl stuff */
        !            35: #include <curl/curl.h>
        !            36: 
        !            37: #ifndef CURLPIPE_MULTIPLEX
        !            38: /* This little trick will just make sure that we don't enable pipelining for
        !            39:    libcurls old enough to not have this symbol. It is _not_ defined to zero in
        !            40:    a recent libcurl header. */
        !            41: #define CURLPIPE_MULTIPLEX 0
        !            42: #endif
        !            43: 
        !            44: struct transfer {
        !            45:   CURL *easy;
        !            46:   unsigned int num;
        !            47:   FILE *out;
        !            48: };
        !            49: 
        !            50: #define NUM_HANDLES 1000
        !            51: 
        !            52: static
        !            53: void dump(const char *text, int num, unsigned char *ptr, size_t size,
        !            54:           char nohex)
        !            55: {
        !            56:   size_t i;
        !            57:   size_t c;
        !            58: 
        !            59:   unsigned int width = 0x10;
        !            60: 
        !            61:   if(nohex)
        !            62:     /* without the hex output, we can fit more on screen */
        !            63:     width = 0x40;
        !            64: 
        !            65:   fprintf(stderr, "%d %s, %lu bytes (0x%lx)\n",
        !            66:           num, text, (unsigned long)size, (unsigned long)size);
        !            67: 
        !            68:   for(i = 0; i<size; i += width) {
        !            69: 
        !            70:     fprintf(stderr, "%4.4lx: ", (unsigned long)i);
        !            71: 
        !            72:     if(!nohex) {
        !            73:       /* hex not disabled, show it */
        !            74:       for(c = 0; c < width; c++)
        !            75:         if(i + c < size)
        !            76:           fprintf(stderr, "%02x ", ptr[i + c]);
        !            77:         else
        !            78:           fputs("   ", stderr);
        !            79:     }
        !            80: 
        !            81:     for(c = 0; (c < width) && (i + c < size); c++) {
        !            82:       /* check for 0D0A; if found, skip past and start a new line of output */
        !            83:       if(nohex && (i + c + 1 < size) && ptr[i + c] == 0x0D &&
        !            84:          ptr[i + c + 1] == 0x0A) {
        !            85:         i += (c + 2 - width);
        !            86:         break;
        !            87:       }
        !            88:       fprintf(stderr, "%c",
        !            89:               (ptr[i + c] >= 0x20) && (ptr[i + c]<0x80)?ptr[i + c]:'.');
        !            90:       /* check again for 0D0A, to avoid an extra \n if it's at width */
        !            91:       if(nohex && (i + c + 2 < size) && ptr[i + c + 1] == 0x0D &&
        !            92:          ptr[i + c + 2] == 0x0A) {
        !            93:         i += (c + 3 - width);
        !            94:         break;
        !            95:       }
        !            96:     }
        !            97:     fputc('\n', stderr); /* newline */
        !            98:   }
        !            99: }
        !           100: 
        !           101: static
        !           102: int my_trace(CURL *handle, curl_infotype type,
        !           103:              char *data, size_t size,
        !           104:              void *userp)
        !           105: {
        !           106:   const char *text;
        !           107:   struct transfer *t = (struct transfer *)userp;
        !           108:   unsigned int num = t->num;
        !           109:   (void)handle; /* prevent compiler warning */
        !           110: 
        !           111:   switch(type) {
        !           112:   case CURLINFO_TEXT:
        !           113:     fprintf(stderr, "== %u Info: %s", num, data);
        !           114:     /* FALLTHROUGH */
        !           115:   default: /* in case a new one is introduced to shock us */
        !           116:     return 0;
        !           117: 
        !           118:   case CURLINFO_HEADER_OUT:
        !           119:     text = "=> Send header";
        !           120:     break;
        !           121:   case CURLINFO_DATA_OUT:
        !           122:     text = "=> Send data";
        !           123:     break;
        !           124:   case CURLINFO_SSL_DATA_OUT:
        !           125:     text = "=> Send SSL data";
        !           126:     break;
        !           127:   case CURLINFO_HEADER_IN:
        !           128:     text = "<= Recv header";
        !           129:     break;
        !           130:   case CURLINFO_DATA_IN:
        !           131:     text = "<= Recv data";
        !           132:     break;
        !           133:   case CURLINFO_SSL_DATA_IN:
        !           134:     text = "<= Recv SSL data";
        !           135:     break;
        !           136:   }
        !           137: 
        !           138:   dump(text, num, (unsigned char *)data, size, 1);
        !           139:   return 0;
        !           140: }
        !           141: 
        !           142: static void setup(struct transfer *t, int num)
        !           143: {
        !           144:   char filename[128];
        !           145:   CURL *hnd;
        !           146: 
        !           147:   hnd = t->easy = curl_easy_init();
        !           148: 
        !           149:   snprintf(filename, 128, "dl-%d", num);
        !           150: 
        !           151:   t->out = fopen(filename, "wb");
        !           152: 
        !           153:   /* write to this file */
        !           154:   curl_easy_setopt(hnd, CURLOPT_WRITEDATA, t->out);
        !           155: 
        !           156:   /* set the same URL */
        !           157:   curl_easy_setopt(hnd, CURLOPT_URL, "https://localhost:8443/index.html");
        !           158: 
        !           159:   /* please be verbose */
        !           160:   curl_easy_setopt(hnd, CURLOPT_VERBOSE, 1L);
        !           161:   curl_easy_setopt(hnd, CURLOPT_DEBUGFUNCTION, my_trace);
        !           162:   curl_easy_setopt(hnd, CURLOPT_DEBUGDATA, t);
        !           163: 
        !           164:   /* HTTP/2 please */
        !           165:   curl_easy_setopt(hnd, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_2_0);
        !           166: 
        !           167:   /* we use a self-signed test server, skip verification during debugging */
        !           168:   curl_easy_setopt(hnd, CURLOPT_SSL_VERIFYPEER, 0L);
        !           169:   curl_easy_setopt(hnd, CURLOPT_SSL_VERIFYHOST, 0L);
        !           170: 
        !           171: #if (CURLPIPE_MULTIPLEX > 0)
        !           172:   /* wait for pipe connection to confirm */
        !           173:   curl_easy_setopt(hnd, CURLOPT_PIPEWAIT, 1L);
        !           174: #endif
        !           175: }
        !           176: 
        !           177: /*
        !           178:  * Download many transfers over HTTP/2, using the same connection!
        !           179:  */
        !           180: int main(int argc, char **argv)
        !           181: {
        !           182:   struct transfer trans[NUM_HANDLES];
        !           183:   CURLM *multi_handle;
        !           184:   int i;
        !           185:   int still_running = 0; /* keep number of running handles */
        !           186:   int num_transfers;
        !           187:   if(argc > 1) {
        !           188:     /* if given a number, do that many transfers */
        !           189:     num_transfers = atoi(argv[1]);
        !           190:     if((num_transfers < 1) || (num_transfers > NUM_HANDLES))
        !           191:       num_transfers = 3; /* a suitable low default */
        !           192:   }
        !           193:   else
        !           194:     num_transfers = 3; /* suitable default */
        !           195: 
        !           196:   /* init a multi stack */
        !           197:   multi_handle = curl_multi_init();
        !           198: 
        !           199:   for(i = 0; i < num_transfers; i++) {
        !           200:     setup(&trans[i], i);
        !           201: 
        !           202:     /* add the individual transfer */
        !           203:     curl_multi_add_handle(multi_handle, trans[i].easy);
        !           204:   }
        !           205: 
        !           206:   curl_multi_setopt(multi_handle, CURLMOPT_PIPELINING, CURLPIPE_MULTIPLEX);
        !           207: 
        !           208:   /* we start some action by calling perform right away */
        !           209:   curl_multi_perform(multi_handle, &still_running);
        !           210: 
        !           211:   while(still_running) {
        !           212:     struct timeval timeout;
        !           213:     int rc; /* select() return code */
        !           214:     CURLMcode mc; /* curl_multi_fdset() return code */
        !           215: 
        !           216:     fd_set fdread;
        !           217:     fd_set fdwrite;
        !           218:     fd_set fdexcep;
        !           219:     int maxfd = -1;
        !           220: 
        !           221:     long curl_timeo = -1;
        !           222: 
        !           223:     FD_ZERO(&fdread);
        !           224:     FD_ZERO(&fdwrite);
        !           225:     FD_ZERO(&fdexcep);
        !           226: 
        !           227:     /* set a suitable timeout to play around with */
        !           228:     timeout.tv_sec = 1;
        !           229:     timeout.tv_usec = 0;
        !           230: 
        !           231:     curl_multi_timeout(multi_handle, &curl_timeo);
        !           232:     if(curl_timeo >= 0) {
        !           233:       timeout.tv_sec = curl_timeo / 1000;
        !           234:       if(timeout.tv_sec > 1)
        !           235:         timeout.tv_sec = 1;
        !           236:       else
        !           237:         timeout.tv_usec = (curl_timeo % 1000) * 1000;
        !           238:     }
        !           239: 
        !           240:     /* get file descriptors from the transfers */
        !           241:     mc = curl_multi_fdset(multi_handle, &fdread, &fdwrite, &fdexcep, &maxfd);
        !           242: 
        !           243:     if(mc != CURLM_OK) {
        !           244:       fprintf(stderr, "curl_multi_fdset() failed, code %d.\n", mc);
        !           245:       break;
        !           246:     }
        !           247: 
        !           248:     /* On success the value of maxfd is guaranteed to be >= -1. We call
        !           249:        select(maxfd + 1, ...); specially in case of (maxfd == -1) there are
        !           250:        no fds ready yet so we call select(0, ...) --or Sleep() on Windows--
        !           251:        to sleep 100ms, which is the minimum suggested value in the
        !           252:        curl_multi_fdset() doc. */
        !           253: 
        !           254:     if(maxfd == -1) {
        !           255: #ifdef _WIN32
        !           256:       Sleep(100);
        !           257:       rc = 0;
        !           258: #else
        !           259:       /* Portable sleep for platforms other than Windows. */
        !           260:       struct timeval wait = { 0, 100 * 1000 }; /* 100ms */
        !           261:       rc = select(0, NULL, NULL, NULL, &wait);
        !           262: #endif
        !           263:     }
        !           264:     else {
        !           265:       /* Note that on some platforms 'timeout' may be modified by select().
        !           266:          If you need access to the original value save a copy beforehand. */
        !           267:       rc = select(maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout);
        !           268:     }
        !           269: 
        !           270:     switch(rc) {
        !           271:     case -1:
        !           272:       /* select error */
        !           273:       break;
        !           274:     case 0:
        !           275:     default:
        !           276:       /* timeout or readable/writable sockets */
        !           277:       curl_multi_perform(multi_handle, &still_running);
        !           278:       break;
        !           279:     }
        !           280:   }
        !           281: 
        !           282:   for(i = 0; i < num_transfers; i++) {
        !           283:     curl_multi_remove_handle(multi_handle, trans[i].easy);
        !           284:     curl_easy_cleanup(trans[i].easy);
        !           285:   }
        !           286: 
        !           287:   curl_multi_cleanup(multi_handle);
        !           288: 
        !           289:   return 0;
        !           290: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>