| 1 | /*************************************************************************** | 
|---|
| 2 | *                                  _   _ ____  _ | 
|---|
| 3 | *  Project                     ___| | | |  _ \| | | 
|---|
| 4 | *                             / __| | | | |_) | | | 
|---|
| 5 | *                            | (__| |_| |  _ <| |___ | 
|---|
| 6 | *                             \___|\___/|_| \_\_____| | 
|---|
| 7 | * | 
|---|
| 8 | * Copyright (C) 1998 - 2017, 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 | #include "tool_setup.h" | 
|---|
| 23 |  | 
|---|
| 24 | #include "slist_wc.h" | 
|---|
| 25 |  | 
|---|
| 26 | #ifndef CURL_DISABLE_LIBCURL_OPTION | 
|---|
| 27 |  | 
|---|
| 28 | #define ENABLE_CURLX_PRINTF | 
|---|
| 29 | /* use our own printf() functions */ | 
|---|
| 30 | #include "curlx.h" | 
|---|
| 31 |  | 
|---|
| 32 | #include "tool_cfgable.h" | 
|---|
| 33 | #include "tool_easysrc.h" | 
|---|
| 34 | #include "tool_msgs.h" | 
|---|
| 35 |  | 
|---|
| 36 | #include "memdebug.h" /* keep this as LAST include */ | 
|---|
| 37 |  | 
|---|
| 38 | /* global variable definitions, for easy-interface source code generation */ | 
|---|
| 39 |  | 
|---|
| 40 | struct slist_wc *easysrc_decl = NULL; /* Variable declarations */ | 
|---|
| 41 | struct slist_wc *easysrc_data = NULL; /* Build slists, forms etc. */ | 
|---|
| 42 | struct slist_wc *easysrc_code = NULL; /* Setopt calls */ | 
|---|
| 43 | struct slist_wc *easysrc_toohard = NULL; /* Unconvertible setopt */ | 
|---|
| 44 | struct slist_wc *easysrc_clean = NULL;  /* Clean up allocated data */ | 
|---|
| 45 | int easysrc_mime_count = 0; | 
|---|
| 46 | int easysrc_slist_count = 0; | 
|---|
| 47 |  | 
|---|
| 48 | static const char *const srchead[]={ | 
|---|
| 49 | "/********* Sample code generated by the curl command line tool **********", | 
|---|
| 50 | " * All curl_easy_setopt() options are documented at:", | 
|---|
| 51 | " * https://curl.haxx.se/libcurl/c/curl_easy_setopt.html", | 
|---|
| 52 | " ************************************************************************/", | 
|---|
| 53 | "#include <curl/curl.h>", | 
|---|
| 54 | "", | 
|---|
| 55 | "int main(int argc, char *argv[])", | 
|---|
| 56 | "{", | 
|---|
| 57 | "  CURLcode ret;", | 
|---|
| 58 | "  CURL *hnd;", | 
|---|
| 59 | NULL | 
|---|
| 60 | }; | 
|---|
| 61 | /* easysrc_decl declarations come here */ | 
|---|
| 62 | /* easysrc_data initialisations come here */ | 
|---|
| 63 | /* easysrc_code statements come here */ | 
|---|
| 64 | static const char *const srchard[]={ | 
|---|
| 65 | "/* Here is a list of options the curl code used that cannot get generated", | 
|---|
| 66 | "   as source easily. You may select to either not use them or implement", | 
|---|
| 67 | "   them yourself.", | 
|---|
| 68 | "", | 
|---|
| 69 | NULL | 
|---|
| 70 | }; | 
|---|
| 71 | static const char *const srcend[]={ | 
|---|
| 72 | "", | 
|---|
| 73 | "  return (int)ret;", | 
|---|
| 74 | "}", | 
|---|
| 75 | "/**** End of sample code ****/", | 
|---|
| 76 | NULL | 
|---|
| 77 | }; | 
|---|
| 78 |  | 
|---|
| 79 | /* Clean up all source code if we run out of memory */ | 
|---|
| 80 | static void easysrc_free(void) | 
|---|
| 81 | { | 
|---|
| 82 | slist_wc_free_all(easysrc_decl); | 
|---|
| 83 | easysrc_decl = NULL; | 
|---|
| 84 | slist_wc_free_all(easysrc_data); | 
|---|
| 85 | easysrc_data = NULL; | 
|---|
| 86 | slist_wc_free_all(easysrc_code); | 
|---|
| 87 | easysrc_code = NULL; | 
|---|
| 88 | slist_wc_free_all(easysrc_toohard); | 
|---|
| 89 | easysrc_toohard = NULL; | 
|---|
| 90 | slist_wc_free_all(easysrc_clean); | 
|---|
| 91 | easysrc_clean = NULL; | 
|---|
| 92 | } | 
|---|
| 93 |  | 
|---|
| 94 | /* Add a source line to the main code or remarks */ | 
|---|
| 95 | CURLcode easysrc_add(struct slist_wc **plist, const char *line) | 
|---|
| 96 | { | 
|---|
| 97 | CURLcode ret = CURLE_OK; | 
|---|
| 98 | struct slist_wc *list = slist_wc_append(*plist, line); | 
|---|
| 99 | if(!list) { | 
|---|
| 100 | easysrc_free(); | 
|---|
| 101 | ret = CURLE_OUT_OF_MEMORY; | 
|---|
| 102 | } | 
|---|
| 103 | else | 
|---|
| 104 | *plist = list; | 
|---|
| 105 | return ret; | 
|---|
| 106 | } | 
|---|
| 107 |  | 
|---|
| 108 | CURLcode easysrc_addf(struct slist_wc **plist, const char *fmt, ...) | 
|---|
| 109 | { | 
|---|
| 110 | CURLcode ret; | 
|---|
| 111 | char *bufp; | 
|---|
| 112 | va_list ap; | 
|---|
| 113 | va_start(ap, fmt); | 
|---|
| 114 | bufp = curlx_mvaprintf(fmt, ap); | 
|---|
| 115 | va_end(ap); | 
|---|
| 116 | if(! bufp) { | 
|---|
| 117 | ret = CURLE_OUT_OF_MEMORY; | 
|---|
| 118 | } | 
|---|
| 119 | else { | 
|---|
| 120 | ret = easysrc_add(plist, bufp); | 
|---|
| 121 | curl_free(bufp); | 
|---|
| 122 | } | 
|---|
| 123 | return ret; | 
|---|
| 124 | } | 
|---|
| 125 |  | 
|---|
| 126 | #define CHKRET(v) do {CURLcode ret = (v); if(ret) return ret;} WHILE_FALSE | 
|---|
| 127 |  | 
|---|
| 128 | CURLcode easysrc_init(void) | 
|---|
| 129 | { | 
|---|
| 130 | CHKRET(easysrc_add(&easysrc_code, | 
|---|
| 131 | "hnd = curl_easy_init();")); | 
|---|
| 132 | return CURLE_OK; | 
|---|
| 133 | } | 
|---|
| 134 |  | 
|---|
| 135 | CURLcode easysrc_perform(void) | 
|---|
| 136 | { | 
|---|
| 137 | /* Note any setopt calls which we could not convert */ | 
|---|
| 138 | if(easysrc_toohard) { | 
|---|
| 139 | int i; | 
|---|
| 140 | struct curl_slist *ptr; | 
|---|
| 141 | const char *c; | 
|---|
| 142 | CHKRET(easysrc_add(&easysrc_code, "")); | 
|---|
| 143 | /* Preamble comment */ | 
|---|
| 144 | for(i = 0; ((c = srchard[i]) != NULL); i++) | 
|---|
| 145 | CHKRET(easysrc_add(&easysrc_code, c)); | 
|---|
| 146 | /* Each unconverted option */ | 
|---|
| 147 | if(easysrc_toohard) { | 
|---|
| 148 | for(ptr = easysrc_toohard->first; ptr; ptr = ptr->next) | 
|---|
| 149 | CHKRET(easysrc_add(&easysrc_code, ptr->data)); | 
|---|
| 150 | } | 
|---|
| 151 | CHKRET(easysrc_add(&easysrc_code, "")); | 
|---|
| 152 | CHKRET(easysrc_add(&easysrc_code, "*/")); | 
|---|
| 153 |  | 
|---|
| 154 | slist_wc_free_all(easysrc_toohard); | 
|---|
| 155 | easysrc_toohard = NULL; | 
|---|
| 156 | } | 
|---|
| 157 |  | 
|---|
| 158 | CHKRET(easysrc_add(&easysrc_code, "")); | 
|---|
| 159 | CHKRET(easysrc_add(&easysrc_code, "ret = curl_easy_perform(hnd);")); | 
|---|
| 160 | CHKRET(easysrc_add(&easysrc_code, "")); | 
|---|
| 161 |  | 
|---|
| 162 | return CURLE_OK; | 
|---|
| 163 | } | 
|---|
| 164 |  | 
|---|
| 165 | CURLcode easysrc_cleanup(void) | 
|---|
| 166 | { | 
|---|
| 167 | CHKRET(easysrc_add(&easysrc_code, "curl_easy_cleanup(hnd);")); | 
|---|
| 168 | CHKRET(easysrc_add(&easysrc_code, "hnd = NULL;")); | 
|---|
| 169 |  | 
|---|
| 170 | return CURLE_OK; | 
|---|
| 171 | } | 
|---|
| 172 |  | 
|---|
| 173 | void dumpeasysrc(struct GlobalConfig *config) | 
|---|
| 174 | { | 
|---|
| 175 | struct curl_slist *ptr; | 
|---|
| 176 | char *o = config->libcurl; | 
|---|
| 177 |  | 
|---|
| 178 | FILE *out; | 
|---|
| 179 | bool fopened = FALSE; | 
|---|
| 180 | if(strcmp(o, "-")) { | 
|---|
| 181 | out = fopen(o, FOPEN_WRITETEXT); | 
|---|
| 182 | fopened = TRUE; | 
|---|
| 183 | } | 
|---|
| 184 | else | 
|---|
| 185 | out = stdout; | 
|---|
| 186 | if(!out) | 
|---|
| 187 | warnf(config, "Failed to open %s to write libcurl code!\n", o); | 
|---|
| 188 | else { | 
|---|
| 189 | int i; | 
|---|
| 190 | const char *c; | 
|---|
| 191 |  | 
|---|
| 192 | for(i = 0; ((c = srchead[i]) != NULL); i++) | 
|---|
| 193 | fprintf(out, "%s\n", c); | 
|---|
| 194 |  | 
|---|
| 195 | /* Declare variables used for complex setopt values */ | 
|---|
| 196 | if(easysrc_decl) { | 
|---|
| 197 | for(ptr = easysrc_decl->first; ptr; ptr = ptr->next) | 
|---|
| 198 | fprintf(out, "  %s\n", ptr->data); | 
|---|
| 199 | } | 
|---|
| 200 |  | 
|---|
| 201 | /* Set up complex values for setopt calls */ | 
|---|
| 202 | if(easysrc_data) { | 
|---|
| 203 | fprintf(out, "\n"); | 
|---|
| 204 |  | 
|---|
| 205 | for(ptr = easysrc_data->first; ptr; ptr = ptr->next) | 
|---|
| 206 | fprintf(out, "  %s\n", ptr->data); | 
|---|
| 207 | } | 
|---|
| 208 |  | 
|---|
| 209 | fprintf(out, "\n"); | 
|---|
| 210 | if(easysrc_code) { | 
|---|
| 211 | for(ptr = easysrc_code->first; ptr; ptr = ptr->next) { | 
|---|
| 212 | if(ptr->data[0]) { | 
|---|
| 213 | fprintf(out, "  %s\n", ptr->data); | 
|---|
| 214 | } | 
|---|
| 215 | else { | 
|---|
| 216 | fprintf(out, "\n"); | 
|---|
| 217 | } | 
|---|
| 218 | } | 
|---|
| 219 | } | 
|---|
| 220 |  | 
|---|
| 221 | if(easysrc_clean) { | 
|---|
| 222 | for(ptr = easysrc_clean->first; ptr; ptr = ptr->next) | 
|---|
| 223 | fprintf(out, "  %s\n", ptr->data); | 
|---|
| 224 | } | 
|---|
| 225 |  | 
|---|
| 226 | for(i = 0; ((c = srcend[i]) != NULL); i++) | 
|---|
| 227 | fprintf(out, "%s\n", c); | 
|---|
| 228 |  | 
|---|
| 229 | if(fopened) | 
|---|
| 230 | fclose(out); | 
|---|
| 231 | } | 
|---|
| 232 |  | 
|---|
| 233 | easysrc_free(); | 
|---|
| 234 | } | 
|---|
| 235 |  | 
|---|
| 236 | #endif /* CURL_DISABLE_LIBCURL_OPTION */ | 
|---|
| 237 |  | 
|---|