Kai Tietz | 92432c5 | 2012-07-31 20:32:13 +0000 | [diff] [blame] | 1 | #include <windows.h> |
| 2 | #include <malloc.h> |
| 3 | #include <errno.h> |
Jacek Caban | a5c387c | 2013-06-17 13:12:21 +0000 | [diff] [blame] | 4 | #include <msvcrt.h> |
David Wohlferd | 5af3b0e | 2016-08-26 17:15:40 -0700 | [diff] [blame] | 5 | #include <sec_api/conio_s.h> |
Kai Tietz | 92432c5 | 2012-07-31 20:32:13 +0000 | [diff] [blame] | 6 | |
Kai Tietz | 92432c5 | 2012-07-31 20:32:13 +0000 | [diff] [blame] | 7 | static errno_t __cdecl _int_cgets_s (char *, size_t, size_t *); |
| 8 | static errno_t __cdecl _stub (char *, size_t, size_t *); |
| 9 | |
| 10 | errno_t __cdecl (*__MINGW_IMP_SYMBOL(_cgets_s))(char *, size_t, size_t *) = |
| 11 | _stub; |
| 12 | |
| 13 | static errno_t __cdecl |
| 14 | _stub (char *s, size_t l, size_t *r_len) |
| 15 | { |
| 16 | errno_t __cdecl (*f)(char *, size_t, size_t *) = __MINGW_IMP_SYMBOL(_cgets_s); |
| 17 | |
| 18 | if (f == _stub) |
| 19 | { |
Martin Storsjö | f4f972f | 2024-09-18 15:05:06 +0300 | [diff] [blame] | 20 | f = (void*) |
Kai Tietz | 92432c5 | 2012-07-31 20:32:13 +0000 | [diff] [blame] | 21 | GetProcAddress (__mingw_get_msvcrt_handle (), "_cgets_s"); |
| 22 | if (!f) |
| 23 | f = _int_cgets_s; |
| 24 | __MINGW_IMP_SYMBOL(_cgets_s) = f; |
| 25 | } |
| 26 | return (*f)(s, l, r_len); |
| 27 | } |
| 28 | |
Kai Tietz | 62d02ec | 2012-08-01 13:53:15 +0000 | [diff] [blame] | 29 | errno_t __cdecl |
| 30 | _cgets_s (char *s, size_t l, size_t *r_len) |
| 31 | { |
| 32 | return _stub (s, l, r_len); |
| 33 | } |
| 34 | |
Kai Tietz | 92432c5 | 2012-07-31 20:32:13 +0000 | [diff] [blame] | 35 | static errno_t __cdecl |
| 36 | _int_cgets_s (char *s, size_t l, size_t *r_len) |
| 37 | { |
| 38 | char *h, *p; |
| 39 | |
| 40 | if (s && l) |
| 41 | s[0] = 0; |
| 42 | if (!s || !l || !r_len) |
| 43 | { |
| 44 | _cgets (NULL); |
| 45 | return EINVAL; |
| 46 | } |
| 47 | p = (char *) alloca (l + 2); |
| 48 | p[0] = l; |
| 49 | h = _cgets (s); |
| 50 | if (!h) |
| 51 | return EINVAL; |
| 52 | *r_len = (size_t) p[1]; |
| 53 | memcpy (s, &p[2], *r_len); |
| 54 | return 0; |
| 55 | } |