Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * rmtree.c
4 : : *
5 : : * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
6 : : * Portions Copyright (c) 1994, Regents of the University of California
7 : : *
8 : : * IDENTIFICATION
9 : : * src/common/rmtree.c
10 : : *
11 : : *-------------------------------------------------------------------------
12 : : */
13 : :
14 : : #ifndef FRONTEND
15 : : #include "postgres.h"
16 : : #else
17 : : #include "postgres_fe.h"
18 : : #endif
19 : :
20 : : #include <unistd.h>
21 : : #include <sys/stat.h>
22 : :
23 : : #include "common/file_utils.h"
24 : :
25 : : #ifndef FRONTEND
26 : : #include "storage/fd.h"
27 : : #define pg_log_warning(...) elog(WARNING, __VA_ARGS__)
28 : : #define LOG_LEVEL WARNING
29 : : #define OPENDIR(x) AllocateDir(x)
30 : : #define CLOSEDIR(x) FreeDir(x)
31 : : #else
32 : : #include "common/logging.h"
33 : : #define LOG_LEVEL PG_LOG_WARNING
34 : : #define OPENDIR(x) opendir(x)
35 : : #define CLOSEDIR(x) closedir(x)
36 : : #endif
37 : :
38 : : /*
39 : : * rmtree
40 : : *
41 : : * Delete a directory tree recursively.
42 : : * Assumes path points to a valid directory.
43 : : * Deletes everything under path.
44 : : * If rmtopdir is true deletes the directory too.
45 : : * Returns true if successful, false if there was any problem.
46 : : * (The details of the problem are reported already, so caller
47 : : * doesn't really have to say anything more, but most do.)
48 : : */
49 : : bool
4340 peter_e@gmx.net 50 :CBC 3343 : rmtree(const char *path, bool rmtopdir)
51 : : {
52 : : char pathbuf[MAXPGPATH];
53 : : DIR *dir;
54 : : struct dirent *de;
949 tmunro@postgresql.or 55 : 3343 : bool result = true;
56 : 3343 : size_t dirnames_size = 0;
57 : 3343 : size_t dirnames_capacity = 8;
58 : : char **dirnames;
59 : :
60 : 3343 : dir = OPENDIR(path);
61 [ - + ]: 3343 : if (dir == NULL)
62 : : {
949 tmunro@postgresql.or 63 [ # # ]:UBC 0 : pg_log_warning("could not open directory \"%s\": %m", path);
4340 peter_e@gmx.net 64 : 0 : return false;
65 : : }
66 : :
768 michael@paquier.xyz 67 :CBC 3343 : dirnames = (char **) palloc(sizeof(char *) * dirnames_capacity);
68 : :
949 tmunro@postgresql.or 69 [ + + ]: 149633 : while (errno = 0, (de = readdir(dir)))
70 : : {
71 [ + + ]: 146290 : if (strcmp(de->d_name, ".") == 0 ||
72 [ + + ]: 142947 : strcmp(de->d_name, "..") == 0)
4340 peter_e@gmx.net 73 : 6686 : continue;
949 tmunro@postgresql.or 74 : 139604 : snprintf(pathbuf, sizeof(pathbuf), "%s/%s", path, de->d_name);
75 [ - + + ]: 139604 : switch (get_dirent_type(pathbuf, de, false, LOG_LEVEL))
76 : : {
949 tmunro@postgresql.or 77 :UBC 0 : case PGFILETYPE_ERROR:
78 : : /* already logged, press on */
79 : 0 : break;
949 tmunro@postgresql.or 80 :CBC 2590 : case PGFILETYPE_DIR:
81 : :
82 : : /*
83 : : * Defer recursion until after we've closed this directory, to
84 : : * avoid using more than one file descriptor at a time.
85 : : */
86 [ + + ]: 2590 : if (dirnames_size == dirnames_capacity)
87 : : {
88 : 183 : dirnames = repalloc(dirnames,
89 : : sizeof(char *) * dirnames_capacity * 2);
90 : 183 : dirnames_capacity *= 2;
91 : : }
92 : 2590 : dirnames[dirnames_size++] = pstrdup(pathbuf);
93 : 2590 : break;
94 : 137014 : default:
95 [ - + - - ]: 137014 : if (unlink(pathbuf) != 0 && errno != ENOENT)
96 : : {
841 peter@eisentraut.org 97 [ # # ]:UBC 0 : pg_log_warning("could not remove file \"%s\": %m", pathbuf);
4340 peter_e@gmx.net 98 : 0 : result = false;
99 : : }
949 tmunro@postgresql.or 100 :CBC 137014 : break;
101 : : }
102 : : }
103 : :
104 [ - + ]: 3343 : if (errno != 0)
105 : : {
949 tmunro@postgresql.or 106 [ # # ]:UBC 0 : pg_log_warning("could not read directory \"%s\": %m", path);
107 : 0 : result = false;
108 : : }
109 : :
949 tmunro@postgresql.or 110 :CBC 3343 : CLOSEDIR(dir);
111 : :
112 : : /* Now recurse into the subdirectories we found. */
113 [ + + ]: 5933 : for (size_t i = 0; i < dirnames_size; ++i)
114 : : {
115 [ - + ]: 2590 : if (!rmtree(dirnames[i], true))
949 tmunro@postgresql.or 116 :UBC 0 : result = false;
949 tmunro@postgresql.or 117 :CBC 2590 : pfree(dirnames[i]);
118 : : }
119 : :
4340 peter_e@gmx.net 120 [ + - ]: 3343 : if (rmtopdir)
121 : : {
122 [ - + ]: 3343 : if (rmdir(path) != 0)
123 : : {
949 tmunro@postgresql.or 124 [ # # ]:UBC 0 : pg_log_warning("could not remove directory \"%s\": %m", path);
4340 peter_e@gmx.net 125 : 0 : result = false;
126 : : }
127 : : }
128 : :
949 tmunro@postgresql.or 129 :CBC 3343 : pfree(dirnames);
130 : :
4340 peter_e@gmx.net 131 : 3343 : return result;
132 : : }
|
