Python code coverage for Modules/_scproxy.c

#	count	content
1	n/a	/*
2	n/a	* Helper method for urllib to fetch the proxy configuration settings
3	n/a	* using the SystemConfiguration framework.
4	n/a	*/
5	n/a	#include <Python.h>
6	n/a	#include <SystemConfiguration/SystemConfiguration.h>
7	n/a
8	n/a	static int32_t
9	n/a	cfnum_to_int32(CFNumberRef num)
10	n/a	{
11	n/a	int32_t result;
12	n/a
13	n/a	CFNumberGetValue(num, kCFNumberSInt32Type, &result);
14	n/a	return result;
15	n/a	}
16	n/a
17	n/a	static PyObject*
18	n/a	cfstring_to_pystring(CFStringRef ref)
19	n/a	{
20	n/a	const char* s;
21	n/a
22	n/a	s = CFStringGetCStringPtr(ref, kCFStringEncodingUTF8);
23	n/a	if (s) {
24	n/a	return PyUnicode_DecodeUTF8(
25	n/a	s, strlen(s), NULL);
26	n/a
27	n/a	} else {
28	n/a	CFIndex len = CFStringGetLength(ref);
29	n/a	Boolean ok;
30	n/a	PyObject* result;
31	n/a	char* buf;
32	n/a
33	n/a	buf = PyMem_Malloc(len*4);
34	n/a	if (buf == NULL) {
35	n/a	PyErr_NoMemory();
36	n/a	return NULL;
37	n/a	}
38	n/a
39	n/a	ok = CFStringGetCString(ref,
40	n/a	buf, len * 4,
41	n/a	kCFStringEncodingUTF8);
42	n/a	if (!ok) {
43	n/a	PyMem_Free(buf);
44	n/a	return NULL;
45	n/a	} else {
46	n/a	result = PyUnicode_DecodeUTF8(
47	n/a	buf, strlen(buf), NULL);
48	n/a	PyMem_Free(buf);
49	n/a	}
50	n/a	return result;
51	n/a	}
52	n/a	}
53	n/a
54	n/a
55	n/a	static PyObject*
56	n/a	get_proxy_settings(PyObject* mod __attribute__((__unused__)))
57	n/a	{
58	n/a	CFDictionaryRef proxyDict = NULL;
59	n/a	CFNumberRef aNum = NULL;
60	n/a	CFArrayRef anArray = NULL;
61	n/a	PyObject* result = NULL;
62	n/a	PyObject* v;
63	n/a	int r;
64	n/a
65	n/a	proxyDict = SCDynamicStoreCopyProxies(NULL);
66	n/a	if (!proxyDict) {
67	n/a	Py_RETURN_NONE;
68	n/a	}
69	n/a
70	n/a	result = PyDict_New();
71	n/a	if (result == NULL) goto error;
72	n/a
73	n/a	aNum = CFDictionaryGetValue(proxyDict,
74	n/a	kSCPropNetProxiesExcludeSimpleHostnames);
75	n/a	if (aNum == NULL) {
76	n/a	v = PyBool_FromLong(0);
77	n/a	} else {
78	n/a	v = PyBool_FromLong(cfnum_to_int32(aNum));
79	n/a	}
80	n/a
81	n/a	if (v == NULL) goto error;
82	n/a
83	n/a	r = PyDict_SetItemString(result, "exclude_simple", v);
84	n/a	Py_DECREF(v); v = NULL;
85	n/a	if (r == -1) goto error;
86	n/a
87	n/a	anArray = CFDictionaryGetValue(proxyDict,
88	n/a	kSCPropNetProxiesExceptionsList);
89	n/a	if (anArray != NULL) {
90	n/a	CFIndex len = CFArrayGetCount(anArray);
91	n/a	CFIndex i;
92	n/a	v = PyTuple_New(len);
93	n/a	if (v == NULL) goto error;
94	n/a
95	n/a	r = PyDict_SetItemString(result, "exceptions", v);
96	n/a	Py_DECREF(v);
97	n/a	if (r == -1) goto error;
98	n/a
99	n/a	for (i = 0; i < len; i++) {
100	n/a	CFStringRef aString = NULL;
101	n/a
102	n/a	aString = CFArrayGetValueAtIndex(anArray, i);
103	n/a	if (aString == NULL) {
104	n/a	PyTuple_SetItem(v, i, Py_None);
105	n/a	Py_INCREF(Py_None);
106	n/a	} else {
107	n/a	PyObject* t = cfstring_to_pystring(aString);
108	n/a	if (!t) {
109	n/a	PyTuple_SetItem(v, i, Py_None);
110	n/a	Py_INCREF(Py_None);
111	n/a	} else {
112	n/a	PyTuple_SetItem(v, i, t);
113	n/a	}
114	n/a	}
115	n/a	}
116	n/a	}
117	n/a
118	n/a	CFRelease(proxyDict);
119	n/a	return result;
120	n/a
121	n/a	error:
122	n/a	if (proxyDict) CFRelease(proxyDict);
123	n/a	Py_XDECREF(result);
124	n/a	return NULL;
125	n/a	}
126	n/a
127	n/a	static int
128	n/a	set_proxy(PyObject* proxies, const char* proto, CFDictionaryRef proxyDict,
129	n/a	CFStringRef enabledKey,
130	n/a	CFStringRef hostKey, CFStringRef portKey)
131	n/a	{
132	n/a	CFNumberRef aNum;
133	n/a
134	n/a	aNum = CFDictionaryGetValue(proxyDict, enabledKey);
135	n/a	if (aNum && cfnum_to_int32(aNum)) {
136	n/a	CFStringRef hostString;
137	n/a
138	n/a	hostString = CFDictionaryGetValue(proxyDict, hostKey);
139	n/a	aNum = CFDictionaryGetValue(proxyDict, portKey);
140	n/a
141	n/a	if (hostString) {
142	n/a	int r;
143	n/a	PyObject* h = cfstring_to_pystring(hostString);
144	n/a	PyObject* v;
145	n/a	if (h) {
146	n/a	if (aNum) {
147	n/a	int32_t port = cfnum_to_int32(aNum);
148	n/a	v = PyUnicode_FromFormat("http://%U:%ld",
149	n/a	h, (long)port);
150	n/a	} else {
151	n/a	v = PyUnicode_FromFormat("http://%U", h);
152	n/a	}
153	n/a	Py_DECREF(h);
154	n/a	if (!v) return -1;
155	n/a	r = PyDict_SetItemString(proxies, proto,
156	n/a	v);
157	n/a	Py_DECREF(v);
158	n/a	return r;
159	n/a	}
160	n/a	}
161	n/a
162	n/a	}
163	n/a	return 0;
164	n/a	}
165	n/a
166	n/a
167	n/a
168	n/a	static PyObject*
169	n/a	get_proxies(PyObject* mod __attribute__((__unused__)))
170	n/a	{
171	n/a	PyObject* result = NULL;
172	n/a	int r;
173	n/a	CFDictionaryRef proxyDict = NULL;
174	n/a
175	n/a	proxyDict = SCDynamicStoreCopyProxies(NULL);
176	n/a	if (proxyDict == NULL) {
177	n/a	return PyDict_New();
178	n/a	}
179	n/a
180	n/a	result = PyDict_New();
181	n/a	if (result == NULL) goto error;
182	n/a
183	n/a	r = set_proxy(result, "http", proxyDict,
184	n/a	kSCPropNetProxiesHTTPEnable,
185	n/a	kSCPropNetProxiesHTTPProxy,
186	n/a	kSCPropNetProxiesHTTPPort);
187	n/a	if (r == -1) goto error;
188	n/a	r = set_proxy(result, "https", proxyDict,
189	n/a	kSCPropNetProxiesHTTPSEnable,
190	n/a	kSCPropNetProxiesHTTPSProxy,
191	n/a	kSCPropNetProxiesHTTPSPort);
192	n/a	if (r == -1) goto error;
193	n/a	r = set_proxy(result, "ftp", proxyDict,
194	n/a	kSCPropNetProxiesFTPEnable,
195	n/a	kSCPropNetProxiesFTPProxy,
196	n/a	kSCPropNetProxiesFTPPort);
197	n/a	if (r == -1) goto error;
198	n/a	r = set_proxy(result, "gopher", proxyDict,
199	n/a	kSCPropNetProxiesGopherEnable,
200	n/a	kSCPropNetProxiesGopherProxy,
201	n/a	kSCPropNetProxiesGopherPort);
202	n/a	if (r == -1) goto error;
203	n/a
204	n/a	CFRelease(proxyDict);
205	n/a	return result;
206	n/a	error:
207	n/a	if (proxyDict) CFRelease(proxyDict);
208	n/a	Py_XDECREF(result);
209	n/a	return NULL;
210	n/a	}
211	n/a
212	n/a	static PyMethodDef mod_methods[] = {
213	n/a	{
214	n/a	"_get_proxy_settings",
215	n/a	(PyCFunction)get_proxy_settings,
216	n/a	METH_NOARGS,
217	n/a	NULL,
218	n/a	},
219	n/a	{
220	n/a	"_get_proxies",
221	n/a	(PyCFunction)get_proxies,
222	n/a	METH_NOARGS,
223	n/a	NULL,
224	n/a	},
225	n/a	{ 0, 0, 0, 0 }
226	n/a	};
227	n/a
228	n/a
229	n/a
230	n/a	static struct PyModuleDef mod_module = {
231	n/a	PyModuleDef_HEAD_INIT,
232	n/a	"_scproxy",
233	n/a	NULL,
234	n/a	-1,
235	n/a	mod_methods,
236	n/a	NULL,
237	n/a	NULL,
238	n/a	NULL,
239	n/a	NULL
240	n/a	};
241	n/a
242	n/a
243	n/a	#ifdef __cplusplus
244	n/a	extern "C" {
245	n/a	#endif
246	n/a
247	n/a	PyMODINIT_FUNC
248	n/a	PyInit__scproxy(void)
249	n/a	{
250	n/a	return PyModule_Create(&mod_module);
251	n/a	}
252	n/a
253	n/a	#ifdef __cplusplus
254	n/a	}
255	n/a	#endif
256	n/a