使用函数修改结构内部字符串的内容

时间:2013-10-16 04:13:25

标签: c

我想使用函数修改字符串的内容,该字符串是结构的一部分。问题是当我在函数外面打印字符串时没有输出,但是如果我在函数内部打印它,则输出是FOO,这是正确的输出。我在我认为问题所在的行中添加了评论。

#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define ACCOUNT_NUM_LEN 15
#define NAME_LEN 255
#define PIN_LEN 4
#define BAL_LEN 50

typedef struct account
{
    // account  info
    int account_num[ACCOUNT_NUM_LEN];
    int pin[PIN_LEN];
    float bal;

    // name of the account owner
    char* fname;
    char* lname;

    // link to next account
    struct account *next;
}
account;

account* root;
int num_of_accounts;

bool init(void)
{
    // example
    char account_inf[80] = "FOO|BAZ|123123000012300|1234|5000.00";
    const char delimiter[2] = "|";

    // initialize root and set number of accounts to 0
    root = NULL;
    num_of_accounts = 0;

    // get the first token
    char* token = strtok(account_inf, delimiter);

    // create a new user each line
    account* new_user = malloc(sizeof(account));
    if (new_user == NULL)
        return false;

    // initialize new user info
    new_user->lname = NULL;
    new_user->fname = NULL;
    new_user->next = NULL;

    // walk through other tokens
    int info = 0;
    while (token != NULL)
    {
        // filter info
        if (info == 0)
        {
            new_user->lname = token; // problem
            info++;
        }
        else if (info == 1)
        {
            new_user->fname = token; // problem
            info++;
        }
        else if (info == 2)
        {
            for (int i = 0; i < ACCOUNT_NUM_LEN; i++)
                new_user->account_num[i] = token[i] - '0';
            info++;
        }
        else if (info == 3)
        {
            for (int i = 0; i < PIN_LEN; i++)
                new_user->pin[i] = token[i] - '0';
            info++;
        }
        else if (info == 4)
        {
            new_user->bal = atof(token);
            info++;
        }

        token = strtok(NULL, delimiter);
    }

    root = new_user;
    printf("%s\n", root->lname);
    printf("%s\n", root->fname);
    for (int i = 0; i < ACCOUNT_NUM_LEN; i++)
        printf("%d", root->account_num[i]);
    printf("\n");
    for (int i = 0; i < PIN_LEN; i++)
        printf("%d", root->pin[i]);
    printf("\n");
    printf("%f\n\n", root->bal);

    return true;
}

int main(void)
{
    // load up all accounts. exit if no account is found or made
    if (!init())
        return 1;

    printf("%s\n", root->lname);
    printf("%s\n", root->fname);
    for (int i = 0; i < ACCOUNT_NUM_LEN; i++)
        printf("%d", root->account_num[i]);
    printf("\n");
    for (int i = 0; i < PIN_LEN; i++)
        printf("%d", root->pin[i]);
    printf("\n");
    printf("%f\n\n", root->bal);

    return 0;
}

1 个答案:

答案 0 :(得分:1)

您应该阅读manpage for strtok。该函数将指针返回到原始字符串。由于您在堆栈上分配输入字符串,因此当您从init函数返回时,它将被释放。当您在printf中对main进行后续调用时,将覆盖堆栈上的旧字符串。为了说明这一点,你可以在输入字符串前添加一堆填充,这样就不会被printf调用覆盖:

bool init(void)
{
    // adding this padding should make the program print the desired output
    char padding[1024];
    // example
    char account_inf[80] = "FOO|BAZ|123123000012300|1234|5000.00";
    // . . .

然而,这不是一个真正的修复,因为你仍在使用已经从堆栈中弹出的内存。幸运的是,要真正解决它所需要做的就是堆分配输入字符串:

const char INPUT_STR[] = "FOO|BAZ|123123000012300|1234|5000.00";

bool init(void)
{
    // example
    char * account_inf = malloc(sizeof(INPUT_STR));
    strcpy(account_inf, INPUT_STR);
    // . . .

或者,您可以将字符串放在静态内存中(即关闭const INPUT_STR并使用它),但如果您这样做,请记住strtok将修改存储在静态内存中的原始字符串,以便以后再也不能再使用它。