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

8. В программе есть несколько участков, в которых хотелось бы спросить пользователя о том, уверен ли он в своем запросе. Вместо того чтобы вставлять в эти места программный код, задающий вопрос, поместим его в отдельную функцию get_confirm:

static int get_confirm(const char *question) {

char tmp_str[TMP_STRING_LEN + 1];

printf("%s", question);

fgets(tmp_str, TMP_STRING_LEN, stdin);

if (tmp_str[0] == 'Y' || tmp_str[0] = 'y') {

return(1);

}

return(0);

}

9. Функция enter_new_cat_entryпозволяет вводить новый элемент каталога. Вам не нужно сохранять перевод строки, который возвращает функция fgets, поэтому отбросьте его:

static int enter_new_cat_entry(cdc_entry *entry_to_update) {

cdc_entry new_entry;

char tmp_str[TMP_STRING_LEN + 1];

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

printf("Enter catalog entry: ");

(void)fgets(tmp_str, TMP_STRING_LEN, stdin);

strip_return(tmp_str);

strncpy(new_entry.catalog, tmp_str, CAT_CAT_LEN - 1);

printf("Enter title: ");

(void)fgets(tmp_str, TMP_STRING_LEN, stdin);

strip_return(tmp_str);

strncpy(new_entry.title, tmp_str, CAT_TITLE_LEN - 1);

printf("Enter type: ");

(void)fgets(tmp_str, TMP_STRING_LEN, stdin);

strip_return(tmp_str);

strncpy(new_entry.type, tmp_str, CAT_TYPE_LEN - 1);

printf("Enter artist: ");

(void)fgets(tmp_str, TMP_STRING_LEN, stdin);

strip_return(tmp_str);

strncpy(new_entry.artist, tmp_str, CAT_ARTIST_LEN - 1);

printf("\nNew catalog entry entry is :-\n");

display_cdc(&new_entry);

if (get_confirm("Add this entry ? ")) {

memcpy(entry_to_update, &new_entry, sizeof(new_entry));

return(1);

}

return(0);

}

Обратите внимание на то, что вы не применяете функцию gets, поскольку нет способа проверить переполнение буфера. Всегда избегайте применения функции gets!

10. Теперь вы переходите к функции enter_new_track_entriesдля ввода информации о дорожке. Эта функция немного сложнее функции ввода элемента каталога, поскольку вы разрешаете существующему элементу-дорожке оставаться неизменным:

static void enter_new_track_entries(const cdc_entry *entry_to_add_to) {

cdt_entry new_track, existing_track;

char tmp_str[TMP_STRING_LEN + 1];

int track_no = 1;

if (entry_to_add_to->catalog[0] == '\0') return;

printf("\nUpdating tracks for %s\n", entry_to_add_to->catalog);

printf("Press return to leave existing description unchanged, \n");

printf(" a single d to delete this and remaining tracks, \n");

printf(" or new track description\n");

while(1) {

11. Сначала вы должны проверить, существует ли уже дорожка с текущим номером дорожки. В зависимости от результатов проверки меняется строка приглашения:

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

existing_track = get_cdt_entry(entry_to_add_to->catalog,

track_no);

if (existing_track.catalog[0]) {

printf("\tTrack %d: %s\n", track_no,

existing_track.track_txt);

printf("\tNew text: ");

} else {

printf("\tTrack %d description: ", track_no);

}

fgets(tmp_str, TMP_STRING_LEN, stdin);

strip_return(tmp_str);

12. Если для данной дорожки не существует элемент и пользователь его не добавил, предположите, что больше нет дорожек, которые надо добавить:

if (strlen(tmp_str) == 0) {

if (existing_track.catalog[0] == '\0') {

/* Нет в наличии элемента, поэтому вставка завершается */

break;

} else {

/* Оставляем существующий элемент,

переходам к следующей дорожке */

track_no++;

continue;

}

}

13. Если пользователь введет единичный символ d, это приведет к удалению текущей дорожки и дорожек с большими номерами. Функция del_cdt_entryвернет false, если не сможет найти дорожку, которую следует удалить:

if ((strlen(tmp_str) == 1) && tmp_str[0] == 'd') { /* Удаляет эту и оставшиеся дорожки */

while (del_cdt_entry(entry_to_add_to->catalog, track_no)) {

track_no++;

}

break;

}

14. В этом пункте приводится код для вставки новой дорожки или обновления существующей. Вы формируете элемент cdt_entryструктуры new_trackи затем вызываете функцию базы данных add_cdt_entryдля того, чтобы включить его в базу данных:

strncpy(new_track. track_txt, tmp_str, TRACK_TTEXT_LEN - 1);

strcpy(new_track.catalog, entry_to_add_to->catalog);

new_track.track_no = track_no;

if (!add_cdt_entry(new_track)) {

fprintf(stderr, "Failed to add new track\n");

break;

}

track_no++;

} /* while */

}

15. Функция del_cat_entryудаляет элемент каталога. Никогда не разрешайте хранить дорожки для несуществующего элемента каталога.

static void del_cat_entry(const cdc_entry *entry_to_delete) {

int track_no = 1;

int delete_ok;

display_cdc(entry_to_delete);

if (get_confirm("Delete this entry and all it's tracks? ")) {

do {

delete_ok = del_cdt_entry(entry_to_delete->catalog, track_no);

track_no++;

} while(delete_ok);

if (!del_cdc_entry(entry_to_delete->catalog)) {

fprintf(stderr, "Failed to delete entry\n");

}

}

}

16. Следующая функция — утилита для удаления всех дорожек элемента каталога:

static void del_track_entries(const cdc_entry *entry_to_delete) {

int track_no = 1;

int delete_ok;

display_cdc(entry_to_delete);

if (get_confirm("Delete tracks for this entry? ")) {

do {

delete_ok = del_cdt_entry(entry_to_delete->catalog, track_no);

track_no++;

} while(delete_ok);

}

}

17. Создайте очень простое средство поиска, в котором разрешите пользователю ввести строку и затем поищите элементы каталога, содержащие строку. Поскольку может быть несколько элементов с такой строкой, просто по очереди предлагаются пользователю все найденные:

static cdc_entry find_cat(void) {

cdc_entry item_found;

char tmp_str[TMP_STRING_LEN + 1];

int first_call = 1;

int any_entry_found = 0;

int string ok;

int entry_selected = 0;

do {

string_ok = 1;

printf("Enter string to search for in catalog entry: ");

fgets(tmp_str, TMP_STRING_LEN, stdin);

strip_return(tmp_str);

if (strlen(tmp_str) > CAT_CAT_LEN) {

fprintf(stderr, "Sorry, string too long, maximum %d \

characters\n", CAT_CAT_LEN);

string_ok = 0;

}

} while (!string_ok);

while (!entry_selected) {

item_found = search_cdc_entry(tmp_str, &firstcall);

if (item_found.catalog[0] != '\0') {

any_entry_found = 1;

printf("\n");

display_cdc(&item_found);

if (get_confirm("This entry? ")) {

entry_selected = 1;

}

} else {

if (any_entry_found) printf("Sorry, no more matches found\n");

else printf("Sorry, nothing found\n");

break;

}

}

return(item_found);

}