我正在通过 Learn C The Hard Way 重温我的 C 技能,目前我在 17th Exercise .
我正在做“额外学分”部分。使该页面上给出的数据库代码得到 我正在努力
“更改代码以接受 MAX_DATA 和 MAX_ROWS 参数,将它们存储在数据库结构中,并将其写入文件,从而创建一个可以任意大小的数据库”
因此,我注释掉了#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 < MAX_ROWS; i++) {
struct Address addr = {.id = i, .set = 0};
conn->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; i<MAX_ROWS; i++) {
struct Address *row = conn->db->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 中遇到错误,内存正在泄漏。我不明白上面代码中的错误。 然而,主要问题似乎已经解决了:)
最佳答案
我修改了 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
关于c - 指向另一个结构中的一个结构的指针,从文件中写入和读取它会产生 SegFault,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/14950241/