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	aNum = CFDictionaryGetValue(proxyDict,
68	n/a	kSCPropNetProxiesExcludeSimpleHostnames);
69	n/a	if (aNum == NULL) {
70	n/a	v = PyBool_FromLong(0);
71	n/a	} else {
72	n/a	v = PyBool_FromLong(cfnum_to_int32(aNum));
73	n/a	}
74	n/a	if (v == NULL) goto error;
75	n/a
76	n/a	r = PyDict_SetItemString(result, "exclude_simple", v);
77	n/a	Py_DECREF(v); v = NULL;
78	n/a	if (r == -1) goto error;
79	n/a
80	n/a	anArray = CFDictionaryGetValue(proxyDict,
81	n/a	kSCPropNetProxiesExceptionsList);
82	n/a	if (anArray != NULL) {
83	n/a	CFIndex len = CFArrayGetCount(anArray);
84	n/a	CFIndex i;
85	n/a	v = PyTuple_New(len);
86	n/a	if (v == NULL) goto error;
87	n/a
88	n/a	r = PyDict_SetItemString(result, "exceptions", v);
89	n/a	Py_DECREF(v);
90	n/a	if (r == -1) goto error;
91	n/a
92	n/a	for (i = 0; i < len; i++) {
93	n/a	CFStringRef aString = NULL;
94	n/a
95	n/a	aString = CFArrayGetValueAtIndex(anArray, i);
96	n/a	if (aString == NULL) {
97	n/a	PyTuple_SetItem(v, i, Py_None);
98	n/a	Py_INCREF(Py_None);
99	n/a	} else {
100	n/a	PyObject* t = cfstring_to_pystring(aString);
101	n/a	if (!t) {
102	n/a	PyTuple_SetItem(v, i, Py_None);
103	n/a	Py_INCREF(Py_None);
104	n/a	} else {
105	n/a	PyTuple_SetItem(v, i, t);
106	n/a	}
107	n/a	}
108	n/a	}
109	n/a	}
110	n/a
111	n/a	CFRelease(proxyDict);
112	n/a	return result;
113	n/a
114	n/a	error:
115	n/a	if (proxyDict) CFRelease(proxyDict);
116	n/a	Py_XDECREF(result);
117	n/a	return NULL;
118	n/a	}
119	n/a
120	n/a	static int
121	n/a	set_proxy(PyObject* proxies, char* proto, CFDictionaryRef proxyDict,
122	n/a	CFStringRef enabledKey,
123	n/a	CFStringRef hostKey, CFStringRef portKey)
124	n/a	{
125	n/a	CFNumberRef aNum;
126	n/a
127	n/a	aNum = CFDictionaryGetValue(proxyDict, enabledKey);
128	n/a	if (aNum && cfnum_to_int32(aNum)) {
129	n/a	CFStringRef hostString;
130	n/a
131	n/a	hostString = CFDictionaryGetValue(proxyDict, hostKey);
132	n/a	aNum = CFDictionaryGetValue(proxyDict, portKey);
133	n/a
134	n/a	if (hostString) {
135	n/a	int r;
136	n/a	PyObject* h = cfstring_to_pystring(hostString);
137	n/a	PyObject* v;
138	n/a	if (h) {
139	n/a	if (aNum) {
140	n/a	int32_t port = cfnum_to_int32(aNum);
141	n/a	v = PyString_FromFormat("http://%s:%ld",
142	n/a	PyString_AS_STRING(h),
143	n/a	(long)port);
144	n/a	} else {
145	n/a	v = PyString_FromFormat("http://%s",
146	n/a	PyString_AS_STRING(h));
147	n/a	}
148	n/a	Py_DECREF(h);
149	n/a	if (!v) return -1;
150	n/a	r = PyDict_SetItemString(proxies, proto,
151	n/a	v);
152	n/a	Py_DECREF(v);
153	n/a	return r;
154	n/a	}
155	n/a	}
156	n/a
157	n/a	}
158	n/a	return 0;
159	n/a	}
160	n/a
161	n/a
162	n/a
163	n/a	static PyObject*
164	n/a	get_proxies(PyObject* mod __attribute__((__unused__)))
165	n/a	{
166	n/a	PyObject* result = NULL;
167	n/a	int r;
168	n/a	CFDictionaryRef proxyDict = NULL;
169	n/a
170	n/a	proxyDict = SCDynamicStoreCopyProxies(NULL);
171	n/a	if (proxyDict == NULL) {
172	n/a	return PyDict_New();
173	n/a	}
174	n/a
175	n/a	result = PyDict_New();
176	n/a	if (result == NULL) goto error;
177	n/a
178	n/a	r = set_proxy(result, "http", proxyDict,
179	n/a	kSCPropNetProxiesHTTPEnable,
180	n/a	kSCPropNetProxiesHTTPProxy,
181	n/a	kSCPropNetProxiesHTTPPort);
182	n/a	if (r == -1) goto error;
183	n/a	r = set_proxy(result, "https", proxyDict,
184	n/a	kSCPropNetProxiesHTTPSEnable,
185	n/a	kSCPropNetProxiesHTTPSProxy,
186	n/a	kSCPropNetProxiesHTTPSPort);
187	n/a	if (r == -1) goto error;
188	n/a	r = set_proxy(result, "ftp", proxyDict,
189	n/a	kSCPropNetProxiesFTPEnable,
190	n/a	kSCPropNetProxiesFTPProxy,
191	n/a	kSCPropNetProxiesFTPPort);
192	n/a	if (r == -1) goto error;
193	n/a	r = set_proxy(result, "gopher", proxyDict,
194	n/a	kSCPropNetProxiesGopherEnable,
195	n/a	kSCPropNetProxiesGopherProxy,
196	n/a	kSCPropNetProxiesGopherPort);
197	n/a	if (r == -1) goto error;
198	n/a
199	n/a	CFRelease(proxyDict);
200	n/a	return result;
201	n/a	error:
202	n/a	if (proxyDict) CFRelease(proxyDict);
203	n/a	Py_XDECREF(result);
204	n/a	return NULL;
205	n/a	}
206	n/a
207	n/a	static PyMethodDef mod_methods[] = {
208	n/a	{
209	n/a	"_get_proxy_settings",
210	n/a	(PyCFunction)get_proxy_settings,
211	n/a	METH_NOARGS,
212	n/a	NULL,
213	n/a	},
214	n/a	{
215	n/a	"_get_proxies",
216	n/a	(PyCFunction)get_proxies,
217	n/a	METH_NOARGS,
218	n/a	NULL,
219	n/a	},
220	n/a	{ 0, 0, 0, 0 }
221	n/a	};
222	n/a
223	n/a	void init_scproxy(void)
224	n/a	{
225	n/a	(void)Py_InitModule4("_scproxy", mod_methods, NULL, NULL, PYTHON_API_VERSION);
226	n/a	}