Нейл Мэтью - Основы программирования в Linux

18. Функция list_tracks— утилита, которая выводит все дорожки для заданного элемента каталога:

static void list_tracks(const cdc_entry *entry_to_use) {

int track_no = 1;

cdt_entry entry_found;

display_cdc(entry_to_use);

printf("\nTracks\n");

do {

entry_found = get_cdt_entry(entry_to_use->catalog, track_no);

if (entry_found.catalog[0]) {

display_cdt(&entry_found);

track_no++;

}

} while(entry_found.catalog[0]);

(void)get_confirm("Press return");

} /* list_tracks */

19. Функция count_all_entriesподсчитывает все дорожки:

static void count_all_entries(void) {

int cd_entries_found = 0;

int track_entries_found = 0;

cdc_entry cdc_found;

cdt_entry cdt_found;

int track_no = 1;

int first_time = 1;

char *search_string = "";

do {

cdc_found = search_cdc_entry(search_string, &first_time);

if (cdc_found.catalog[0]) {

cd_entries_found++;

track_no = 1;

do {

cdt_found = get_cdt_entry(cdc_found.catalog, track_no);

if (cdt_found.catalog[0]) {

track_entries_found++;

track_no++;

}

} while (cdt_found.catalog[0]);

}

} while (cdc_found.catalog[0]);

printf("Found %d CDs, with a total of %d tracks\n",

cd_entries_found, track_entries_found);

(void)get_confirm("Press return");

}

20. Теперь у вас есть утилита display_cdcдля вывода элемента каталога:

static void display_cdc(const cdc_entry *cdc_to_show) {

printf("Catalog: %s\n", cdc_to_show->catalog);

printf("\ttitle: %s\n", cdc_to_show->title);

printf("\ttype: %s\n", cdc_to_show->type);

printf("\tartist: %s\n", cdc_to_show->artist);

}

и утилита display_cdtдля отображения элемента-дорожки:

static void display_cdt(const cdt_entry *cdt_to_show) {

printf("%d: %s\n", cdt_to_show->track_no,

cdt_to_show->track_txt);

}

21. Служебная функция strip_returnудаляет завершающий строку символ перевода строки. Помните о том, что Linux, как и UNIX, использует один символ перевода строки для обозначения конца строки.

static void strip_return(char *string_to_strip) {

int len;

len = strlen(string_to_strip);

if (string_to_strip[len - 1] == '\n')

string_to_strip[len - 1] = '\0';

}

22. Функция command_modeпредназначена для синтаксического анализа аргументов командной строки. Функция getopt— хороший способ убедиться в том, что ваша программа принимает аргументы, соответствующие стандартным соглашениям, принятым в системе Linux.

static int command_mode(int argc, char *argv[]) {

int c;

int result = EXIT_SUCCESS;

char *prog_name = argv[0];

/* Эти внешние переменные используются функцией getopt */

extern char *optarg;

extern optind, opterr, optopt;

while ((c = getopt(argc, argv, ":i")) != -1) {

switch(c) {

case 'i':

if (!database_initialize(1)) {

result = EXIT_FAILURE;

fprintf(stderr, "Failed to initialize database\n");

}

break;

case ':':

case '?':

default:

fprintf(stderr, "Usage: %s [-i]\n", prog_name);

result = EXIT_FAILURE;

break;

} /* switch */

} /* while */

return(result);

}

Теперь переходите к функциям доступа к базе данных dbm.

1. Как обычно, начните с нескольких файлов #include. Далее примените директивы #defineдля задания файлов, которые будут использоваться для хранения данных:

#define _XOPEN_SOURCE

#include

#include

#include

#include

#include

#include

/* В некоторых дистрибутивах файл в предыдущей строке может быть придется заменить на gdbm-ndbm.h */

#include "cd_data.h"

#define CDC_FILE_BASE "cdc_data"

#define CDT_FILE_BASE "cdt_data"

#define CDC_FILE_DIR "cdc_data.dir"

#define CDC_FILE_PAG "cdc_data.pag"

#define CDT_FILE_DIR "cdt_data.dir"

#define CDT_FILE_PAG "cdt_data.pag"

2. Используйте эти две переменные области действия файла для отслеживания текущей базы данных:

static DBM *cdc_dbm_ptr = NULL;

static DBM *cdt_dbm_ptr = NULL;

3. По умолчанию функция database_initializeоткрывает существующую базу данных, но передав ненулевой (т.е. true) параметр new_database, вы можете заставить ее создать новую (пустую) базу данных, при этом существующая база данных удаляется. Если база данных успешно инициализирована, также инициализированы и два ее указателя, указывающие на то, что база данных открыта.

int database_initialize(const int new_database) {

int open_mode = O_CREAT | O_RDWR;

/* Если открыта какая-либо имеющаяся база данных, закрывает ее */

if (cdc_dbm_ptr) dbm_close(cdc_dbm_ptr);

if (cdt_dbm_ptr) dbm_close(cdt_dbm_ptr);

if (new_database) {

/* Удаляет старые файлы */

(void)unlink(CDC_FILE_PAG);

(void)unlink(CDC_FILE_DIR);

(void)unlink(CDT_FILE_PAG);

(void)unlink(CDT_FILE_DIR);

}

/* Открывает несколько новых файлов, создавая их при необходимости */

cdc_dbm_ptr = dbm_open(CDC_FILE_BASE, open_mode, 0644);

cdt_dbm_ptr = dbm_open(CDT_FILE_BASE, open_mode, 0644);

if (!cdc_dbm_ptr || !cdt_dbm_ptr) {

fprintf(stderr, "Unable to create database\n");

cdc_dbm_ptr = cdt_dbm_ptr = NULL;

return (0);

}

return (1);

}

4. Функция database_closeпросто закрывает базу данных, если она была открыта и устанавливает указатели базы данных в null, чтобы показать, что нет открытой базы данных:

void database_close(void) {

if (cdc_dbm_ptr) dbm_close(cdc_dbm_ptr);

if (cdt_dbm_ptr) dbm_close(cdt_dbm_ptr);

cdc_dbm_ptr = cdt_dbm_ptr = NULL;

}

5. Далее у вас появляется функция, извлекающая единственный элемент каталога, когда передан указатель на строку текста из каталога. Если элемент не найден, у возвращенных данных пустое поле каталога:

cdc_entry get_cdc_entry(const char *cd_catalog_ptr) {

cdc_entry entry_to_return;

char entry_to_find[CAT_CAT_LEN + 1];

datum local data datum;

datum local_key_datum;

memset(&entry_to_return, '\0', sizeof(entry_to_return));

6. Начните с некоторых имеющих смысл проверок, чтобы убедиться в том, что база данных открыта, и вы передали приемлемые параметры, т.е. ключ поиска содержит только допустимую строку и значения null:

if (!cdc_dbm_ptr || !cdt_dbm_ptr) return (entry_to_return);

if (!cd_catalog_ptr) return (entry_to_return);

if (strlen(cd_catalog_ptr) >= CAT_CAT_LEN) return (entry_to_return);

memset(&entry_to_find, '\0', sizeof(entry_to_find));

strcpy(entry_to_find, cd_catalog_ptr);

7. Задайте структуру datum, нужную функциям базы данных dbm, и используйте функцию dbm_fetchдля извлечения данных. Если не извлечены никакие данные, вы возвращаете пустую структуру entry_to_return, которая была инициализирована ранее: