fdlimit_bsd.go 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. // Copyright 2016 The go-ethereum Authors
  2. // This file is part of the go-ethereum library.
  3. //
  4. // The go-ethereum library is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Lesser General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // The go-ethereum library is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Lesser General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Lesser General Public License
  15. // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16. // +build freebsd dragonfly
  17. package fdlimit
  18. import "syscall"
  19. // This file is largely identical to fdlimit_unix.go,
  20. // but Rlimit fields have type int64 on *BSD so it needs
  21. // an extra conversion.
  22. // Raise tries to maximize the file descriptor allowance of this process
  23. // to the maximum hard-limit allowed by the OS.
  24. func Raise(max uint64) (uint64, error) {
  25. // Get the current limit
  26. var limit syscall.Rlimit
  27. if err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &limit); err != nil {
  28. return 0, err
  29. }
  30. // Try to update the limit to the max allowance
  31. limit.Cur = limit.Max
  32. if limit.Cur > int64(max) {
  33. limit.Cur = int64(max)
  34. }
  35. if err := syscall.Setrlimit(syscall.RLIMIT_NOFILE, &limit); err != nil {
  36. return 0, err
  37. }
  38. if err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &limit); err != nil {
  39. return 0, err
  40. }
  41. return uint64(limit.Cur), nil
  42. }
  43. // Current retrieves the number of file descriptors allowed to be opened by this
  44. // process.
  45. func Current() (int, error) {
  46. var limit syscall.Rlimit
  47. if err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &limit); err != nil {
  48. return 0, err
  49. }
  50. return int(limit.Cur), nil
  51. }
  52. // Maximum retrieves the maximum number of file descriptors this process is
  53. // allowed to request for itself.
  54. func Maximum() (int, error) {
  55. var limit syscall.Rlimit
  56. if err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &limit); err != nil {
  57. return 0, err
  58. }
  59. return int(limit.Max), nil
  60. }