Антивирус

#include "stdafx.h"
#include "dos.h"
#include "windows.h"
#include <fcntl.h>
#include <sys\types.h>
#include <sys\stat.h>
#include "stdio.h"
#include <stdlib.h>
#include <iostream>
#include <sstream>
#include <direct.h>
#include "io.h"
#define F_FOUND 0
#define PATH_LEN 128
#define DRIVE_LEN 4
#define BLANK_LEN 80
#define BAD 1
#define GOOD 0
#define DBG
int a = -1740;
/* Сигнатура */
char sign[7] = { (char)0xB4,
(char)0x83,
(char)0xCD,
(char)0x21,
(char)0x5E,
(char)0x56,
'\0' };
char
/* Строка імені поточного підкаталога */
path[PATH_LEN],
/* Строка імені початкового знаходження */
old_path[PATH_LEN],
/* Строка імені необхідного пристроя */
drive[DRIVE_LEN],
/* Пуста строка */
blank[BLANK_LEN];
int
/* Кількість відсканованих каталогів */
n_dir,
/* Кількість досліджених файлів */
n_fil,
/* Кількість хворих і вилікуваних файлів */
n_ill;
int
/* Довжина імені файлу */
l,
/* Тимчасовий індекс */
i;
/* Рекурсивна процедура обходу дерева каталогу */
int infected(char *fn)
{
    int f = 0;
    int r, q = 0;
    char buf[7]; /* Буфер під сигнатуру */
                 /* Відкриваємо файл */
    r = _open(fn, O_RDONLY, f);
    if (r) { printf(" – ошибка открытия!"); return GOOD; }
    /* Читаємо 6 байт */
    _lseek(f, a, SEEK_END);
    r = _read(f, buf, 6); buf[6] = '\0';
    if ((r) || (q != 6)) {
        printf(" – ошибка чтения!"); _close(f); return GOOD;
    }
    /* Закриваємо файл */
    _close(f);
    /* Порівнюємо байти з сигнатурою */
    if (strcmp(buf, sign) == 0)
    {
        printf(" – был болен и..."); n_ill++; return BAD;
    } /* Заражений !!! */
    system("pause");
    return GOOD;
}
int cure(char *fn)
{
    int f = 0;
    int mz;
    int r, q = 0;
    char buf[24]; /* Буфер під байти */
                  /* Відкриваємо файл */
    r = _open(fn, O_RDWR, f);
    if (r) { printf(" – ошибка открытия!"); return 0; }
    /* Читаємо перші два байти для визначення типа програми */
    r = _read(f, &mz, 2);
    if ((r) || (q != 2)) { printf(" – ошибка чтения!"); _close(f); return 0; }
    /* Читаємо збережені вірусом 24 байти старого початку */
    int a1 = -80;
    _lseek(f, a1, SEEK_END);
    r = _read(f, buf, 24);
    if ((r) || (q != 24)) { printf(" – ошибка чтения!"); _close(f); return 0; }
    /* Визначаємо тип програми */
    if ((mz == 0x4D5A) || (mz == 0x5A4D))
    { /* Это exe */
      /* Пишемо правильні PartPag и PageCnt */
        _lseek(f, 2, SEEK_SET);
        r = _write(f, &buf[2], 4);
        if ((r) || (q != 4)) { printf(" – ошибка записи!"); _close(f); return 0; }
        /* Пишимо правильні ReloSS и ExeSP */
        _lseek(f, 14, SEEK_SET);
        r = _write(f, &buf[14], 4);
        if ((r) || (q != 4)) { printf(" – ошибка записи!"); _close(f); return 0; }
        /* Пишимо правильні ReloCS и ExeIP */
        _lseek(f, 20, SEEK_SET);
        r = _write(f, &buf[20], 4);
        if ((r) || (q != 4)) { printf(" – ошибка записи!"); _close(f); return 0; }
    }
    else
    { /* Это com */
      /* Відновлюємо збережені 3 байти програми */
        _lseek(f, 0, SEEK_SET);
        r = _write(f, &buf[0], 3);
        if ((r) || (q != 3)) { printf(" – ошибка записи!"); _close(f); return 0; }
    }
    /* Усікаємо файл (переходимо на початок
    вірусу і записуємо 0 байт) */
    _lseek(f, a, SEEK_END);
    r = _write(f, buf, 0);
    /* Закриваємо файл */
    _close(f);
    printf("теперь исцелен!\n");
    system("pause");
    return 0;
}
int walk()
{
    int found_d, found_f;
    _finddata64i32_t buf;
    /* Пошук каталогів */
    found_d = _findfirst(".", &buf);
 
    while (found_d == F_FOUND)
    {
        if ((buf.name[0] != '.') && (buf.attrib & _A_SUBDIR))
        {
            _chdir(buf.name);
            walk();
            _chdir("..");
        }
        found_d = _findnext(found_d, &buf);
    }
    /* До цього моменту не відсканованих нижележачих каталогів
    Більше не залишилося – скануємо файли */
    n_dir++;
    _getcwd(path, PATH_LEN);
    /* Пошук файлів */
    found_f = _findfirst(".", &buf);
    while (found_f == F_FOUND)
    {
        l = strlen(buf.name);
        if (((buf.name[l - 3] == 'C') &&
            (buf.name[l - 2] == 'O') &&
            (buf.name[l - 1] == 'M')) ||
            ((buf.name[l - 3] == 'E') &&
            (buf.name[l - 2] == 'X') &&
                (buf.name[l - 1] == 'E')))
        {
            n_fil++;
            printf("%c%s", 13, blank);
            printf("%c%s\\%s ", 13, path, buf.name);
            /* Знайшли новий файл – необхідно перевірити, інфікований він чи ні.
            Якщо інфікований, то лікуємо */
            if (infected(buf.name) == BAD) cure(buf.name);
        }
        found_f = _findnext(found_f, &buf);
    }
    system("pause");
    return 0;
}
int main(int argc, char *argv[])
{
    setlocale(LC_ALL, "Russian");
    if (argc < 2)
    {
        puts("Введите имя диска в качестве параметра"); exit(2);
    }
    if (((toupper(argv[1][0]))>'Z') || ((toupper(argv[1][0]))<'A'))
    {
        puts("Неверно задано имя диска”"); exit(3);
    }
    drive[0] = argv[1][0]; drive[1] = ':'; drive[3] = '\0';
    n_dir = 0;
    n_fil = 0;
    _getcwd(old_path, PATH_LEN);
    drive[2] = '\0'; system(drive);
    drive[2] = '\\'; _chdir(drive);
    /* Запускаємо рекурсивний обхід дерева каталогів
    Для вибраного диску */
    walk();
    old_path[2] = '0'; system(old_path);
    old_path[2] = '\\'; _chdir(old_path);
    printf("\nКаталогов : %d\nФайлов : %d\nОбнаружено больных и излечено : %d", n_dir, n_fil, n_ill);
    if (n_ill) exit(1); else exit(0);
    system("pause");
    return 0;
}

while (found_d == F_FOUND)

    _finddata64i32_t buf;
    /* Пошук каталогів */
    found_d = _findfirst64i32(".", &buf);

found_d = _findfirst(".\\*.*", &buf);

while (found_d == F_FOUND)
    {
        if ((buf.name[0] != '*.*') && (buf.attrib & _A_SUBDIR))
        {
            _chdir(buf.name);
            walk();
            _chdir(".\\.\\*.*");
        }
        n_dir++;
        found_d = _findnext(found_d, &buf);
        printf("ff = %s",found_d);
    }

printf("ff = %s",found_d);

    while (found_d >= F_FOUND)
    {
        if ((buf.name[0] != '.') && (buf.attrib & _A_SUBDIR))
        {
            _chdir(buf.name);
            walk();
            _chdir("..");
        }
        n_dir++;
        found_d = _findnext(found_d, &buf);
    }