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 ret = vsnprintf(nullptr, 0, fmt, args);
  31   if (ret <= 0) {
  32     return -1;
  33   }
  34   size_t len = size_t(ret);
  35   char* buf = new char[len + 1];
  36   SCOPE_EXIT { delete[] buf; };
  37   if (size_t(vsnprintf(buf, len + 1, fmt, args)) == len &&
  38       write(fd, buf, len) == ssize_t(len)) {
  39     return ret;
  40   }
  41
  42   return -1;
  43 }
  44
  45 int pclose(FILE* f) { return _pclose(f); }
  46
  47 FILE* popen(const char* name, const char* mode) { return _popen(name, mode); }
  48
  49 void setbuffer(FILE* f, char* buf, size_t size) {
  50   setvbuf(f, buf, _IOFBF, size);
  51 }
  52
  53 int vasprintf(char** dest, const char* format, va_list ap) {
  54   int len = vsnprintf(nullptr, 0, format, ap);
  55   if (len <= 0) {
  56     return -1;
  57   }
  58   char* buf = *dest = (char*)malloc(size_t(len + 1));
  59   if (vsnprintf(buf, size_t(len + 1), format, ap) == len) {
  60     return len;
  61   }
  62   free(buf);
  63   return -1;
  64 }
  65 }
  66 #endif