Show More
@@ -1,323 +1,323 b'' | |||||
1 | /* |
|
1 | /* | |
2 | osutil.c - native operating system services |
|
2 | osutil.c - native operating system services | |
3 |
|
3 | |||
4 | Copyright 2007 Matt Mackall and others |
|
4 | Copyright 2007 Matt Mackall and others | |
5 |
|
5 | |||
6 | This software may be used and distributed according to the terms of |
|
6 | This software may be used and distributed according to the terms of | |
7 | the GNU General Public License, incorporated herein by reference. |
|
7 | the GNU General Public License, incorporated herein by reference. | |
8 | */ |
|
8 | */ | |
9 |
|
9 | |||
10 | #define _ATFILE_SOURCE |
|
10 | #define _ATFILE_SOURCE | |
11 | #include <Python.h> |
|
11 | #include <Python.h> | |
12 | #include <alloca.h> |
|
12 | #include <alloca.h> | |
13 | #include <dirent.h> |
|
13 | #include <dirent.h> | |
14 | #include <fcntl.h> |
|
14 | #include <fcntl.h> | |
15 | #include <string.h> |
|
15 | #include <string.h> | |
16 | #include <sys/stat.h> |
|
16 | #include <sys/stat.h> | |
17 | #include <sys/types.h> |
|
17 | #include <sys/types.h> | |
18 | #include <unistd.h> |
|
18 | #include <unistd.h> | |
19 |
|
19 | |||
20 | struct listdir_stat { |
|
20 | struct listdir_stat { | |
21 |
|
|
21 | PyObject_HEAD | |
22 |
|
|
22 | struct stat st; | |
23 | }; |
|
23 | }; | |
24 |
|
24 | |||
25 | #define listdir_slot(name) \ |
|
25 | #define listdir_slot(name) \ | |
26 | static PyObject *listdir_stat_##name(PyObject *self, void *x) \ |
|
26 | static PyObject *listdir_stat_##name(PyObject *self, void *x) \ | |
27 | { \ |
|
27 | { \ | |
28 |
return PyInt_FromLong(((struct listdir_stat *) |
|
28 | return PyInt_FromLong(((struct listdir_stat *)self)->st.name); \ | |
29 | } |
|
29 | } | |
30 |
|
30 | |||
31 | listdir_slot(st_dev); |
|
31 | listdir_slot(st_dev); | |
32 | listdir_slot(st_mode); |
|
32 | listdir_slot(st_mode); | |
33 | listdir_slot(st_nlink); |
|
33 | listdir_slot(st_nlink); | |
34 | listdir_slot(st_size); |
|
34 | listdir_slot(st_size); | |
35 | listdir_slot(st_mtime); |
|
35 | listdir_slot(st_mtime); | |
36 | listdir_slot(st_ctime); |
|
36 | listdir_slot(st_ctime); | |
37 |
|
37 | |||
38 | static struct PyGetSetDef listdir_stat_getsets[] = { |
|
38 | static struct PyGetSetDef listdir_stat_getsets[] = { | |
39 |
|
|
39 | {"st_dev", listdir_stat_st_dev, 0, 0, 0}, | |
40 |
|
|
40 | {"st_mode", listdir_stat_st_mode, 0, 0, 0}, | |
41 |
|
|
41 | {"st_nlink", listdir_stat_st_nlink, 0, 0, 0}, | |
42 |
|
|
42 | {"st_size", listdir_stat_st_size, 0, 0, 0}, | |
43 |
|
|
43 | {"st_mtime", listdir_stat_st_mtime, 0, 0, 0}, | |
44 |
|
|
44 | {"st_ctime", listdir_stat_st_ctime, 0, 0, 0}, | |
45 |
|
|
45 | {0, 0, 0, 0, 0} | |
46 | }; |
|
46 | }; | |
47 |
|
47 | |||
48 | static PyObject *listdir_stat_new(PyTypeObject *t, PyObject *a, PyObject *k) |
|
48 | static PyObject *listdir_stat_new(PyTypeObject *t, PyObject *a, PyObject *k) | |
49 | { |
|
49 | { | |
50 |
|
|
50 | return t->tp_alloc(t, 0); | |
51 | } |
|
51 | } | |
52 |
|
52 | |||
53 | static void listdir_stat_dealloc(PyObject *o) |
|
53 | static void listdir_stat_dealloc(PyObject *o) | |
54 | { |
|
54 | { | |
55 |
|
|
55 | o->ob_type->tp_free(o); | |
56 | } |
|
56 | } | |
57 |
|
57 | |||
58 | static PyTypeObject listdir_stat_type = { |
|
58 | static PyTypeObject listdir_stat_type = { | |
59 |
|
|
59 | PyObject_HEAD_INIT(NULL) | |
60 |
|
|
60 | 0, /*ob_size*/ | |
61 |
|
|
61 | "osutil.stat", /*tp_name*/ | |
62 |
|
|
62 | sizeof(struct listdir_stat), /*tp_basicsize*/ | |
63 |
|
|
63 | 0, /*tp_itemsize*/ | |
64 |
|
|
64 | (destructor)listdir_stat_dealloc, /*tp_dealloc*/ | |
65 |
|
|
65 | 0, /*tp_print*/ | |
66 |
|
|
66 | 0, /*tp_getattr*/ | |
67 |
|
|
67 | 0, /*tp_setattr*/ | |
68 |
|
|
68 | 0, /*tp_compare*/ | |
69 |
|
|
69 | 0, /*tp_repr*/ | |
70 |
|
|
70 | 0, /*tp_as_number*/ | |
71 |
|
|
71 | 0, /*tp_as_sequence*/ | |
72 |
|
|
72 | 0, /*tp_as_mapping*/ | |
73 |
|
|
73 | 0, /*tp_hash */ | |
74 |
|
|
74 | 0, /*tp_call*/ | |
75 |
|
|
75 | 0, /*tp_str*/ | |
76 |
|
|
76 | 0, /*tp_getattro*/ | |
77 |
|
|
77 | 0, /*tp_setattro*/ | |
78 |
|
|
78 | 0, /*tp_as_buffer*/ | |
79 |
|
|
79 | Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/ | |
80 |
|
|
80 | "stat objects", /* tp_doc */ | |
81 |
|
|
81 | 0, /* tp_traverse */ | |
82 |
|
|
82 | 0, /* tp_clear */ | |
83 |
|
|
83 | 0, /* tp_richcompare */ | |
84 |
|
|
84 | 0, /* tp_weaklistoffset */ | |
85 |
|
|
85 | 0, /* tp_iter */ | |
86 |
|
|
86 | 0, /* tp_iternext */ | |
87 |
|
|
87 | 0, /* tp_methods */ | |
88 |
|
|
88 | 0, /* tp_members */ | |
89 |
|
|
89 | listdir_stat_getsets, /* tp_getset */ | |
90 |
|
|
90 | 0, /* tp_base */ | |
91 |
|
|
91 | 0, /* tp_dict */ | |
92 |
|
|
92 | 0, /* tp_descr_get */ | |
93 |
|
|
93 | 0, /* tp_descr_set */ | |
94 |
|
|
94 | 0, /* tp_dictoffset */ | |
95 |
|
|
95 | 0, /* tp_init */ | |
96 |
|
|
96 | 0, /* tp_alloc */ | |
97 |
|
|
97 | listdir_stat_new, /* tp_new */ | |
98 | }; |
|
98 | }; | |
99 |
|
99 | |||
100 | static inline int mode_to_kind(int mode) |
|
100 | static inline int mode_to_kind(int mode) | |
101 | { |
|
101 | { | |
102 |
|
|
102 | if (S_ISREG(mode)) return S_IFREG; | |
103 |
|
|
103 | if (S_ISDIR(mode)) return S_IFDIR; | |
104 |
|
|
104 | if (S_ISLNK(mode)) return S_IFLNK; | |
105 |
|
|
105 | if (S_ISBLK(mode)) return S_IFBLK; | |
106 |
|
|
106 | if (S_ISCHR(mode)) return S_IFCHR; | |
107 |
|
|
107 | if (S_ISFIFO(mode)) return S_IFIFO; | |
108 |
|
|
108 | if (S_ISSOCK(mode)) return S_IFSOCK; | |
109 |
|
|
109 | return mode; | |
110 | } |
|
110 | } | |
111 |
|
111 | |||
112 | static PyObject *listdir(PyObject *self, PyObject *args, PyObject *kwargs) |
|
112 | static PyObject *listdir(PyObject *self, PyObject *args, PyObject *kwargs) | |
113 | { |
|
113 | { | |
|
114 | static char *kwlist[] = { "path", "stat", NULL }; | |||
|
115 | PyObject *statobj = NULL; | |||
114 |
|
|
116 | DIR *dir = NULL; | |
115 |
|
|
117 | struct dirent *ent; | |
116 |
|
|
118 | PyObject *list = NULL; | |
117 |
|
|
119 | PyObject *ctor_args = NULL; | |
118 |
|
|
120 | int all_kinds = 1; | |
119 |
|
|
121 | char *full_path; | |
120 |
|
|
122 | int path_len; | |
121 |
|
|
123 | int do_stat; | |
122 |
|
|
124 | char *path; | |
123 |
|
|
125 | int ret; | |
124 |
|
126 | ssize_t size; | ||
125 | { |
|
127 | ssize_t i; | |
126 | static char *kwlist[] = { "path", "stat", NULL }; |
|
128 | int dfd; | |
127 | PyObject *statobj = NULL; |
|
|||
128 |
|
129 | |||
129 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s#|O:listdir", kwlist, |
|
130 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s#|O:listdir", kwlist, | |
130 | &path, &path_len, &statobj)) |
|
131 | &path, &path_len, &statobj)) | |
131 |
|
|
132 | goto bail; | |
132 |
|
133 | |||
133 | do_stat = statobj && PyObject_IsTrue(statobj); |
|
134 | do_stat = statobj && PyObject_IsTrue(statobj); | |
134 | } |
|
|||
135 |
|
135 | |||
136 |
|
|
136 | dir = opendir(path); | |
|
137 | if (!dir) { | |||
137 | list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, path); |
|
138 | list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, path); | |
138 |
|
|
139 | goto bail; | |
139 | } |
|
140 | } | |
140 |
|
141 | |||
141 |
|
|
142 | list = PyList_New(0); | |
|
143 | if (!list) | |||
142 |
|
|
144 | goto bail; | |
143 |
|
145 | |||
144 |
|
|
146 | full_path = alloca(path_len + PATH_MAX + 2); | |
145 |
|
|
147 | memcpy(full_path, path, path_len); | |
146 |
|
|
148 | full_path[path_len] = '/'; | |
147 |
|
149 | |||
148 | while ((ent = readdir(dir))) { |
|
150 | for (ent = readdir(dir); ent; ent = readdir(dir)) { | |
149 |
|
|
151 | PyObject *name = NULL; | |
150 |
|
|
152 | PyObject *py_kind = NULL; | |
151 |
|
|
153 | PyObject *val = NULL; | |
152 |
|
|
154 | unsigned char d_type; | |
153 |
|
|
155 | int kind = -1; | |
154 |
|
156 | |||
155 |
|
|
157 | if (!strcmp(ent->d_name, ".") || !strcmp(ent->d_name, "..")) | |
156 | continue; |
|
158 | continue; | |
157 |
|
159 | |||
158 | #ifdef DT_REG |
|
160 | #ifdef DT_REG | |
159 | if (do_stat) |
|
161 | if (do_stat) | |
160 |
|
|
162 | d_type = 0; | |
161 | else |
|
163 | else | |
162 |
|
|
164 | d_type = ent->d_type; | |
163 | #else |
|
165 | #else | |
164 |
|
|
166 | d_type = 0; | |
165 | #endif |
|
167 | #endif | |
166 |
|
168 | |||
167 |
|
|
169 | switch (d_type) { | |
168 | #ifdef DT_REG |
|
170 | #ifdef DT_REG | |
169 |
|
|
171 | case DT_REG: kind = S_IFREG; break; | |
170 |
|
|
172 | case DT_DIR: kind = S_IFDIR; break; | |
171 |
|
|
173 | case DT_LNK: kind = S_IFLNK; break; | |
172 |
|
|
174 | case DT_BLK: kind = S_IFBLK; break; | |
173 |
|
|
175 | case DT_CHR: kind = S_IFCHR; break; | |
174 |
|
|
176 | case DT_FIFO: kind = S_IFIFO; break; | |
175 |
|
|
177 | case DT_SOCK: kind = S_IFSOCK; break; | |
176 | #endif |
|
178 | #endif | |
177 | default: |
|
179 | default: | |
178 |
|
|
180 | if (all_kinds) | |
179 | all_kinds = 0; |
|
181 | all_kinds = 0; | |
180 |
|
|
182 | break; | |
181 | } |
|
183 | } | |
182 |
|
184 | |||
183 |
|
|
185 | name = PyString_FromString(ent->d_name); | |
184 | if (kind != -1) |
|
186 | if (kind != -1) | |
185 |
|
|
187 | py_kind = PyInt_FromLong(kind); | |
186 | else { |
|
188 | else { | |
187 |
|
|
189 | py_kind = Py_None; | |
188 |
|
|
190 | Py_INCREF(Py_None); | |
189 | } |
|
191 | } | |
190 |
|
192 | |||
191 |
|
|
193 | val = PyTuple_New(do_stat ? 3 : 2); | |
192 |
|
194 | |||
193 |
|
|
195 | if (!name || !py_kind || !val) { | |
194 |
|
|
196 | Py_XDECREF(name); | |
195 |
|
|
197 | Py_XDECREF(py_kind); | |
196 |
|
|
198 | Py_XDECREF(val); | |
197 |
|
||||
198 | goto bail; |
|
199 | goto bail; | |
199 | } |
|
200 | } | |
200 |
|
201 | |||
201 |
|
|
202 | PyTuple_SET_ITEM(val, 0, name); | |
202 |
|
|
203 | PyTuple_SET_ITEM(val, 1, py_kind); | |
203 | if (do_stat) { |
|
204 | if (do_stat) { | |
204 |
|
|
205 | PyTuple_SET_ITEM(val, 2, Py_None); | |
205 |
|
|
206 | Py_INCREF(Py_None); | |
206 | } |
|
207 | } | |
207 |
|
208 | |||
208 |
|
|
209 | PyList_Append(list, val); | |
209 |
|
|
210 | Py_DECREF(val); | |
210 | } |
|
211 | } | |
211 |
|
212 | |||
212 |
|
|
213 | PyList_Sort(list); | |
|
214 | size = PyList_Size(list); | |||
|
215 | #ifdef AT_SYMLINK_NOFOLLOW | |||
|
216 | dfd = dirfd(dir); | |||
|
217 | #endif | |||
213 |
|
218 | |||
214 |
|
|
219 | if (do_stat || !all_kinds) { | |
215 | ssize_t size = PyList_Size(list); |
|
|||
216 | ssize_t i; |
|
|||
217 | #ifdef AT_SYMLINK_NOFOLLOW |
|
|||
218 | int dfd = dirfd(dir); |
|
|||
219 | #endif |
|
|||
220 |
|
||||
221 | for (i = 0; i < size; i++) { |
|
220 | for (i = 0; i < size; i++) { | |
222 |
|
|
221 | PyObject *elt = PyList_GetItem(list, i); | |
223 |
|
|
222 | char *name = PyString_AsString(PyTuple_GET_ITEM(elt, 0)); | |
224 |
|
|
223 | PyObject *py_st = NULL; | |
225 |
|
|
224 | PyObject *py_kind = PyTuple_GET_ITEM(elt, 1); | |
226 |
|
|
225 | int kind; | |
227 |
|
226 | |||
228 |
|
|
227 | kind = py_kind == Py_None ? -1 : PyInt_AsLong(py_kind); | |
229 |
|
228 | |||
230 |
|
|
229 | if (kind != -1 && !do_stat) | |
231 | continue; |
|
230 | continue; | |
232 |
|
231 | |||
233 |
|
|
232 | strcpy(full_path + path_len + 1, name); | |
234 |
|
233 | |||
235 |
|
|
234 | if (do_stat) { | |
236 | struct listdir_stat *st; |
|
235 | struct listdir_stat *st; | |
237 |
|
236 | |||
238 |
if (ctor_args |
|
237 | if (!ctor_args) { | |
239 |
|
|
238 | ctor_args = PyTuple_New(0); | |
240 |
|
|
239 | if (!ctor_args) | |
241 | goto bail; |
|
240 | goto bail; | |
242 | } |
|
241 | } | |
243 |
|
242 | |||
244 | st = (struct listdir_stat *) |
|
243 | st = (struct listdir_stat *) | |
245 |
|
|
244 | PyObject_CallObject((PyObject *)&listdir_stat_type, | |
246 | ctor_args); |
|
245 | ctor_args); | |
247 | if (st == NULL) |
|
246 | ||
|
247 | if (!st) | |||
248 |
|
|
248 | goto bail; | |
249 | #ifdef AT_SYMLINK_NOFOLLOW |
|
249 | #ifdef AT_SYMLINK_NOFOLLOW | |
250 | ret = fstatat(dfd, name, &st->st, AT_SYMLINK_NOFOLLOW); |
|
250 | ret = fstatat(dfd, name, &st->st, AT_SYMLINK_NOFOLLOW); | |
251 | #else |
|
251 | #else | |
252 | ret = lstat(full_path, &st->st); |
|
252 | ret = lstat(full_path, &st->st); | |
253 | #endif |
|
253 | #endif | |
254 | if (ret == -1) { |
|
254 | if (ret == -1) { | |
255 |
|
|
255 | list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, | |
256 | full_path); |
|
256 | full_path); | |
257 |
|
|
257 | goto bail; | |
258 | } |
|
258 | } | |
259 | if (kind == -1) |
|
259 | if (kind == -1) | |
260 |
|
|
260 | kind = mode_to_kind(st->st.st_mode); | |
261 |
py_st = (PyObject *) |
|
261 | py_st = (PyObject *)st; | |
262 |
|
|
262 | } else { | |
263 | struct stat buf; |
|
263 | struct stat buf; | |
264 | #ifdef AT_SYMLINK_NOFOLLOW |
|
264 | #ifdef AT_SYMLINK_NOFOLLOW | |
265 | ret = fstatat(dfd, ent->d_name, &buf, AT_SYMLINK_NOFOLLOW); |
|
265 | ret = fstatat(dfd, ent->d_name, &buf, AT_SYMLINK_NOFOLLOW); | |
266 | #else |
|
266 | #else | |
267 | ret = lstat(full_path, &buf); |
|
267 | ret = lstat(full_path, &buf); | |
268 | #endif |
|
268 | #endif | |
269 | if (ret == -1) { |
|
269 | if (ret == -1) { | |
270 |
|
|
270 | list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, | |
271 | full_path); |
|
271 | full_path); | |
272 |
|
|
272 | goto bail; | |
273 | } |
|
273 | } | |
274 | if (kind == -1) |
|
274 | if (kind == -1) | |
275 |
|
|
275 | kind = mode_to_kind(buf.st_mode); | |
276 | } |
|
276 | } | |
277 |
|
277 | |||
278 |
|
|
278 | if (py_kind == Py_None && kind != -1) { | |
279 | py_kind = PyInt_FromLong(kind); |
|
279 | py_kind = PyInt_FromLong(kind); | |
280 |
if (py_kind |
|
280 | if (!py_kind) | |
281 |
|
|
281 | goto bail; | |
282 | Py_XDECREF(Py_None); |
|
282 | Py_XDECREF(Py_None); | |
283 | PyTuple_SET_ITEM(elt, 1, py_kind); |
|
283 | PyTuple_SET_ITEM(elt, 1, py_kind); | |
284 | } |
|
284 | } | |
285 |
|
285 | |||
286 |
|
|
286 | if (do_stat) { | |
287 |
if (py_st |
|
287 | if (!py_st) { | |
288 |
|
|
288 | py_st = Py_None; | |
289 |
|
|
289 | Py_INCREF(Py_None); | |
290 | } |
|
290 | } | |
291 | PyTuple_SET_ITEM(elt, 2, py_st); |
|
291 | PyTuple_SET_ITEM(elt, 2, py_st); | |
292 | } |
|
292 | } | |
293 | } |
|
293 | } | |
294 | } |
|
294 | } | |
295 |
|
295 | |||
296 |
|
|
296 | goto done; | |
297 |
|
297 | |||
298 | bail: |
|
298 | bail: | |
299 |
|
|
299 | Py_XDECREF(list); | |
300 |
|
300 | |||
301 | done: |
|
301 | done: | |
302 |
|
|
302 | Py_XDECREF(ctor_args); | |
303 |
|
|
303 | if (dir) | |
304 |
|
|
304 | closedir(dir); | |
305 |
|
||||
306 |
|
|
305 | return list; | |
307 | } |
|
306 | } | |
308 |
|
307 | |||
|
308 | ||||
309 | static char osutil_doc[] = "Native operating system services."; |
|
309 | static char osutil_doc[] = "Native operating system services."; | |
310 |
|
310 | |||
311 | static PyMethodDef methods[] = { |
|
311 | static PyMethodDef methods[] = { | |
312 |
|
|
312 | {"listdir", (PyCFunction)listdir, METH_VARARGS | METH_KEYWORDS, | |
313 |
|
|
313 | "list a directory\n"}, | |
314 |
|
|
314 | {NULL, NULL} | |
315 | }; |
|
315 | }; | |
316 |
|
316 | |||
317 | PyMODINIT_FUNC initosutil(void) |
|
317 | PyMODINIT_FUNC initosutil(void) | |
318 | { |
|
318 | { | |
319 |
|
|
319 | if (PyType_Ready(&listdir_stat_type) == -1) | |
320 | return; |
|
320 | return; | |
321 |
|
321 | |||
322 |
|
|
322 | Py_InitModule3("osutil", methods, osutil_doc); | |
323 | } |
|
323 | } |
General Comments 0
You need to be logged in to leave comments.
Login now