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_cgetws_s (wchar_t *, size_t, size_t *);
8 static errno_t __cdecl _stub (wchar_t *, size_t, size_t *);
9
10 errno_t __cdecl (*__MINGW_IMP_SYMBOL(_cgetws_s))(wchar_t *, size_t, size_t *) =
11 _stub;
12
13 static errno_t __cdecl
_stub(wchar_t * s,size_t l,size_t * r_len)14 _stub (wchar_t *s, size_t l, size_t *r_len)
15 {
16 errno_t __cdecl (*f)(wchar_t *, size_t, size_t *) = __MINGW_IMP_SYMBOL(_cgetws_s);
17
18 if (f == _stub)
19 {
20 f = (errno_t __cdecl (*)(wchar_t *, size_t, size_t *))
21 GetProcAddress (__mingw_get_msvcrt_handle (), "_cgetws_s");
22 if (!f)
23 f = _int_cgetws_s;
24 __MINGW_IMP_SYMBOL(_cgetws_s) = f;
25 }
26 return (*f)(s, l, r_len);
27 }
28
29 errno_t __cdecl
_cgetws_s(wchar_t * s,size_t l,size_t * r_len)30 _cgetws_s (wchar_t *s, size_t l, size_t *r_len)
31 {
32 return _stub (s, l, r_len);
33 }
34
35 static errno_t __cdecl
_int_cgetws_s(wchar_t * s,size_t l,size_t * r_len)36 _int_cgetws_s (wchar_t *s, size_t l, size_t *r_len)
37 {
38 wchar_t *h, *p;
39
40 if (s && l)
41 s[0] = 0;
42 if (!s || !l || !r_len)
43 {
44 _cgetws (NULL);
45 return EINVAL;
46 }
47 p = (wchar_t *) alloca ((l + 2) * sizeof (wchar_t));
48 p[0] = l;
49 h = _cgetws (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