123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) 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 https://curl.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.
- *
- * SPDX-License-Identifier: curl
- *
- ***************************************************************************/
- #include "tool_setup.h"
- #include <sys/stat.h>
- #ifdef WIN32
- #include <tchar.h>
- #endif
- #ifdef HAVE_SIGNAL_H
- #include <signal.h>
- #endif
- #ifdef HAVE_FCNTL_H
- #include <fcntl.h>
- #endif
- #ifdef USE_NSS
- #include <nspr.h>
- #include <plarenas.h>
- #endif
- #define ENABLE_CURLX_PRINTF
- /* use our own printf() functions */
- #include "curlx.h"
- #include "tool_cfgable.h"
- #include "tool_doswin.h"
- #include "tool_msgs.h"
- #include "tool_operate.h"
- #include "tool_vms.h"
- #include "tool_main.h"
- #include "tool_libinfo.h"
- /*
- * This is low-level hard-hacking memory leak tracking and similar. Using
- * the library level code from this client-side is ugly, but we do this
- * anyway for convenience.
- */
- #include "memdebug.h" /* keep this as LAST include */
- #ifdef __VMS
- /*
- * vms_show is a global variable, used in main() as parameter for
- * function vms_special_exit() to allow proper curl tool exiting.
- * Its value may be set in other tool_*.c source files thanks to
- * forward declaration present in tool_vms.h
- */
- int vms_show = 0;
- #endif
- #ifdef __MINGW32__
- /*
- * There seems to be no way to escape "*" in command-line arguments with MinGW
- * when command-line argument globbing is enabled under the MSYS shell, so turn
- * it off.
- */
- int _CRT_glob = 0;
- #endif /* __MINGW32__ */
- /* if we build a static library for unit tests, there is no main() function */
- #ifndef UNITTESTS
- #if defined(HAVE_PIPE) && defined(HAVE_FCNTL)
- /*
- * Ensure that file descriptors 0, 1 and 2 (stdin, stdout, stderr) are
- * open before starting to run. Otherwise, the first three network
- * sockets opened by curl could be used for input sources, downloaded data
- * or error logs as they will effectively be stdin, stdout and/or stderr.
- *
- * fcntl's F_GETFD instruction returns -1 if the file descriptor is closed,
- * otherwise it returns "the file descriptor flags (which typically can only
- * be FD_CLOEXEC, which is not set here).
- */
- static int main_checkfds(void)
- {
- int fd[2];
- while((fcntl(STDIN_FILENO, F_GETFD) == -1) ||
- (fcntl(STDOUT_FILENO, F_GETFD) == -1) ||
- (fcntl(STDERR_FILENO, F_GETFD) == -1))
- if(pipe(fd))
- return 1;
- return 0;
- }
- #else
- #define main_checkfds() 0
- #endif
- #ifdef CURLDEBUG
- static void memory_tracking_init(void)
- {
- char *env;
- /* if CURL_MEMDEBUG is set, this starts memory tracking message logging */
- env = curlx_getenv("CURL_MEMDEBUG");
- if(env) {
- /* use the value as file name */
- char fname[CURL_MT_LOGFNAME_BUFSIZE];
- if(strlen(env) >= CURL_MT_LOGFNAME_BUFSIZE)
- env[CURL_MT_LOGFNAME_BUFSIZE-1] = '\0';
- strcpy(fname, env);
- curl_free(env);
- curl_dbg_memdebug(fname);
- /* this weird stuff here is to make curl_free() get called before
- curl_dbg_memdebug() as otherwise memory tracking will log a free()
- without an alloc! */
- }
- /* if CURL_MEMLIMIT is set, this enables fail-on-alloc-number-N feature */
- env = curlx_getenv("CURL_MEMLIMIT");
- if(env) {
- char *endptr;
- long num = strtol(env, &endptr, 10);
- if((endptr != env) && (endptr == env + strlen(env)) && (num > 0))
- curl_dbg_memlimit(num);
- curl_free(env);
- }
- }
- #else
- # define memory_tracking_init() Curl_nop_stmt
- #endif
- /*
- * This is the main global constructor for the app. Call this before
- * _any_ libcurl usage. If this fails, *NO* libcurl functions may be
- * used, or havoc may be the result.
- */
- static CURLcode main_init(struct GlobalConfig *config)
- {
- CURLcode result = CURLE_OK;
- #if defined(__DJGPP__) || defined(__GO32__)
- /* stop stat() wasting time */
- _djstat_flags |= _STAT_INODE | _STAT_EXEC_MAGIC | _STAT_DIRSIZE;
- #endif
- /* Initialise the global config */
- config->showerror = -1; /* Will show errors */
- config->errors = stderr; /* Default errors to stderr */
- config->styled_output = TRUE; /* enable detection */
- config->parallel_max = PARALLEL_DEFAULT;
- /* Allocate the initial operate config */
- config->first = config->last = malloc(sizeof(struct OperationConfig));
- if(config->first) {
- /* Perform the libcurl initialization */
- result = curl_global_init(CURL_GLOBAL_DEFAULT);
- if(!result) {
- /* Get information about libcurl */
- result = get_libcurl_info();
- if(!result) {
- /* Initialise the config */
- config_init(config->first);
- config->first->global = config;
- }
- else {
- errorf(config, "error retrieving curl library information\n");
- free(config->first);
- }
- }
- else {
- errorf(config, "error initializing curl library\n");
- free(config->first);
- }
- }
- else {
- errorf(config, "error initializing curl\n");
- result = CURLE_FAILED_INIT;
- }
- return result;
- }
- static void free_globalconfig(struct GlobalConfig *config)
- {
- Curl_safefree(config->trace_dump);
- if(config->errors_fopened && config->errors)
- fclose(config->errors);
- config->errors = NULL;
- if(config->trace_fopened && config->trace_stream)
- fclose(config->trace_stream);
- config->trace_stream = NULL;
- Curl_safefree(config->libcurl);
- }
- /*
- * This is the main global destructor for the app. Call this after
- * _all_ libcurl usage is done.
- */
- static void main_free(struct GlobalConfig *config)
- {
- /* Cleanup the easy handle */
- /* Main cleanup */
- curl_global_cleanup();
- #ifdef USE_NSS
- if(PR_Initialized()) {
- /* prevent valgrind from reporting still reachable mem from NSPR arenas */
- PL_ArenaFinish();
- /* prevent valgrind from reporting possibly lost memory (fd cache, ...) */
- PR_Cleanup();
- }
- #endif
- free_globalconfig(config);
- /* Free the config structures */
- config_free(config->last);
- config->first = NULL;
- config->last = NULL;
- }
- /*
- ** curl tool main function.
- */
- #ifdef _UNICODE
- int wmain(int argc, wchar_t *argv[])
- #else
- int main(int argc, char *argv[])
- #endif
- {
- CURLcode result = CURLE_OK;
- struct GlobalConfig global;
- memset(&global, 0, sizeof(global));
- #ifdef WIN32
- /* Undocumented diagnostic option to list the full paths of all loaded
- modules. This is purposely pre-init. */
- if(argc == 2 && !_tcscmp(argv[1], _T("--dump-module-paths"))) {
- struct curl_slist *item, *head = GetLoadedModulePaths();
- for(item = head; item; item = item->next)
- printf("%s\n", item->data);
- curl_slist_free_all(head);
- return head ? 0 : 1;
- }
- /* win32_init must be called before other init routines. */
- result = win32_init();
- if(result) {
- fprintf(stderr, "curl: (%d) Windows-specific init failed.\n", result);
- return result;
- }
- #endif
- if(main_checkfds()) {
- fprintf(stderr, "curl: out of file descriptors\n");
- return CURLE_FAILED_INIT;
- }
- #if defined(HAVE_SIGNAL) && defined(SIGPIPE)
- (void)signal(SIGPIPE, SIG_IGN);
- #endif
- /* Initialize memory tracking */
- memory_tracking_init();
- /* Initialize the curl library - do not call any libcurl functions before
- this point */
- result = main_init(&global);
- if(!result) {
- /* Start our curl operation */
- result = operate(&global, argc, argv);
- /* Perform the main cleanup */
- main_free(&global);
- }
- #ifdef WIN32
- /* Flush buffers of all streams opened in write or update mode */
- fflush(NULL);
- #endif
- #ifdef __VMS
- vms_special_exit(result, vms_show);
- #else
- return (int)result;
- #endif
- }
- #endif /* ndef UNITTESTS */
|