C语言写的一个通信录小程序（原创）

原创代码，转载需注明。
/*
 *CopyRight@sh365
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct _node{
    char name[32];
    char phone[13];
    char qq[12];
    char addres[128];
    struct _node *pNext;
}NODE;


char *myfgets(char *s, int n, FILE *stream)
{
    int i = 0;

    fgets(s, n, stream);
    i = (int)strlen(s);
    i = (i<=n-1) ? i : n-1;
    s[i-1] = ((s[i-1] == '\n')||(s[i-1] == '\r')) ? '\0' : s[i-1];

    return s;
}

int create_address_book(NODE **header)
{
    static int create_flag = 0;

    if (!create_flag) {
        (*header) = (NODE*)malloc(sizeof(NODE));
        if (NULL == *header) {
            printf("malloc error,create_address_book failed.\n");
            return -1;
        }
        memset(*header, 0, sizeof(NODE));
        create_flag = 1;
    }

    return 0;
}

int add_address_item(NODE *header)
{
    NODE *p = (NODE*)malloc(sizeof(NODE));
    char dumy[8];

    if (NULL == p) {
        printf("malloc error,add_address_item failed.\n");
        return -1;
    }
    memset(p, 0, sizeof(NODE));
    fgets(dumy, 7, stdin);
    printf("Input  Name: ");
    myfgets(p->name, 31, stdin);

    printf("Input Phone: ");
    myfgets(p->phone, 12, stdin);

    printf("Input    QQ: ");
    myfgets(p->qq, 11, stdin);

    printf("Input  Addr: ");
    myfgets(p->addres, 127, stdin);


    while (NULL != header->pNext) {
        header = header->pNext;
    }
    header->pNext = p;
    p->pNext = NULL;

    return 0;
}

int query_address_item(NODE *header)
{
    char name[32] = {'\0'};
    int find_flag = -1;
    myfgets(name,31,stdin);
    memset(name, '\0', sizeof(name));
    printf("Input Name for search: ");
    myfgets(name,31,stdin);

    find_flag = -1;
    header = header->pNext;
    while (NULL != header) {
        if (NULL != strstr(header->name, name)) {
            printf(" Name: %s\n", header->name);
            printf("Phone: %s\n", header->phone);
            printf("     QQ: %s\n", header->qq);
            printf(" Addr: %s\n\n", header->addres);
            find_flag = 0;
        }
        header = header->pNext;
    }

    return find_flag;
}

int delete_address_item(NODE *header)
{
    char name[32] = {'\0'};
    int find_flag = -1;
    NODE *p = NULL;

    memset(name, '\0', sizeof(name));
    printf("Input Name for delete: ");
    myfgets(name,31,stdin);

    find_flag = -1;
    while (NULL != header->pNext) {
        if (!strcmp(header->pNext->name, name)) {
            printf("will delete this item:\n");
            printf(" Name: %s\n", header->name);
            printf("Phone: %s\n", header->phone);
            printf("     QQ: %s\n", header->qq);
            printf(" Addr: %s\n\n", header->addres);

            p = header->pNext;
            header->pNext = p->pNext;
            free(p);
            find_flag = 0;
        }
        header = header->pNext;
    }

    return find_flag;

}

int output_address_book(NODE *header)
{
    header = header->pNext;
    while (NULL != header) {
        printf(" Name: %s\n", header->name);
        printf("Phone: %s\n", header->phone);
        printf("    QQ: %s\n", header->qq);
        printf(" Addr: %s\n\n", header->addres);

        header = header->pNext;
    }

    return 0;
}

int save_address_book(NODE *header)
{
    FILE *fp = NULL;
    fp = fopen("my_address_book.txt", "wt");
    if (NULL == fp) {
        printf("file open failed, can not save\n");
        return -1;
    }

    header = header->pNext;
    while (NULL != header) {
        fprintf(fp, " Name: %s\n", header->name);
        fprintf(fp, "Phone: %s\n", header->phone);
        fprintf(fp, "   QQ: %s\n", header->qq);
        fprintf(fp, " Addr: %s\n\n", header->addres);

        header = header->pNext;
    }
    fclose(fp);

    return 0;
}

int delete_address_book(NODE *header)
{
    NODE *p = NULL;

    while (NULL != header->pNext) {
        p = header->pNext;
        header->pNext = p->pNext;
        free(p);
    }
    free(header);

    return 0;
}


void usage()
{
    printf("\t|--------------------------------------------------------------|\n");
    printf("\t|            X Welcom to Students AddressBook    X             |\n");
    printf("\t|--------------------------------------------------------------|\n");
    printf("\t|****************X   1-Create AddressBook    X**************** |\n");
    printf("\t|****************X   2-Add    AddressBook    X**************** |\n");
    printf("\t|****************X   3-Query  AddressBook    X**************** |\n");
    printf("\t|****************X   4-Delete AddressBook    X**************** |\n");
    printf("\t|****************X   5-Output AddressBook    X**************** |\n");
    printf("\t|****************X   6-Save   AddressBook    X**************** |\n");
    printf("\t|****************X   0-Exit   AddressBook    X**************** |\n");
    printf("\t|--------------------------------------------------------------|\n");
    printf("Input a command(0--6): ");
}

/*
 *CopyRight @ sh365
 */

int main(int argc, char *argv[])
{
    int comd = 0;
    int quit = 0;
    int ret = 0;
    NODE *header = NULL;

    ret = create_address_book(&header);
    while (!quit) {
        usage();
        scanf("%d", &comd);

        switch (comd) {
            case 1:
                ret = create_address_book(&header);
                break;

            case 2:
                ret = add_address_item(header);
                break;

            case 3:
                ret = query_address_item(header);
                break;

            case 4:
                ret = delete_address_item(header);
                break;

            case 5:
                ret = output_address_book(header);
                break;

            case 6:
                ret = save_address_book(header);
                break;

            case 0:
                quit = 1;
                ret = delete_address_book(header);
                break;

            default:
                break;
        }
    }
}