1 #include <windows.h>
2 #include <malloc.h>
3 #include <errno.h>
4 #include <msvcrt.h>
5 #include <sec_api/conio_s.h>
6
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
_stub(char * s,size_t l,size_t * r_len)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 {
20 f = (errno_t __cdecl (*)(char *, size_t, size_t *))
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
29 errno_t __cdecl
_cgets_s(char * s,size_t l,size_t * r_len)30 _cgets_s (char *s, size_t l, size_t *r_len)
31 {
32 return _stub (s, l, r_len);
33 }
34
35 static errno_t __cdecl
_int_cgets_s(char * s,size_t l,size_t * r_len)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 }
56