src/tool_writeenv.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2012, 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 http://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 #include "tool_setup.h"
  23
  24 #ifdef USE_ENVIRONMENT
  25
  26 #ifdef __riscos__
  27 #  include <kernel.h>
  28 #endif
  29
  30 #define _MPRINTF_REPLACE /* use our functions only */
  31 #include <curl/mprintf.h>
  32
  33 #include "tool_writeenv.h"
  34
  35 #include "memdebug.h" /* keep this as LAST include */
  36
  37 static const struct
  38 {
  39   const char * name;
  40   CURLINFO id;
  41   enum {
  42     writeenv_NONE,
  43     writeenv_DOUBLE,
  44     writeenv_LONG,
  45     writeenv_STRING
  46   } type;
  47 } variables[14] =
  48 {
  49   {"curl_url_effective", CURLINFO_EFFECTIVE_URL, writeenv_STRING},
  50   {"curl_http_code", CURLINFO_RESPONSE_CODE, writeenv_LONG},
  51   {"curl_time_total", CURLINFO_TOTAL_TIME, writeenv_DOUBLE},
  52   {"curl_time_namelookup", CURLINFO_NAMELOOKUP_TIME, writeenv_DOUBLE},
  53   {"curl_time_connect", CURLINFO_CONNECT_TIME, writeenv_DOUBLE},
  54   {"curl_time_pretransfer", CURLINFO_PRETRANSFER_TIME, writeenv_DOUBLE},
  55   {"curl_time_starttransfer", CURLINFO_STARTTRANSFER_TIME, writeenv_DOUBLE},
  56   {"curl_size_header", CURLINFO_HEADER_SIZE, writeenv_LONG},
  57   {"curl_size_request", CURLINFO_REQUEST_SIZE, writeenv_LONG},
  58   {"curl_size_download", CURLINFO_SIZE_DOWNLOAD, writeenv_DOUBLE},
  59   {"curl_size_upload", CURLINFO_SIZE_UPLOAD, writeenv_DOUBLE},
  60   {"curl_speed_download", CURLINFO_SPEED_DOWNLOAD, writeenv_DOUBLE},
  61   {"curl_speed_upload", CURLINFO_SPEED_UPLOAD, writeenv_DOUBLE},
  62   {NULL, 0, writeenv_NONE}
  63  };
  64
  65 static void internalSetEnv(const char * name, char * value)
  66 {
  67   /* Add your OS-specific code here. */
  68 #ifdef __riscos__
  69   _kernel_setenv(name, value);
  70 #elif defined (CURLDEBUG)
  71   curl_memlog("ENV %s = %s\n", name, value);
  72 #endif
  73   return;
  74 }
  75
  76 void ourWriteEnv(CURL *curl)
  77 {
  78   unsigned int i;
  79   char *string, numtext[10];
  80   long longinfo;
  81   double doubleinfo;
  82
  83   for(i=0; variables[i].name; i++) {
  84     switch (variables[i].type) {
  85     case writeenv_STRING:
  86       if(curl_easy_getinfo(curl, variables[i].id, &string) == CURLE_OK)
  87         internalSetEnv(variables[i].name, string);
  88       else
  89         internalSetEnv(variables[i].name, NULL);
  90       break;
  91
  92     case writeenv_LONG:
  93       if(curl_easy_getinfo(curl, variables[i].id, &longinfo) == CURLE_OK) {
  94         curl_msprintf(numtext, "%5ld", longinfo);
  95         internalSetEnv(variables[i].name, numtext);
  96       }
  97       else
  98         internalSetEnv(variables[i].name, NULL);
  99       break;
 100     case writeenv_DOUBLE:
 101       if(curl_easy_getinfo(curl, variables[i].id, &doubleinfo) == CURLE_OK) {
 102         curl_msprintf(numtext, "%6.2f", doubleinfo);
 103         internalSetEnv(variables[i].name, numtext);
 104       }
 105       else
 106         internalSetEnv(variables[i].name, NULL);
 107       break;
 108     default:
 109       break;
 110     }
 111   }
 112
 113   return;
 114 }
 115
 116 #endif