Python code coverage for Mac/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 PyString_FromString(s);
25	n/a
26	n/a	} else {
27	n/a	CFIndex len = CFStringGetLength(ref);
28	n/a	Boolean ok;
29	n/a	PyObject* result;
30	n/a	result = PyString_FromStringAndSize(NULL, len*4);
31	n/a
32	n/a	ok = CFStringGetCString(ref,
33	n/a	PyString_AS_STRING(result),
34	n/a	PyString_GET_SIZE(result),
35	n/a	kCFStringEncodingUTF8);
36	n/a	if (!ok) {
37	n/a	Py_DECREF(result);
38	n/a	return NULL;
39	n/a	} else {
40	n/a	_PyString_Resize(&result,
41	n/a	strlen(PyString_AS_STRING(result)));
42	n/a	}
43	n/a	return result;
44	n/a	}
45	n/a	}
46	n/a
47	n/a
48	n/a	static PyObject*
49	n/a	get_proxy_settings(PyObject* mod __attribute__((__unused__)))
50	n/a	{
51	n/a	CFDictionaryRef proxyDict = NULL;
52	n/a	CFNumberRef aNum = NULL;
53	n/a	CFArrayRef anArray = NULL;
54	n/a	PyObject* result = NULL;
55	n/a	PyObject* v;
56	n/a	int r;
57	n/a
58	n/a	proxyDict = SCDynamicStoreCopyProxies(NULL);
59	n/a	if (!proxyDict) {
60	n/a	Py_INCREF(Py_None);
61	n/a	return Py_None;
62	n/a	}
63	n/a
64	n/a	result = PyDict_New();
65	n/a	if (result == NULL) goto error;
66	n/a
67	n/a	if (&kSCPropNetProxiesExcludeSimpleHostnames != NULL) {
68	n/a	aNum = CFDictionaryGetValue(proxyDict,
69	n/a	kSCPropNetProxiesExcludeSimpleHostnames);
70	n/a	if (aNum == NULL) {
71	n/a	v = PyBool_FromLong(0);
72	n/a	} else {
73	n/a	v = PyBool_FromLong(cfnum_to_int32(aNum));
74	n/a	}
75	n/a	} else {
76	n/a	v = PyBool_FromLong(1);
77	n/a	}
78	n/a
79	n/a	if (v == NULL) goto error;
80	n/a
81	n/a	r = PyDict_SetItemString(result, "exclude_simple", v);
82	n/a	Py_DECREF(v); v = NULL;
83	n/a	if (r == -1) goto error;
84	n/a
85	n/a	anArray = CFDictionaryGetValue(proxyDict,
86	n/a	kSCPropNetProxiesExceptionsList);
87	n/a	if (anArray != NULL) {
88	n/a	CFIndex len = CFArrayGetCount(anArray);
89	n/a	CFIndex i;
90	n/a	v = PyTuple_New(len);
91	n/a	if (v == NULL) goto error;
92	n/a
93	n/a	r = PyDict_SetItemString(result, "exceptions", v);
94	n/a	Py_DECREF(v);
95	n/a	if (r == -1) goto error;
96	n/a
97	n/a	for (i = 0; i < len; i++) {
98	n/a	CFStringRef aString = NULL;
99	n/a
100	n/a	aString = CFArrayGetValueAtIndex(anArray, i);
101	n/a	if (aString == NULL) {
102	n/a	PyTuple_SetItem(v, i, Py_None);
103	n/a	Py_INCREF(Py_None);
104	n/a	} else {
105	n/a	PyObject* t = cfstring_to_pystring(aString);
106	n/a	if (!t) {
107	n/a	PyTuple_SetItem(v, i, Py_None);
108	n/a	Py_INCREF(Py_None);
109	n/a	} else {
110	n/a	PyTuple_SetItem(v, i, t);
111	n/a	}
112	n/a	}
113	n/a	}
114	n/a	}
115	n/a
116	n/a	CFRelease(proxyDict);
117	n/a	return result;
118	n/a
119	n/a	error:
120	n/a	if (proxyDict) CFRelease(proxyDict);
121	n/a	Py_XDECREF(result);
122	n/a	return NULL;
123	n/a	}
124	n/a
125	n/a	static int
126	n/a	set_proxy(PyObject* proxies, char* proto, CFDictionaryRef proxyDict,
127	n/a	CFStringRef enabledKey,
128	n/a	CFStringRef hostKey, CFStringRef portKey)
129	n/a	{
130	n/a	CFNumberRef aNum;
131	n/a
132	n/a	aNum = CFDictionaryGetValue(proxyDict, enabledKey);
133	n/a	if (aNum && cfnum_to_int32(aNum)) {
134	n/a	CFStringRef hostString;
135	n/a
136	n/a	hostString = CFDictionaryGetValue(proxyDict, hostKey);
137	n/a	aNum = CFDictionaryGetValue(proxyDict, portKey);
138	n/a
139	n/a	if (hostString) {
140	n/a	int r;
141	n/a	PyObject* h = cfstring_to_pystring(hostString);
142	n/a	PyObject* v;
143	n/a	if (h) {
144	n/a	if (aNum) {
145	n/a	int32_t port = cfnum_to_int32(aNum);
146	n/a	v = PyString_FromFormat("http://%s:%ld",
147	n/a	PyString_AS_STRING(h),
148	n/a	(long)port);
149	n/a	} else {
150	n/a	v = PyString_FromFormat("http://%s",
151	n/a	PyString_AS_STRING(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	void init_scproxy(void)
229	n/a	{
230	n/a	(void)Py_InitModule4("_scproxy", mod_methods, NULL, NULL, PYTHON_API_VERSION);
231	n/a	}