第 14 章 结构和其他数据形式（names3）

 /*-----------------------------------

     names3.c -- 使用指针和 malloc()

 -----------------------------------*/

 #include <stdio.h>

 #include <string.h>        //提供 strcpy()、strlen() 原型

 #include <stdlib.h>        //提供 malloc()、free() 原型

 #define SLEN 81

 struct namect

 {

     char *fname;        //fname、lname 分别保存内存分配的地址

     char *lname;

     int letters;

 };

 void getinfo(struct namect *);        //分配内存

 void makeinfo(struct namect *);

 void showinfo(const struct namect *);

 void cleanup(struct namect *);

 char* s_gets(char *st, int n);

 int main()

 {

     struct namect person;

     getinfo(&person);

     makeinfo(&person);

     showinfo(&person);

     cleanup(&person);

     return ;

 }

 void getinfo(struct namect *pst)

 {

     char temp[SLEN];

     printf("Please enter your first name.\n");

     s_gets(temp, SLEN);

     pst->fname = (char*)malloc((strlen(temp) + ) * sizeof(char));

     strcpy(pst->fname, temp);

     printf("Please enter your last name.\n");

     s_gets(temp, SLEN);

     pst->lname = (char*)malloc((strlen(temp) + ) * sizeof(char));

     strcpy(pst->lname, temp);

 }

 void makeinfo(struct namect *pst)

 {

     pst->letters = strlen(pst->fname) + strlen(pst->lname);

 }

 void showinfo(const struct namect *pst)

 {

     printf("%s %s, your name contains %d letters.\n"

         , pst->fname, pst->lname, pst->letters);

 }

 void cleanup(struct namect *pst)

 {

     free(pst->fname);

     free(pst->lname);

 }

 char* s_gets(char *st, int n)

 {

     char *ret_val;

     char *find;

     if (ret_val = fgets(st, n, stdin))

     {

         if (find = strchr(st, '\n'))

             *find = '\0';

         else

             while (getchar() != '\n') continue;

     }

     return ret_val;

 }

names3.c