Функция работает только один раз - C - PullRequest
Новая
       13

Функция работает только один раз - C

3 голосов
/ 13 марта 2012

Я пытался создать базовый файл заголовка XOR для использования в некоторых будущих программах.До сих пор я получил почти все для работы, но я не могу использовать одну и ту же функцию дважды.Если я вызываю функцию для шифрования строки, она работает, но затем, если я вызываю ее снова, происходит сбой.Я не знаю, делаю ли я что-то неправильно с памятью или упускаю что-то очевидное.Надеюсь, что кто-то может указать на недостаток в этом, потому что я не могу найти что-то неправильное.

Редактировать: Если публикация этого слишком много, не стесняйтесь обрезать код.Я уже немного потрудился, поэтому я не просто вставляю свой проект и надеюсь, что кто-то исправит его. 

// Main.c
#define MAX_LENGTH 255
#define KEY_SIZE 8
int main(int argc, char *argv[]) {
    //Get String to XOR
    char *input = malloc (MAX_LENGTH);
    printf("Enter a string to encrypt: ");
    fgets(input, MAX_LENGTH, stdin);

    if(input[strlen (input) - 1] == '\n') {
        input[strlen (input) - 1] = '\0';
    }

    //Create a random key
    char *pass = _create_key(KEY_SIZE);
    int len = strlen (input);
    printf("Length of key is %d\n", KEY_SIZE);
    printf("Entered String: %s - Password: %s\n", input, pass);

    //Encrypt works fine
    char *encrypted = malloc (sizeof (input));
    _xor_str_s(input, pass, len, encrypted);
    printf("Encrypted String: %s\n", encrypted);

    char *decrypted = malloc (sizeof (input));
    //Crashes here
    _xor_str_s(encrypted, pass, len, decrypted);
    printf("Decrypted String: %s\n", decrypted);
    return 0;
}

//Header File Function
void _xor_str_s(char *str, char *pass, int len, char *out) {
    int i = 0;
    for(i = 0; i < len; i++) {
        *(out + i) = str[i] ^ pass[i % strlen (pass)];
    }
    *(out + i) = 0;
}

char * _create_key(int len) {
    len = !len ? 16 : len;
    char *ret = (char *)malloc (len);
    unsigned int _GLOBAL_SEED_ = (unsigned int)time(NULL);
    srand (_GLOBAL_SEED_);
    int i = 0;
    for(i = 0; i < len; i++) {
        ret[i] = (char)(rand() + 1); //+1 avoids NULL
    }
    ret[i] = '\0';
    return ret;
}

1 Ответ

11 голосов
/ 13 марта 2012 
char *encrypted = malloc (sizeof (input));

, вероятно, проблема, поскольку это всегда будет sizeof(char *). Я думаю, что вы хотите 

char *encrypted = malloc (strlen (input) + 1);
...