Option Explicit
Enum EnuCT
    AllFile = 0
    EntryPoint = 1
    StartFile = 2
End Enum
Type VirusData
    CheckType As EnuCT
    Signature As String
    Length As Long
    Name As String
End Type
Public VirusDatas() As VirusData
Public VirusCount As Long
Public Declare Function GetTickCount Lib "kernel32" () As Long
Public Declare Sub CopyMemory Lib "kernel32" Alias "RtlMoveMemory" (Destination As Any, Source As Any, ByVal Length As Long)

Public Function CheckVirus(FileName As String) As Boolean
    Dim i As Long
    Dim T As EnuCT
    Dim sT As Long
    If InStr(1, FileName, "Qurantine") Then Exit Function
    If InStr(1, FileName, "Samples") Then Exit Function
    Dim HashObj As New cHash
    sT = GetTickCount()
    For i = 0 To VirusCount - 1
        T = VirusDatas(i).CheckType
        Select Case T
        Case AllFile
            If VirusDatas(i).Signature = HashObj.HashFile(FileName) Then \'GetHash(ByteArr, , , i * 2) & GetHash2(ByteArr, , , (i * 2) + 1) Then
                CheckVirus = True
                Exit Function
            End If
        Case EntryPoint
        Case StartFile
        End Select
    Next
End Function

Public Function AddSignature(FileName As String)
    Dim HashObj As New cHash
    ReDim Preserve VirusDatas(VirusCount)
    VirusDatas(VirusCount).CheckType = AllFile
    VirusDatas(VirusCount).Signature = HashObj.HashFile(FileName)
    VirusDatas(VirusCount).Name = Mid(FileName, InStrRev(FileName, "\") + 1)
    VirusDatas(VirusCount).Name = Replace(Left(VirusDatas(VirusCount).Name, Len(VirusDatas(VirusCount).Name)), " ", "-")
    VirusCount = VirusCount + 1
End Function

Public Function Save()
    Dim FN As Long
    FN = FreeFile
    Open App.Path & "\\Virusdb.ini" For Output As #FN
        Print #FN, VirusCount
        Dim i As Long
        For i = 0 To VirusCount - 1
            Print #FN, VirusDatas(i).Name & " " & VirusDatas(i).CheckType & " " & VirusDatas(i).Signature
        Next
    Close #FN
End Function

Public Function Init()
    Dim FN As Long, i As Long
    FN = FreeFile
    On Error GoTo UDB
    Open "Virusdb.ini" For Input As #1
    Input #1, VirusCount
    ReDim VirusDatas(VirusCount)
    For i = 0 To VirusCount - 1
        Input #1, VirusDatas(i).Name, VirusDatas(i).CheckType, VirusDatas(i).Signature
    Next
    Close #1
UDB:
    UpdateDB
End Function

Public Function UpdateDB()
    Dim tHand As Long, wfd As WIN32_FIND_DATA, tPath As String, Files As String, i As Long
    tHand = FindFirstFile(App.Path & "\\samples\\*.vir", wfd)
    On Error Resume Next
    Do
        tPath = TrimNull(wfd.cFileName)
        Files = Files & tPath & "\"
        wfd.cFileName = ""
    Loop While FindNextFile(tHand, wfd)
    For i = 0 To UBound(Split(Files, "\")) - 1
        tPath = Split(Files, "\")(i)
        If CheckVirus(App.Path & "\\samples\" & tPath) = False Then
            AddSignature App.Path & "\\samples\" & tPath
        End If
    Next
    Save
End Function

Public Function CheckFile(ByVal FileName As String)
    Dim vf As frmVirusFound
    If CheckVirus(FileName) Then
        frmMain.lstStatus.AddItem "바이러스:" & FileName, 0
        frmMain.lblCaught = Val(frmMain.lblCaught) + 1
        frmMain.lblChecked = Val(frmMain.lblChecked) + 1
        Set vf = New frmVirusFound
        vf.Warn FileName
        vf.Show
    Else
        frmMain.lstStatus.AddItem "정상:" & FileName, 0
        frmMain.lblChecked = Val(frmMain.lblChecked) + 1
    End If
    frmMain.lstStatus.TopIndex = 0
End Function

Public Function CheckMemory()

End Function