C - Почему мой код ломается при удалении объявления неиспользуемой переменной - PullRequest
Новая
       9

C - Почему мой код ломается при удалении объявления неиспользуемой переменной

0 голосов
/ 05 февраля 2020

Я пишу программу на C для восстановления изображений из необработанного файла для CS50, и у меня странная проблема. У меня есть переменная int cnt, которую я использовал для целей отладки, и я заставил программу работать, поэтому я удалял оставшийся отладочный код. Но когда я удаляю объявление cnt, я начинаю выводить поврежденные файлы. Перед удалением строки 25 ниже я выводил файлы .jpg, которые я мог открыть и просматривать, затем я удалил строку, перекомпилировал, удалил фотографии из последнего запуска и перезапустил программу с теми же данными .raw и новыми файлами, которые я получил были не признаны. Поэтому я поместил декларацию обратно, перекомпилировал, удалил старые фотографии, снова запустил программу и получил хорошие файлы. Кто-нибудь знает, почему удаление неиспользованной декларации портит мои результаты? Оскорбительная декларация находится в строке 25. 

  #include <stdio.h>
  #include <stdlib.h>
  #include <stdint.h>

  int main(int argc, char *argv[])
  {
      if (argc != 2)
      {
          printf("Usage: ./recover image\n");
         return 1;
     }

     int filesFound = 0;

     FILE *inFile = fopen(argv[1], "r");
     FILE *outFile = NULL;

     if (inFile == NULL)
     {
         printf("Image file could not be opened\n");
         return 1;
     }

     uint8_t buffer[512];
     int cnt = 0;

     while (!feof(inFile))
     {
         fread(buffer, 512, 1, inFile);

         // check for start of jpg file
         if (buffer[0] == 0xff && buffer[1] == 0xd8 && buffer[2] == 0xff && (buffer[3] & 0xf0) == 0xe0)
         {
             // start of jpg was found
             if (outFile != NULL)
             {
                 // close the current file and then open a new file to write to
                 fclose(outFile);
                 outFile = NULL;
             }

             // open a file to write to
            char fName[4];
             sprintf(fName, "%03i.jpg", filesFound);
             outFile = fopen(fName, "w");

             filesFound++;
         }

         if (outFile != NULL){
             // we have found data to write and opened a file
             fwrite(buffer, 512, 1, outFile);
         }
     }

     //Be sure to close my files
     fclose(inFile);

     if (outFile != NULL)
     {
         fclose(outFile);
     }

     return 0;
 }

1 Ответ

0 голосов
/ 05 февраля 2020

char fName[4] недостаточно места для имени, сгенерированного "%03i.jpg", поэтому вы переполняете буфер. Увеличьте его и используйте snprintf, а не sprintf, и проверьте возвращаемое значение для обнаружения ошибок: 

int result = snprintf(fName, sizeof fName, "%03i.jpg", filesFound);
if (sizeof fName <= result)
{
    fprintf(stderr, "Internal error, buffer is too small for file name.\n");
    exit(EXIT_FAILURE);
}

Вместо того, чтобы печатать ошибку, вместо этого можно использовать возвращаемое значение snprintf, которая указывает необходимую длину, чтобы выделить память для большего буфера, а затем повторить snprintf с этим буфером.

(Обратите внимание, что snprintf может вернуть отрицательный результат в случае ошибки. Обычно это станет большим числом при преобразовании в size_t для сравнения, поэтому оно вызовет это сообщение об ошибке. Однако в надежной программе может потребоваться вставить отдельный тест для result < 0.)

...