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
37static 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
65static 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
76void 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
117