指向另一个结构的指针，从文件中写入和读取它会产生SegFault

我正在从学习C The Hard Way中刷出我的C技能，目前我正在进行第17次练习 。

我正在做“额外积分”部分。 使该页面上给出的数据库代码得到我正在尝试

“更改代码以接受MAX_DATA和MAX_ROWS的参数，将它们存储在Database结构中，并将其写入文件，从而创建一个可以任意resize的数据库”

所以，我注释掉#define指令，并更改了给定的地址和数据库结构：

struct Address { int id; int set; char *name; char *email; }; struct Database { int MAX_DATA; int MAX_ROWS; struct Address *rows; }; 

这样，我可以从用户那里获得MAX_DATA和MAX_ROWS参数，以根据自己的喜好创建数据库。 我在代码中更改的其他内容是 –

Database_create函数：

 void Database_create(struct Connection *conn, int MAX_DATA, int MAX_ROWS) { int i = 0; conn->db->MAX_DATA = MAX_DATA; conn->db->MAX_ROWS = MAX_ROWS; conn->db->rows = malloc(sizeof(struct Address) * MAX_ROWS); for(i = 0; i db->rows[i] = addr; } } 

我得到’无效读取4个字节’的部分，而SegFault是：

 void Database_set(struct Connection *conn, int id, const char *name, const char *email) { struct Address *addr = &conn->db->rows[id]; int MAX_DATA = conn->db->MAX_DATA; if(addr->set) die("Already set, delete it first"); /* This if statement gives error for addr->set */ addr->set = 1; addr->name = malloc(sizeof(char) * MAX_DATA); addr->email = malloc(sizeof(char) * MAX_DATA); // WARNING: bug, read the "How To Break It" and fix this char *res = strncpy(addr->name, name, MAX_DATA); // demonstrate the strncpy bug if(!res) die("Name copy failed"); res = strncpy(addr->email, email, MAX_DATA); if(!res) die("Email copy failed"); } 

我知道这段代码片段不是整个部分，但我不能在这里粘贴整个代码。 所以，我在这里发布了它： http ： //pastebin.com/EbKShT3r我可以使用’c’选项为第一次运行创建和编写数据库。 但是，要使用’s’选项添加条目，会发生分段错误。

编辑 ：所以，最后，我得到了这个程序使用@WhizCraig给出的解决方案然而，为了释放内存，这是我正在尝试：

 void Database_close(struct Connection *conn) { int i; if (conn) { int MAX_ROWS = conn->db->MAX_ROWS; for (i=0; idb->rows+i; if (row->set) { free(row->name); free(row->email); } } free(conn->db->rows); if(conn->file) fclose(conn->file); if(conn->db) free(conn->db); free(conn); } } 

而且，我在Valgrind遇到错误，内存正在泄漏。 我不明白上面代码中的错误。 但是，主要问题似乎已经解决:)

Answer 1

 struct Address { int id; int set; char *name; char *email; }; struct Database { int MAX_DATA; int MAX_ROWS; Address **rows; // USE ARRAY OF POINTERS }; void Database_create(Connection *conn, int MAX_DATA, int MAX_ROWS) { int i = 0; conn->db->MAX_DATA = MAX_DATA; conn->db->MAX_ROWS = MAX_ROWS; conn->db->rows = (Address**)malloc(sizeof(Address*) * MAX_ROWS); for(i = 0; i < MAX_ROWS; i++) { conn->db->rows[i] = (Address*)malloc(sizeof(Address)); conn->db->rows[i]->id = i; conn->db->rows[i]->set = 0; } } void Database_set(Connection *conn, int id, const char *name, const char *email) { if (!(conn && conn->db && conn->db->rows && conn->db->rows[id])) return; Address *addr = conn->db->rows[id]; int MAX_DATA = conn->db->MAX_DATA; if(addr->set == 0) die("Already set, delete it first"); addr->set = 1; addr->name = malloc(sizeof(char) * MAX_DATA); addr->email = malloc(sizeof(char) * MAX_DATA); char *res = strncpy(addr->name, name, MAX_DATA); if(!res) die("Name copy failed"); res = strncpy(addr->email, email, MAX_DATA); if(!res) die("Email copy failed"); } void Database_close(Connection *conn) { size_t i; if(conn) { if (con->db && conn->db->rows) { for (i = 0; i < conn->db->MAX_ROWS; i++) { Address *cur = conn->db->rows[i]; free(cur); } } if(conn->file) fclose(conn->file); if(conn->db) free(conn->db); free(conn); } } 

Answer 2

我修改了Rohit在这里发布的代码： http ： //pastebin.com/MvLXkDCz来修复内存泄漏。 它通过Valgrind没有任何错误。 相关修改发布如下：

 void Database_close(Connection *conn) { size_t i; if(conn) { if(conn->db && conn->db->rows) { for(i = 0; i < conn->db->max_rows; i++) { Address *cur = conn->db->rows[i]; free(cur->name); free(cur->email); free(cur); } free(conn->db->rows); } if(conn->file) fclose(conn->file); if(conn->db) free(conn->db); free(conn); } } void Database_delete(Connection *conn, int id) { conn->db->rows[id]->set = 0; } 

你可以在这里找到完整的程序： https ： //github.com/sookoor/Learn-C-the-Hard-Way/blob/master/ex17.c