##// END OF EJS Templates
osutil.c: style fix - delete trailing end-of-line spaces
Giorgos Keramidas -
r5416:ca890c0c default
parent child Browse files
Show More
@@ -1,323 +1,323 b''
1 1 /*
2 2 osutil.c - native operating system services
3 3
4 4 Copyright 2007 Matt Mackall and others
5 5
6 6 This software may be used and distributed according to the terms of
7 7 the GNU General Public License, incorporated herein by reference.
8 8 */
9 9
10 10 #define _ATFILE_SOURCE
11 11 #include <Python.h>
12 12 #include <alloca.h>
13 13 #include <dirent.h>
14 14 #include <fcntl.h>
15 15 #include <string.h>
16 16 #include <sys/stat.h>
17 17 #include <sys/types.h>
18 18 #include <unistd.h>
19 19
20 20 struct listdir_stat {
21 21 PyObject_HEAD
22 22 struct stat st;
23 23 };
24 24
25 25 #define listdir_slot(name) \
26 26 static PyObject *listdir_stat_##name(PyObject *self, void *x) \
27 27 { \
28 28 return PyInt_FromLong(((struct listdir_stat *) self)->st.name); \
29 29 }
30 30
31 31 listdir_slot(st_dev);
32 32 listdir_slot(st_mode);
33 33 listdir_slot(st_nlink);
34 34 listdir_slot(st_size);
35 35 listdir_slot(st_mtime);
36 36 listdir_slot(st_ctime);
37 37
38 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 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 53 static void listdir_stat_dealloc(PyObject *o)
54 54 {
55 55 (*o->ob_type->tp_free)(o);
56 56 }
57 57
58 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 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 112 static PyObject *listdir(PyObject *self, PyObject *args, PyObject *kwargs)
113 113 {
114 114 DIR *dir = NULL;
115 115 struct dirent *ent;
116 116 PyObject *list = NULL;
117 117 PyObject *ctor_args = NULL;
118 118 int all_kinds = 1;
119 119 char *full_path;
120 120 int path_len;
121 121 int do_stat;
122 122 char *path;
123 123 int ret;
124
124
125 125 {
126 static char *kwlist[] = { "path", "stat", NULL };
126 static char *kwlist[] = { "path", "stat", NULL };
127 127 PyObject *statobj = NULL;
128 128
129 129 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s#|O:listdir", kwlist,
130 130 &path, &path_len, &statobj))
131 131 goto bail;
132 132
133 133 do_stat = statobj && PyObject_IsTrue(statobj);
134 134 }
135
135
136 136 if ((dir = opendir(path)) == NULL) {
137 137 list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, path);
138 138 goto bail;
139 139 }
140 140
141 141 if ((list = PyList_New(0)) == NULL)
142 142 goto bail;
143 143
144 144 full_path = alloca(path_len + PATH_MAX + 2);
145 145 memcpy(full_path, path, path_len);
146 146 full_path[path_len] = '/';
147 147
148 148 while ((ent = readdir(dir))) {
149 149 PyObject *name = NULL;
150 150 PyObject *py_kind = NULL;
151 151 PyObject *val = NULL;
152 152 unsigned char d_type;
153 153 int kind = -1;
154 154
155 155 if (strcmp(ent->d_name, ".") == 0 || strcmp(ent->d_name, "..") == 0)
156 156 continue;
157 157
158 158 #ifdef DT_REG
159 159 if (do_stat)
160 160 d_type = 0;
161 161 else
162 162 d_type = ent->d_type;
163 163 #else
164 164 d_type = 0;
165 165 #endif
166 166
167 167 switch (d_type) {
168 168 #ifdef DT_REG
169 169 case DT_REG: kind = S_IFREG; break;
170 170 case DT_DIR: kind = S_IFDIR; break;
171 171 case DT_LNK: kind = S_IFLNK; break;
172 172 case DT_BLK: kind = S_IFBLK; break;
173 173 case DT_CHR: kind = S_IFCHR; break;
174 174 case DT_FIFO: kind = S_IFIFO; break;
175 175 case DT_SOCK: kind = S_IFSOCK; break;
176 176 #endif
177 177 default:
178 178 if (all_kinds)
179 179 all_kinds = 0;
180 180 break;
181 181 }
182 182
183 183 name = PyString_FromString(ent->d_name);
184 184 if (kind != -1)
185 185 py_kind = PyInt_FromLong(kind);
186 186 else {
187 187 py_kind = Py_None;
188 188 Py_INCREF(Py_None);
189 189 }
190
190
191 191 val = PyTuple_New(do_stat ? 3 : 2);
192
192
193 193 if (name == NULL || py_kind == NULL || val == NULL) {
194 194 Py_XDECREF(name);
195 195 Py_XDECREF(py_kind);
196 196 Py_XDECREF(val);
197 197
198 198 goto bail;
199 199 }
200 200
201 201 PyTuple_SET_ITEM(val, 0, name);
202 202 PyTuple_SET_ITEM(val, 1, py_kind);
203 203 if (do_stat) {
204 204 PyTuple_SET_ITEM(val, 2, Py_None);
205 205 Py_INCREF(Py_None);
206 206 }
207 207
208 208 PyList_Append(list, val);
209 209 Py_DECREF(val);
210 210 }
211
211
212 212 PyList_Sort(list);
213 213
214 214 if (do_stat || !all_kinds) {
215 215 ssize_t size = PyList_Size(list);
216 216 ssize_t i;
217 217 #ifdef AT_SYMLINK_NOFOLLOW
218 218 int dfd = dirfd(dir);
219 219 #endif
220 220
221 221 for (i = 0; i < size; i++) {
222 222 PyObject *elt = PyList_GetItem(list, i);
223 223 char *name = PyString_AsString(PyTuple_GET_ITEM(elt, 0));
224 224 PyObject *py_st = NULL;
225 225 PyObject *py_kind = PyTuple_GET_ITEM(elt, 1);
226 226 int kind;
227 227
228 228 kind = py_kind == Py_None ? -1 : PyInt_AsLong(py_kind);
229
229
230 230 if (kind != -1 && !do_stat)
231 231 continue;
232 232
233 233 strcpy(full_path + path_len + 1, name);
234 234
235 235 if (do_stat) {
236 236 struct listdir_stat *st;
237 237
238 238 if (ctor_args == NULL) {
239 239 ctor_args = PyTuple_New(0);
240 240 if (ctor_args == NULL)
241 241 goto bail;
242 242 }
243
243
244 244 st = (struct listdir_stat *)
245 245 PyObject_CallObject((PyObject *) &listdir_stat_type,
246 246 ctor_args);
247 247 if (st == NULL)
248 248 goto bail;
249 249 #ifdef AT_SYMLINK_NOFOLLOW
250 250 ret = fstatat(dfd, name, &st->st, AT_SYMLINK_NOFOLLOW);
251 251 #else
252 252 ret = lstat(full_path, &st->st);
253 253 #endif
254 254 if (ret == -1) {
255 255 list = PyErr_SetFromErrnoWithFilename(PyExc_OSError,
256 256 full_path);
257 257 goto bail;
258 258 }
259 259 if (kind == -1)
260 260 kind = mode_to_kind(st->st.st_mode);
261 261 py_st = (PyObject *) st;
262 262 } else {
263 263 struct stat buf;
264 264 #ifdef AT_SYMLINK_NOFOLLOW
265 265 ret = fstatat(dfd, ent->d_name, &buf, AT_SYMLINK_NOFOLLOW);
266 266 #else
267 267 ret = lstat(full_path, &buf);
268 268 #endif
269 269 if (ret == -1) {
270 270 list = PyErr_SetFromErrnoWithFilename(PyExc_OSError,
271 271 full_path);
272 272 goto bail;
273 273 }
274 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 279 py_kind = PyInt_FromLong(kind);
280 280 if (py_kind == NULL)
281 281 goto bail;
282 282 Py_XDECREF(Py_None);
283 283 PyTuple_SET_ITEM(elt, 1, py_kind);
284 284 }
285 285
286 286 if (do_stat) {
287 287 if (py_st == NULL) {
288 288 py_st = Py_None;
289 289 Py_INCREF(Py_None);
290 290 }
291 291 PyTuple_SET_ITEM(elt, 2, py_st);
292 292 }
293 293 }
294 294 }
295 295
296 296 goto done;
297 297
298 298 bail:
299 299 Py_XDECREF(list);
300 300
301 301 done:
302 302 Py_XDECREF(ctor_args);
303 303 if (dir)
304 304 closedir(dir);
305 305
306 306 return list;
307 307 }
308 308
309 309 static char osutil_doc[] = "Native operating system services.";
310 310
311 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 317 PyMODINIT_FUNC initosutil(void)
318 318 {
319 319 if (PyType_Ready(&listdir_stat_type) == -1)
320 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