FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
pymanager/src/_native/helpers.cpp at main · python/pymanager · GitHub
python
pymanager
Repository navigation
Code
Issues
24
(24)
Pull requests
6
(6)
Actions
Security and quality
1
(1)
Insights
Expand file tree
Breadcrumbs
pymanager
/
src
/
_native
/
helpers.cpp
Copy path
More file actions
More file actions
Latest commit
History
History
History
141 lines (129 loc) · 3.96 KB
Breadcrumbs
pymanager
/
src
/
_native
/
helpers.cpp
Copy path
File metadata and controls
141 lines (129 loc) · 3.96 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#
include
<
Python.h
>
#
include
<
windows.h
>
#
include
<
winhttp.h
>
#
include
"
helpers.h
"
int
as_utf16
(PyObject *obj,
wchar_t
**address) {
if
(obj ==
NULL
) {
//
Automatic cleanup
PyMem_Free
(*address);
return
1
;
}
if
(!
PyObject_IsTrue
(obj)) {
if
(
Py_Is
(obj,
Py_GetConstantBorrowed
(Py_CONSTANT_NONE))) {
*address =
NULL
;
return
1
;
}
}
PyObject *wobj =
PyObject_Str
(obj);
if
(!wobj) {
return
0
;
}
PyObject *b =
PyObject_CallMethod
(wobj,
"
encode
"
,
"
ss
"
,
"
utf-16-le
"
,
"
strict
"
);
Py_DECREF
(wobj);
if
(!b) {
return
0
;
}
char
*src;
Py_ssize_t len;
if
(
PyBytes_AsStringAndSize
(b, &src, &len) <
0
) {
Py_DECREF
(b);
return
0
;
}
Py_ssize_t wlen = len /
sizeof
(
wchar_t
);
wchar_t
*result = (
wchar_t
*)
PyMem_Malloc
((wlen +
1
) *
sizeof
(
wchar_t
));
if
(!result) {
Py_DECREF
(b);
PyErr_NoMemory
();
return
0
;
}
wcsncpy_s
(result, wlen +
1
, (
wchar_t
*)src, wlen);
Py_DECREF
(b);
*address = result;
return
Py_CLEANUP_SUPPORTED;
}
void
err_SetFromWindowsErrWithMessage
(
int
error,
const
char
*message,
const
wchar_t
*os_message,
void
*hModule) {
LPWSTR
os_message_buffer =
NULL
;
PyObject *cause =
NULL
;
if
(
PyErr_Occurred
()) {
cause =
PyErr_GetRaisedException
();
}
switch
(
HRESULT_FACILITY
(error)) {
case
FACILITY_WIN32
:
//
Error code is an HRESULT containing a regular Windows error
error &=
0xFFFF
;
break
;
case
FACILITY_CERT
:
case
FACILITY_SECURITY
:
//
crypt32-related errors
if
(!hModule) {
hModule =
GetModuleHandleW
(
L"
crypt32
"
);
}
break
;
}
if
(!hModule && error >=
WINHTTP_ERROR_BASE
&& error <=
WINHTTP_ERROR_LAST
) {
//
Error codes are from WinHTTP, which means we need a module
hModule =
GetModuleHandleW
(
L"
winhttp
"
);
}
if
(!os_message) {
DWORD
len =
FormatMessageW
(
/*
Error API error
*/
FORMAT_MESSAGE_ALLOCATE_BUFFER
|
FORMAT_MESSAGE_FROM_SYSTEM
| (hModule ?
FORMAT_MESSAGE_FROM_HMODULE
:
0
)
|
FORMAT_MESSAGE_IGNORE_INSERTS
,
hModule,
error,
MAKELANGID
(
LANG_NEUTRAL
,
SUBLANG_DEFAULT
),
(
LPWSTR
)&os_message_buffer,
0
,
NULL
);
if
(len) {
while
(len >
0
&&
isspace
(os_message_buffer[--len])) {
os_message_buffer[len] = L
'
\0
'
;
}
os_message = os_message_buffer;
}
}
PyObject *msg;
if
(message && os_message) {
msg =
PyUnicode_FromFormat
(
"
%s: %ls
"
, message, os_message);
}
else
if
(os_message) {
msg =
PyUnicode_FromWideChar
(os_message, -
1
);
}
else
if
(message) {
msg =
PyUnicode_FromString
(message);
}
else
{
msg =
PyUnicode_FromString
(
"
Unknown error
"
);
}
if
(msg) {
//
Hacky way to get OSError without a direct data reference
//
This allows us to delay load the Python DLL
PyObject *builtins =
PyEval_GetFrameBuiltins
();
PyObject *oserr = builtins ?
PyDict_GetItemString
(builtins,
"
OSError
"
) :
NULL
;
if
(oserr) {
PyObject *exc_args =
Py_BuildValue
(
"
(iOOiO)
"
,
(
int
)
0
,
msg,
Py_GetConstantBorrowed
(Py_CONSTANT_NONE),
error,
Py_GetConstantBorrowed
(Py_CONSTANT_NONE)
);
if
(exc_args) {
PyErr_SetObject
(oserr, exc_args);
Py_DECREF
(exc_args);
}
}
Py_XDECREF
(builtins);
Py_DECREF
(msg);
}
if
(os_message_buffer) {
LocalFree
((
void
*)os_message_buffer);
}
if
(cause) {
//
References are all stolen here, so no DECREF required
PyObject *chained =
PyErr_GetRaisedException
();
PyException_SetContext
(chained, cause);
PyErr_SetRaisedException
(chained);
}
}
Back
|
FazBrowse Home
|
New Git URL