folly/portability/Stdio.cpp

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <folly/portability/Stdio.h>
  18
  19 #ifdef _WIN32
  20 #include <cstdlib>
  21 #include <folly/ScopeGuard.h>
  22 #include <folly/portability/Unistd.h>
  23
  24 extern "C" {
  25 int dprintf(int fd, const char* fmt, ...) {
  26   va_list args;
  27   va_start(args, fmt);
  28   SCOPE_EXIT { va_end(args); };
  29
  30   int len = vsnprintf(nullptr, 0, fmt, args);
  31   if (len <= 0) {
  32     return -1;
  33   }
  34   char* buf = new char[len + 1];
  35   SCOPE_EXIT { delete[] buf; };
  36   if (vsnprintf(buf, len + 1, fmt, args) == len && write(fd, buf, len) == len) {
  37     return len;
  38   }
  39
  40   return -1;
  41 }
  42
  43 int pclose(FILE* f) { return _pclose(f); }
  44
  45 FILE* popen(const char* name, const char* mode) { return _popen(name, mode); }
  46
  47 void setbuffer(FILE* f, char* buf, size_t size) {
  48   setvbuf(f, buf, _IOFBF, size);
  49 }
  50
  51 int vasprintf(char** dest, const char* format, va_list ap) {
  52   int len = vsnprintf(nullptr, 0, format, ap);
  53   if (len <= 0) {
  54     return -1;
  55   }
  56   char* buf = *dest = (char*)malloc(len + 1);
  57   if (vsnprintf(buf, len + 1, format, ap) == len) {
  58     return len;
  59   }
  60   free(buf);
  61   return -1;
  62 }
  63 }
  64 #endif