Utilities/cmcurl/lib/fopen.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2022, 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.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  * SPDX-License-Identifier: curl
  22  *
  23  ***************************************************************************/
  24
  25 #include "curl_setup.h"
  26
  27 #if !defined(CURL_DISABLE_COOKIES) || !defined(CURL_DISABLE_ALTSVC) ||  \
  28   !defined(CURL_DISABLE_HSTS)
  29
  30 #ifdef HAVE_FCNTL_H
  31 #include <fcntl.h>
  32 #endif
  33
  34 #include "urldata.h"
  35 #include "rand.h"
  36 #include "fopen.h"
  37 /* The last 3 #include files should be in this order */
  38 #include "curl_printf.h"
  39 #include "curl_memory.h"
  40 #include "memdebug.h"
  41
  42 /*
  43  * Curl_fopen() opens a file for writing with a temp name, to be renamed
  44  * to the final name when completed. If there is an existing file using this
  45  * name at the time of the open, this function will clone the mode from that
  46  * file.  if 'tempname' is non-NULL, it needs a rename after the file is
  47  * written.
  48  */
  49 CURLcode Curl_fopen(struct Curl_easy *data, const char *filename,
  50                     FILE **fh, char **tempname)
  51 {
  52   CURLcode result = CURLE_WRITE_ERROR;
  53   unsigned char randsuffix[9];
  54   char *tempstore = NULL;
  55   struct_stat sb;
  56   int fd = -1;
  57   *tempname = NULL;
  58
  59   if(stat(filename, &sb) == -1 || !S_ISREG(sb.st_mode)) {
  60     /* a non-regular file, fallback to direct fopen() */
  61     *fh = fopen(filename, FOPEN_WRITETEXT);
  62     if(*fh)
  63       return CURLE_OK;
  64     goto fail;
  65   }
  66
  67   result = Curl_rand_hex(data, randsuffix, sizeof(randsuffix));
  68   if(result)
  69     goto fail;
  70
  71   tempstore = aprintf("%s.%s.tmp", filename, randsuffix);
  72   if(!tempstore) {
  73     result = CURLE_OUT_OF_MEMORY;
  74     goto fail;
  75   }
  76
  77   result = CURLE_WRITE_ERROR;
  78   fd = open(tempstore, O_WRONLY | O_CREAT | O_EXCL, 0600);
  79   if(fd == -1)
  80     goto fail;
  81
  82 #ifdef HAVE_FCHMOD
  83   {
  84     struct_stat nsb;
  85     if((fstat(fd, &nsb) != -1) &&
  86        (nsb.st_uid == sb.st_uid) && (nsb.st_gid == sb.st_gid)) {
  87       /* if the user and group are the same, clone the original mode */
  88       if(fchmod(fd, sb.st_mode) == -1)
  89         goto fail;
  90     }
  91   }
  92 #endif
  93
  94   *fh = fdopen(fd, FOPEN_WRITETEXT);
  95   if(!*fh)
  96     goto fail;
  97
  98   *tempname = tempstore;
  99   return CURLE_OK;
 100
 101 fail:
 102   if(fd != -1) {
 103     close(fd);
 104     unlink(tempstore);
 105   }
 106
 107   free(tempstore);
 108
 109   *tempname = NULL;
 110   return result;
 111 }
 112
 113 #endif /* ! disabled */