Марк Митчелл - Программирование для Linux. Профессиональный подход

write(fd, page_start, strlen(page_start));

/* Копирование данных из файла /etc/issue

в клиентский сокет. */

rval = sendfile(fd, input_fd, &offset, file_info.st_size);

if (rval == -1)

/* При отправке файла /etc/issue произошла ошибка.

Выводим соответствующее сообщение. */

write(fd, error_message, strlen(error_message));

/* Конец страницы. */

write(fd, page_end, strlen(page_end));

}

close(input_fd);

}

Сначала модуль пытается открыть файл /etc/issue. Если это не удалось, клиенту возвращается сообщение об ошибке. В противном случае посылается начальный код HTML-страницы, содержащийся в переменной page_start, затем — содержимое файла /etc/issue(это делается с помощью функции sendfile(), о которой рассказывалось в разделе 8.12. "Функция sendfile(): быстрая передача данных") и, наконец конечный код HTML-страницы, содержащийся в переменной page_end.

Этот модуль можно легко настроить на отправку любого другого файла. Если файл содержит HTML-страницу, переменные page_startи page_endбудут не нужны.

Модуль diskfree.so(исходный текст приведен в листинге 11.8) генерирует страницу с информацией о свободном дисковом пространстве в файловых системах, смонтированных на серверном компьютере. Эта информация берется из выходных данных команды df -h. Как и в модуле issue.so, выходные данные заключаются в тэги

#include

#include

#include

#include

#include "server.h"

/* HTML-код начала генерируемой страницы. */

static char* page_start =

"\n"

"

"

"

"

void module_generate(int fd) {

pid_t child_pid;

int rval;

/* Запись начала страницы. */

write(fd, page_start, strlen(page_start));

/* Создание дочернего процесса. */

child_pid = fork();

if (child_pid == 0) {

/* Это дочерний процесс. */

/* Подготовка списка аргументов команды df. */

char* argv[] = { "/bin/df, "-h", NULL };

/* Дублирование потоков stdout и stderr для записи данных

в клиентский сокет. */

rval = dup2(fd, STDOUT_FILENO);

if (rval == -1)

system_error("dup2");

rval = dup2(fd, STDERR_FILENO);

if (rval == -1)

system_error("dup2");

/* Запуск команды df, отображающей объем свободного

пространства в смонтированных файловых системах. */

execv(argv[0], argv);

/* Функция execv() возвращает управление в программу только

при возникновении ошибки. */

system_error("execv");

} else if (child_pid > 0) {

/* Это родительский процесс, дожидаемся завершения дочернего

процесса. */

rval = waitpid(child_pid, NULL, 0);

if (rval == -1)

system_error("waitpid");

} else

/* Вызов функции fork() завершился неудачей. */

system_error("fork");

/* запись конца страницы. */

write(fd, page_end, strlen(page_end));

}

В то время как модуль issue.soпосылает содержимое файла с помощью функции sendfile(), данный модуль должен вызвать внешнюю команду и перенаправить результаты ее работы клиенту. Для этого модуль придерживается такой последовательности действий.

1. Сначала с помощью функции fork()создается дочерний процесс (см. раздел 3.2.2. "Функции fork() и exec()").

2. Дочерний процесс копирует дескриптор сокета в дескрипторы STDOUT_FILENOи STDERR_FILENO, соответствующие стандартным потокам вывода и ошибок (см. раздел 2.1.4, "Стандартный ввод-вывод"). Это копирование осуществляется с помощью системного вызова dup2()(см. раздел 5.4 3. "Перенаправление стандартных потоков ввода, вывода и ошибок"). Все последующие данные, записываемые в эти потоки в рамках дочернего процесса, будут направляться в сокет.

3. Дочерний процесс с помощью функции execv()вызывает команду df -h.

4. Родительский процесс дожидается завершения дочернего процесса, вызывая функцию waitpid()(см. раздел 5.4 2. "Системные вызовы wait()").

Этот модуль можно легко настроить на вызов другой системной команды.

Модуль processes.so(исходный текст приведен в листинге 11.9) сложнее остальных модулей. Он генерирует страницу, в которой содержится таблица процессов, выполняющихся в данный момент на сервере. Каждому процессу отводится в таблице одна строка. В этой строке указан идентификатор процесса, имя исполняемого файла, имена владельца и группы, которым принадлежит процесс, а также размер резидентной части процесса.

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include "server.h"

/* Эта функция записывает в аргументы UID и GID

идентификаторы пользователя и группы, которым

принадлежит процесс с указанным идентификатором,

в случае успешного завершения возвращается нуль,

иначе -- ненулевое значение. */

static int get_uid_gid(pid_t pid, uid_t* uid, gid_t* gid) {

char dir_name[64];

struct stat dir_info;

int rval;

/* Формирование имени каталога процесса

в файловой системе /proc. */

snprintf(dir_name, sizeof(dir_name), "/proc/%d", (int)pid);

/* Получение информации о каталоге. */

rval = stat(dir_name, &dir_info);

if (rval != 0)

/* Каталог не найден. Возможно, процесс больше

не существует. */

return 1;

/* Убеждаемся в том, что это действительно каталог. */

assert(S_ISDIR(dir_info.st_mode));

/* Определяем интересующие нас идентификаторы. */

*uid = dir_info.st_uid;

*gid = dir_info.st_gid;

return 0;

}

/* Эта функция находит имя пользователя,

соответствующее заданному идентификатору.

Возвращаемый буфер должен быть удален

в вызывающей функции. */

static char* get_user_name(uid_t uid) {

struct passwd* entry;

entry = getpwuid(uid);

if (entry == NULL)

system_error("getpwuid");

return xstrdup(entry->pw_name);

}

/* Эта функция находит имя группы, соответствующее

заданному идентификатору, возвращаемый буфер

должен быть удален в вызывающей функции. */

static char* get_group_name(gid_t gid) {

struct group* entry;

entry = getgrgid(gid);

if (entry == NULL)

system_error("getgrgid");

return xstrdup(entry->gr_name);

}