folly/portability/SysResource.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/SysResource.h>
  18
  19 #ifdef _WIN32
  20 #include <folly/portability/Windows.h>
  21
  22 extern "C" {
  23 int getrlimit(int type, rlimit* dst) {
  24   if (type == RLIMIT_STACK) {
  25     NT_TIB* tib = (NT_TIB*)NtCurrentTeb();
  26     dst->rlim_cur = (size_t)tib->StackBase - (size_t)tib->StackLimit;
  27     dst->rlim_max = dst->rlim_cur;
  28     return 0;
  29   }
  30   return -1;
  31 }
  32
  33 int getrusage(int /* who */, rusage* usage) {
  34   // You get NOTHING! Good day to you sir.
  35   ZeroMemory(usage, sizeof(rusage));
  36   return 0;
  37 }
  38
  39 int setrlimit(int /* type */, rlimit* /* src */) {
  40   // Do nothing for setting them for now.
  41   // We couldn't set the stack size at runtime even if we wanted to.
  42   return 0;
  43 }
  44 }
  45 #endif