Refactors folly sync test cases
[folly.git] / folly / portability / Stdio.cpp
1 /*
2  * Copyright 2016-present 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
21 #include <cstdlib>
22
23 #include <folly/ScopeGuard.h>
24 #include <folly/portability/Unistd.h>
25
26 extern "C" {
27 int dprintf(int fd, const char* fmt, ...) {
28   va_list args;
29   va_start(args, fmt);
30   SCOPE_EXIT {
31     va_end(args);
32   };
33
34   int ret = vsnprintf(nullptr, 0, fmt, args);
35   if (ret <= 0) {
36     return -1;
37   }
38   size_t len = size_t(ret);
39   char* buf = new char[len + 1];
40   SCOPE_EXIT {
41     delete[] buf;
42   };
43   if (size_t(vsnprintf(buf, len + 1, fmt, args)) == len &&
44       write(fd, buf, len) == ssize_t(len)) {
45     return ret;
46   }
47
48   return -1;
49 }
50
51 int pclose(FILE* f) {
52   return _pclose(f);
53 }
54
55 FILE* popen(const char* name, const char* mode) {
56   return _popen(name, mode);
57 }
58
59 void setbuffer(FILE* f, char* buf, size_t size) {
60   setvbuf(f, buf, _IOFBF, size);
61 }
62
63 int vasprintf(char** dest, const char* format, va_list ap) {
64   int len = vsnprintf(nullptr, 0, format, ap);
65   if (len <= 0) {
66     return -1;
67   }
68   char* buf = *dest = (char*)malloc(size_t(len + 1));
69   if (vsnprintf(buf, size_t(len + 1), format, ap) == len) {
70     return len;
71   }
72   free(buf);
73   return -1;
74 }
75 }
76
77 #endif