크리티컬 블록 선정 수정중... 중괄호 기반이 아닌 syntax 기반

@@ -240,126 +240,117 @@ namespace VulnCrawler
             {
             {
                 return null;
                 return null;
             }
             }
-            bool hasIf = false;
-            string prevLine = string.Empty;
             bool mainLine = true; /* 현재 라인이 메인 코드 라인인지 */
             bool mainLine = true; /* 현재 라인이 메인 코드 라인인지 */
-            bool criticalBlock = false; /* 현재 라인이 크리티컬 블록 라인인지 */
+           
             int blockNum = 1; /* 블록 번호 */
             int blockNum = 1; /* 블록 번호 */
             
             
+
+            bool group = false;
+            Queue<string> groupQ = new Queue<string>();
+            var mainQ = new Queue<string>();
+            
             foreach (var line in split)
             foreach (var line in split)
             {
             {
-                bool hasRight = false;
-                bool hasIf2 = false;
-                
-                string trim = line.Trim();
-
-                /* 중괄호 수 세기 */
-                int openBracketCount = trim.Count(c => c == '{');
-                int closeBracketCount = trim.Count(c => c == '}');
+                bool criticalBlock = false; /* 현재 라인이 크리티컬 블록 라인인지 */
 
 
-                int subtract = openBracketCount - closeBracketCount;
-                bracketCount += subtract;
+                string trim = line.Trim();
 
 
-            
-                if (trim.Equals("}"))
+                if (Regex.IsMatch(trim, @"^(if|for|while)"))
                 {
                 {
-                    builder.AppendLine(line);
-                    hasRight = true;
+                    group = true;
+                    mainLine = false;
+                    groupQ.Enqueue(line);
+                    if (trim.EndsWith("{"))
+                    {
+                        group = true;
+                    }
+                    else if (trim.EndsWith("}"))
+                    {
+                        group = false;
+                    }
+                    else if(trim.EndsWith(";"))
+                    {
+                        group = false;
+                    }
+                    continue;
                 }
                 }
-                /* 중괄호 연산 결과 1이라는 것은 메인 라인 */
-                if (bracketCount == 1)
+
+                if (group)
                 {
                 {
-                    if (!hasIf)
+                    groupQ.Enqueue(line);
+                    if (trim.EndsWith("}"))
                     {
                     {
-                        if (Regex.IsMatch(trim, @"^(if|for|while).+\)$"))
-                        {
-                            prevLine = line;
-                            hasIf = true;
-                        }
+                        group = false;
                     }
                     }
-                    else
+                    else if (trim.EndsWith(";"))
                     {
                     {
-                        if (!trim.StartsWith("{"))
-                        {
-                            hasIf2 = true;
-                            builder.AppendLine(line);
-                        }
-                        hasIf = false;
+                        group = false;
                     }
                     }
-                    /* 
-                     * 깊이가 1인데 mainLine이 
-                     * false 이면 넘어왔다는 것이니 현재까지 코드
-                     * blockList에 추가
-                     */
-                    if (!mainLine || hasIf2)
+                    continue;
+                }
+
+                mainQ.Enqueue(line);
+
+                StringBuilder mainBuilder = new StringBuilder();
+                if (!mainLine)
+                {
+                    while(mainQ.Count > 0)
                     {
                     {
-                        string s = builder.ToString();
-                        if (!string.IsNullOrWhiteSpace(s))
+                        string s = mainQ.Dequeue();
+                        if (!criticalBlock)
                         {
                         {
-                            blockList.Add(new Block() { HasCritical = criticalBlock, Code = s, Num = blockNum });
-                            blockNum++;
-                            criticalBlock = false;
-                            builder.Clear();
+                            foreach (var var in ExtractCriticalVariant(s))
+                            {
+                                if (crList.Contains(var))
+                                {
+                                    criticalBlock = true;
+                                    break;
+                                }
+                            }
                         }
                         }
-                        
+                        mainBuilder.AppendLine(s);
+                    }
+                    if (mainBuilder.Length > 0)
+                    {
+                        blockList.Add(new Block { Code = mainBuilder.ToString(), HasCritical = criticalBlock, Num = blockNum++ });
+                        //continue;
                     }
                     }
-                    mainLine = true;
                 }
                 }
-                /* 2 이상이라는 건 메인 라인 X */
-                else if(bracketCount >= 2)
+
+                StringBuilder groupBuilder = new StringBuilder();
+                while (groupQ.Count > 0)
                 {
                 {
-                    /* 
-                     * 깊이가 2 이상인데 mainLine이 
-                     * true면 넘어왔다는 것이니 현재까지 코드
-                     * blockList에 추가
-                     */
-                    if (mainLine)
+                    var s = groupQ.Dequeue();
+                    if (!criticalBlock)
                     {
                     {
-                        string s = builder.ToString();
-                        if (!string.IsNullOrWhiteSpace(s))
+                        foreach (var var in ExtractCriticalVariant(s))
                         {
                         {
-                            blockList.Add(new Block() { HasCritical = criticalBlock, Code = s, Num = blockNum });
-                            blockNum++;
-                            criticalBlock = false;
-                            builder.Clear();
+                            if (crList.Contains(var))
+                            {
+                                criticalBlock = true;
+                                break;
+                            }
                         }
                         }
                     }
                     }
-                    mainLine = false;
+                    groupBuilder.AppendLine(s);
                 }
                 }
-                /* 이도 저도 아니면 그냥 넘어감 */
-                else
+                if (groupBuilder.Length > 0)
                 {
                 {
+                    blockList.Add(new Block { Code = groupBuilder.ToString(), HasCritical = criticalBlock, Num = blockNum++ });
                     continue;
                     continue;
                 }
                 }
-                /* 현재 코드 라인에서 변수 추출시켜서 크리티컬 리스트와 대조 */
-                foreach (var var in ExtractCriticalVariant(line))
-                {
-                    /* 크리티컬 리스트에 추출한 변수가 들어있다면 추가 */
-                    if (criticalList.Contains(var))
-                    {
-                        criticalBlock = true;
-                        break;
-                    }
-                }
 
 
-                if (!hasRight && !hasIf2)
-                {
-                    builder.AppendLine(line);
-                    
-                }
-                
-            }
 
 
-            /* 마지막 남은게 있을 수 있으니 추가 */
-            string fs = builder.ToString();
-            if (!string.IsNullOrWhiteSpace(fs))
-            {
-                blockList.Add(new Block() { HasCritical = criticalBlock, Code = fs, Num = blockNum });
-                blockNum++;
-                criticalBlock = false;
-                builder.Clear();
+
+
+                mainLine = true;
+
+
+
+
             }
             }
 
 
+            Console.WriteLine("끝");
             return blockList;
             return blockList;
         }
         }
     }
     }