8efb001bc1185ed1332c46f8fa88db40048ce353
[folly.git] / folly / portability / SysStat.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/SysStat.h>
18
19 #ifdef _WIN32
20 #include <folly/portability/Windows.h>
21
22 extern "C" {
23 int chmod(char const* fn, int am) { return _chmod(fn, am); }
24
25 int fchmod(int fd, mode_t mode) {
26   HANDLE h = (HANDLE)_get_osfhandle(fd);
27   if (h == INVALID_HANDLE_VALUE) {
28     return -1;
29   }
30
31   FILE_ATTRIBUTE_TAG_INFO attr{};
32   if (!GetFileInformationByHandleEx(
33           h, FileAttributeTagInfo, &attr, sizeof(attr))) {
34     return -1;
35   }
36
37   if (mode & _S_IWRITE) {
38     attr.FileAttributes &= ~FILE_ATTRIBUTE_READONLY;
39   } else {
40     attr.FileAttributes |= FILE_ATTRIBUTE_READONLY;
41   }
42
43   if (!SetFileInformationByHandle(
44           h, FileAttributeTagInfo, &attr, sizeof(attr))) {
45     return -1;
46   }
47
48   return 0;
49 }
50
51 // Just return the result of a normal stat for now
52 int lstat(const char* path, struct stat* st) { return stat(path, st); }
53
54 int mkdir(const char* fn, int /* mode */) {
55   return _mkdir(fn);
56 }
57
58 int umask(int md) { return _umask(md); }
59 }
60 #endif