VulnWorker.cs 4.96 KB

using LibGit2Sharp;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;

namespace VulnCrawler
{

    public static class VulnWorker
    {
        // 템플릿 메서드 패턴
        public static void Run<T>(string dirPath) where T : VulnAbstractCrawler, new() {
            var self = new T();
            self.Init(dirPath);
            var commits = self.Commits;
            foreach (var commit in commits) {
                // 커밋 메시지
                string message = commit.Message;
                string cve = self.GetCVE(message);
                if (string.IsNullOrEmpty(cve)) {
                    continue;
                }
                foreach (var parent in commit.Parents) {
                    // 부모 커밋과 현재 커밋을 Compare 하여 패치 내역을 가져옴
                    var patch = self.Repository.Diff.Compare<Patch>(parent.Tree, commit.Tree);
                    // 패치 엔트리 파일 배열 중에 파일 확장자가 .py인 것만 가져옴
                    // (실질적인 코드 변경 커밋만 보기 위해서)
                    var entrys = self.GetPatchEntryChanges(patch);
                    // 현재 커밋에 대한 패치 엔트리 배열을 출력함
                    PrintPatchEntrys(entrys, self, message, cve);
                }
            }
        }


        private static void PrintPatchEntrys(IEnumerable<PatchEntryChanges> entrys, VulnAbstractCrawler self, string commitMsg, string cve) {

            foreach (var entry in entrys) {
                // 기존 소스코드
                var oldOid = entry.OldOid;
                Blob oldBlob = self.Repository.Lookup<Blob>(oldOid);
                string oldContent = oldBlob.GetContentText();

                // 변경된 소스코드
                var newOid = entry.Oid;
                Blob newBlob = self.Repository.Lookup<Blob>(newOid);
                string newContent = newBlob.GetContentText();

                var regs = self.GetMatches(entry.Patch);

                #region 패치   코드 출력
                // 패치 전 코드 (oldContent)
                // 패치 후 코드 (newContent)
                // 패치 코드 (entry.Patch)
                // 출력
                //if (regs.Count > 0) {
                //    Console.BackgroundColor = ConsoleColor.DarkBlue;
                //    Console.WriteLine($"Old Content: \n{oldContent}");
                //    Console.ResetColor();

                //    Console.BackgroundColor = ConsoleColor.DarkMagenta;
                //    Console.WriteLine($"New Content: \n{newContent}");
                //    Console.ResetColor();
                //    Console.BackgroundColor = ConsoleColor.DarkRed;
                //    Console.WriteLine($"Patched: \n{entry.Patch}");

                //    Console.ResetColor();
                //    Console.WriteLine("-----------");
                //    Console.WriteLine(regs.Count);

                //}

                // 패치 코드에서 매칭된 파이썬 함수들로부터 
                // 패치 전 코드 파일(oldBlob)을 탐색하여 원본 파이썬 함수 가져오고(originalFunc)
                // 
#endregion

                foreach (var reg in regs) {
                    var match = reg as Match;
                    string methodName = match.Groups[VulnAbstractCrawler.MethodName].Value;

                    string originalFunc, md5;

                    (originalFunc, md5) = self.Process(oldBlob.GetContentStream(),
                        match.Groups[VulnAbstractCrawler.MethodName].Value);

                    #region 현재 패치 엔트리 정보 출력(추가된  , 삭제된  , 패치 이전 경로, 패치  경로)
                    Console.ForegroundColor = ConsoleColor.Blue;
                    Console.WriteLine($"status: {entry.Status.ToString()}");
                    Console.WriteLine($"added: {entry.LinesAdded.ToString()}, deleted: {entry.LinesDeleted.ToString()}");
                    Console.WriteLine($"old path: {entry.OldPath.ToString()}, new path: {entry.Path.ToString()}");
                    Console.ResetColor();


                    Console.Write($"CVE: ");
                    Console.ForegroundColor = ConsoleColor.Red;
                    Console.Write($"{cve}");
                    Console.WriteLine("");
                    Console.ResetColor();
                    Console.ForegroundColor = ConsoleColor.Yellow;
                    Console.WriteLine($"Commit Message: {commitMsg}");
                    Console.ResetColor();

                    // 패치 전 원본 함수
                    Console.WriteLine($"Original Func: {originalFunc}");
                    // 해쉬 후
                    Console.WriteLine($"Original Func MD5: {md5}");
                    Console.WriteLine("==============================");

                    #endregion

                }
            }
        }

    }
}