Russ Cox | 50199d7 | 2014-08-30 14:53:47 -0400 | [diff] [blame] | 1 | // Copyright 2012 The Go Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
| 5 | package runtime |
| 6 | |
| 7 | import "unsafe" |
| 8 | |
| 9 | func getenv(s *byte) *byte { |
| 10 | val := gogetenv(gostringnocopy(s)) |
| 11 | if val == "" { |
| 12 | return nil |
| 13 | } |
| 14 | // Strings found in environment are NUL-terminated. |
| 15 | return &bytes(val)[0] |
| 16 | } |
| 17 | |
| 18 | var tracebackbuf [128]byte |
| 19 | |
| 20 | func gogetenv(key string) string { |
| 21 | var file [128]byte |
| 22 | if len(key) > len(file)-6 { |
| 23 | return "" |
| 24 | } |
| 25 | |
| 26 | copy(file[:], "/env/") |
| 27 | copy(file[5:], key) |
| 28 | |
| 29 | fd := open(&file[0], _OREAD, 0) |
| 30 | if fd < 0 { |
| 31 | return "" |
| 32 | } |
David du Colombier | 9d06cfc | 2014-10-20 23:03:03 +0200 | [diff] [blame] | 33 | n := seek(fd, 0, 2) |
David du Colombier | b3f224b | 2014-09-03 00:56:31 +0200 | [diff] [blame] | 34 | if n <= 0 { |
| 35 | close(fd) |
| 36 | return "" |
| 37 | } |
Russ Cox | 50199d7 | 2014-08-30 14:53:47 -0400 | [diff] [blame] | 38 | |
David du Colombier | 20d9cc4 | 2014-09-01 23:03:26 -0400 | [diff] [blame] | 39 | p := make([]byte, n) |
Russ Cox | 50199d7 | 2014-08-30 14:53:47 -0400 | [diff] [blame] | 40 | |
David du Colombier | 20d9cc4 | 2014-09-01 23:03:26 -0400 | [diff] [blame] | 41 | r := pread(fd, unsafe.Pointer(&p[0]), int32(n), 0) |
Russ Cox | 50199d7 | 2014-08-30 14:53:47 -0400 | [diff] [blame] | 42 | close(fd) |
| 43 | if r < 0 { |
| 44 | return "" |
| 45 | } |
| 46 | |
David du Colombier | 9d06cfc | 2014-10-20 23:03:03 +0200 | [diff] [blame] | 47 | if p[r-1] == 0 { |
| 48 | r-- |
| 49 | } |
| 50 | |
Russ Cox | 50199d7 | 2014-08-30 14:53:47 -0400 | [diff] [blame] | 51 | var s string |
| 52 | sp := (*_string)(unsafe.Pointer(&s)) |
David du Colombier | 20d9cc4 | 2014-09-01 23:03:26 -0400 | [diff] [blame] | 53 | sp.str = &p[0] |
Russ Cox | 50199d7 | 2014-08-30 14:53:47 -0400 | [diff] [blame] | 54 | sp.len = int(r) |
| 55 | return s |
| 56 | } |
David du Colombier | e9c57d8 | 2014-11-21 19:39:01 +0100 | [diff] [blame] | 57 | |
| 58 | var _cgo_setenv unsafe.Pointer // pointer to C function |
| 59 | var _cgo_unsetenv unsafe.Pointer // pointer to C function |