From: Christopher Dykes Date: Tue, 15 Mar 2016 20:23:20 +0000 (-0700) Subject: Add portability header for sys/file.h X-Git-Tag: 2016.07.26~438 X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=30c8cd934d3a579aa0768486f9e950ac8c41de84;p=folly.git Add portability header for sys/file.h Summary: We don't have it on Windows :( Reviewed By: yfeldblum Differential Revision: D2978397 fb-gh-sync-id: 388bc450ce269559825ebfb78e3646533c2e1153 shipit-source-id: 388bc450ce269559825ebfb78e3646533c2e1153 --- diff --git a/folly/Makefile.am b/folly/Makefile.am index 904d70a3..746350bd 100644 --- a/folly/Makefile.am +++ b/folly/Makefile.am @@ -274,6 +274,7 @@ nobase_follyinclude_HEADERS = \ portability/Memory.h \ portability/Strings.h \ portability/Syscall.h \ + portability/SysFile.h \ portability/SysStat.h \ portability/SysTime.h \ portability/SysTypes.h \ @@ -407,6 +408,7 @@ libfolly_la_SOURCES = \ MemoryMapping.cpp \ portability/Environment.cpp \ portability/Strings.cpp \ + portability/SysFile.cpp \ portability/SysStat.cpp \ portability/SysTime.cpp \ portability/Time.cpp \ diff --git a/folly/portability/SysFile.cpp b/folly/portability/SysFile.cpp new file mode 100755 index 00000000..71287e75 --- /dev/null +++ b/folly/portability/SysFile.cpp @@ -0,0 +1,47 @@ +/* + * Copyright 2016 Facebook, Inc. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include + +#ifdef _WIN32 +#include + +#include + +extern "C" int flock(int fd, int operation) { + HANDLE h = (HANDLE)_get_osfhandle(fd); + if (h == INVALID_HANDLE_VALUE) { + return -1; + } + + constexpr DWORD kMaxDWORD = std::numeric_limits::max(); + if (operation & LOCK_UN) { + if (!UnlockFile(h, 0, 0, kMaxDWORD, kMaxDWORD)) { + return -1; + } + } else { + int flags = 0 + | (operation & LOCK_NB ? LOCKFILE_FAIL_IMMEDIATELY : 0) + | (operation & LOCK_EX ? LOCKFILE_EXCLUSIVE_LOCK : 0) + ; + OVERLAPPED ov = {}; + if (!LockFileEx(h, flags, 0, kMaxDWORD, kMaxDWORD, &ov)) { + return -1; + } + } + return 0; +} +#endif diff --git a/folly/portability/SysFile.h b/folly/portability/SysFile.h new file mode 100755 index 00000000..096bfcc5 --- /dev/null +++ b/folly/portability/SysFile.h @@ -0,0 +1,27 @@ +/* + * Copyright 2016 Facebook, Inc. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#pragma once + +#ifndef _WIN32 +#include +#else +#define LOCK_SH 1 +#define LOCK_EX 2 +#define LOCK_NB 4 +#define LOCK_UN 8 +extern "C" int flock(int fd, int operation); +#endif