Linux программирование в примерах
Шрифт:
31 quote(file));
32 G_fail = 1;
33 return 0;
34
35 case FTW_DCH:
36 /* Нельзя просто вернуться, поскольку, хотя nftw не может войти в
37 каталог, она может использовать stat, постольку у нас есть размер */
38 error(0, errno, _("cannot change to directory %s"), quote(file));
39 G_fail = 1;
40 break;
41
42 case FTW_DNR:
43 /*
Нельзя просто вернуться, поскольку, хотя nftw не может прочесть
44 каталог, она может вызвать stat, постольку у нас есть размер. */
45 error(0, errno, _("cannot read directory %s"), quote(file));
46 G_fail = 1;
47 break;
48
49 default:
50 break;
51 }
52
53 /* Если это первая (предварительная) встреча с каталогом,
54 сразу вернуться. */
55 if (file_type == FTW_DPRE)
56 return 0;
Строки 22–51 являются стандартным оператором
switch
. Ошибки, для которых нет информации о размере, устанавливают глобальную переменную G_fail
в 1 и возвращают 0, чтобы продолжить обработку (см строки 24–27 и 29–33). Ошибки, для которых есть размер, также устанавливают G_fail
, но затем прерывают switch
для того, чтобы обработать статистику (см. строки 35–40 и 42–47). Строки 55–56 сразу завершают функцию, если это первая встреча с каталогом
58 /* Если файл исключается или если он уже учитывался
59 через прямую ссылку, не включать его в сумму. */
60 if (info->skip,
61 || (!opt_count_all
62 && 1 < sb->st_nlink
63 && hash_ins(sb->st_ino, sb->st_dev)))
64 {
65 /* Заметьте, мы не должны здесь просто возвращаться.
66 Нам все еще нужно обновить prev_level и, возможно, передать
67 некоторые суммы выше по иерархии. */
68 size = 0;
69 print = 0;
70 }
71 else
72 {
73 size = (apparent_size
74 ? sb->st_size
75 : ST_NBLOCKS (*sb) * ST_NBLOCKSIZE);
76 }
Теперь становится интересно. По умолчанию
du
подсчитывает пространство, занимаемое прямыми ссылками, лишь одни раз. Опция – -count-links
заставляет ее подсчитывать пространство для каждой ссылки; переменная opt_count_all
равна true, когда указана – -count-links
. Для отслеживания ссылок du
содержит хэш-таблицу [87] уже встречавшихся пар (устройство, индекс).87
Хэш-таблица является структурой данных, позволяющей быстрое получение сохраненной информации, подробности выходят за рамки данной книги — Примеч. автора.
Строки 60–63 проверяют, следует ли не включать файл в подсчет, либо из-за того, что он был исключен (
info->skip
равно true, строка 60), либо потому что не была указана – -count-links
(строка 61) и у файла несколько ссылок (строка 62) и файл уже находится в хеш-таблице (строка 63). В этом случае размер устанавливается в 0, поэтому он не входит в конечную сумму, a print
также устанавливается в false (строки 68–69). Если ни одно из этих условий не отмечается, размер вычисляется либо в соответствии с размером в
struct stat
, либо в соответствии с числом блоков диска (строки 73–75) Это решение основывается на переменной apparent_size
, которая установлена при использовании опции – -apparent-size
.
78 if (first_call)
79 {
80 n_alloc = info->level + 10; /* Allocate arrays */
81 sum_ent = XCALLOC(uintmax_t, n_alloc); /* to hold sums */
82 sum_subdir = XCALLOC(uintmax_t, n_alloc);
83 }
84 else
85 {
86 /* FIXME: Стыдно, что нам нужно приводить к типу size_t для избежания
87 предупреждений gcc о 'сравнении между знаковым и беззнаковым'.
88 Возможно, неизбежно, при условии, что члены структуры FTW
89 имеют тип 'int' (исторически), так как мне нужно, чтобы переменные
90 вроде n_alloc и prev_level имели осмысленные типы. */
91 if (n_alloc <= (size_t)info->level)
92 {
93 n_alloc = info->level * 2; /* Удвоить сумму */
94 sum_ent = XREALLOC(sum_ent, uintmax_t, realloc); /* И выделить повторно */
95 sum_subdir = XREALLOC(sum_subdir, uintmax_t, n_alloc);
96 }
97 }
98
99 size_to_print = size;
Строки 78–97 управляют динамической памятью, используемой для хранения статистики о размере файла,
first_call
является статической переменной (строка 12), которая равна true при первом вызове process_file
. В этом случае вызывается calloc
(через упаковывающий макрос в строках 81–82; это обсуждалось в разделе 3.2.1.8 «Пример чтение строк произвольной длины»). Остальную часть времени first_call
равно false, и используется realloc
(снова через упаковывающий макрос, строки 91–96). Строка 99 заносит значение
size
в size_to_print
; эта переменная может обновляться в зависимости от того, должна ли она включать размеры дочерних элементов. Хотя size
могла бы использоваться повторно, отдельная переменная упрощает чтение кода.
101 if (!first_call)
102 {
103 if ((size_t)info->level == prev_level)
104 {
105 /* Обычно самый частый случай. Ничего не делать. */
Поделиться с друзьями: