alert_attack_on_UTM_based_on_json_.js

/********************************************************

// alert_attack_on_UTM_based_on_json v1.4

★★★★★★ Made by MaMu0228 ★★★★★★

※설명※
UTM 로그인 후, 관제 룰을 적용한 URL로 fetch를 보내 받아온 JSON의 데이터를 분석해서
공격이라 판단되는 패킷이 있으면 노래를 재생하는 프로그램

▣사용법▣
1. 변경하는 곳을 자기 UTM 설정에 맞춰 변경하기
2. 이 파일 전체 코드를 복사하고, console에 붙여넣기
3. 이후 올라오는 창을 통해 알람으로 쓸 노래 선택하기


********************************************************/


//↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓ 변경하는 곳 ↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓

// UTM의 관제 룰을 적는 공간의 id를 넣으면 됩니다.
let queryStringInput = document.querySelector('#queryString');
let queryStringValue = queryStringInput.value;

// 자동으로 UTM 룰을 가져와 JSON을 보내는 변수, URL대신 사용하시면 됩니다.
let ruleURL = 'https://IP:PORT/log/total/totalList?json=true&take=200&skip=0&page=1&pageSize=200&queryString=' + encodeURIComponent(queryStringValue) + '&grid=true&searchType=0&searchFlag=true&dateRangeSelect=3600';

// 위 'ruleURL'이 작동을 안할 경우, UTM 로그인 이후 룰을 적용한 다음 'F12 -> 네트워크'를 들어가서 JSON을 가져오는 URL을 적습니다.
let = URL = 'https://UTM-URL-AFTER-SET-RULE'

// UTM에서 공격이라 판단할 sip(source ip)와 dip(destination ip)를 적습니다.
let attackArray = ['at.ta.ck.ip', '123.23.234.123'];

// 공격이라 판단할 패킷에 적용된 정책 번호를 적습니다.
let attackPolicy = ['number'];

// 한 sip가 몇 개의 서로 다른 dip를 가질 때 공격이라 간주할 지 정하는 개수입니다. 
let NETWORK_SCAN_NUMBER = 10;

// 하나의 sip가 10개 이상의 다른 dport로 패킷이 올 때 포트 스캔으로 간주하는 개수입니다.
let PORT_SCAN_NUMBER =10;

// 네트워크 스캔, 포트 스캔을 json이 몇 개 이상일 때 실행할지 정하는 숫자입니다.
// ex) 1시간 내 패킷이 10개만 들어왔다면, 공격이 왔는지 체크 X
let CHECK_NUMBER = 15;

// 공격이라 간주한 sip가 알람을 울리고나서, 언제 다시 알람을 울릴지 정하는 쿠키, 초 단위입니다.
let COOKIE_TIME =  4000;


//↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑ 변경하는 곳 ↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑


// 공격 발생시 만들 오디오 객체 변수
let audio

////////////////////// CSS ///////////////////////////////////
// 공격이 있을 때 화면에 띄울려고 선언해둔 전역 변수
let eventLog = '';
// 공격이 왔을 때 문구를 띄울 공간용 요소
let warningDiv = document.createElement('div');

// 클릭을 했을 때 띄울 문자열
warningDiv.addEventListener("click", function(){
  warningDiv.style.display = 'none';
  warningDiv.style.textContent = "공격이 의심됩니다. ";
});

////////// 공격일 올 시 화면을 붉게 만드는 요소 //////
warningDiv.id = 'warning';
warningDiv.style.display = 'none';
warningDiv.style.position = 'fixed';
warningDiv.style.top = '0';
warningDiv.style.left = '0';
warningDiv.style.width = '100%';
warningDiv.style.height = '100%';
warningDiv.style.backgroundColor = 'rgba(255, 0, 0, 0.3)';

////////// 공격이 올 시 관련 정보를 띄우는 요소 //////
let warningText = document.createElement('p');
warningText.style.postion = 'absolute';
warningText.style.top = '50%';
warningText.style.left = '50%';
warningText.style.transform = 'translate(-50%, -50%)';
warningText.style.color = 'black';
warningText.style.fontsize = '30px';

warningDiv.appendChild(warningText);

let body = document.getElementsByTagName('body')[0];
body.appendChild(warningDiv);



///////////////// 핵심 함수 //////////////////////

function getDataFromURL() {
    let sipArray = []; // json에서 중복을 제거한 sip를 모아둔 배열
    let dipArray = []; // json에서 중복을 제거한 dip를 모아둔 배열
    let policyArray = []; // json에서 중복을 제거한 정책 번호를 모아둔 배열


///// fetch로 받아온 데이터를 html --> innerText --> jSON 변경 과정 ///////
    fetch(ruleURL)
        .then(response => response.text()) // text 가져오기
        .then(data => {
            const parser = new DOMParser(); // text를 DOM으로 변경
            const htmlDoc = parser.parseFromString(data, 'text/html'); // html로 변경
            const elements = htmlDoc.body.innerText; // html안에 있는 body의 innerText를 가져오기

            if (typeof elements === 'undefined'){
                console.log("elements가 비어있습니다. ");
            }

            const objects = JSON.parse(elements); // 가져온 innerText를 json으로 변경
            const objectList = Object.values(objects); // json에 있는 값들을 

            objectList[1] = objectList[1].reversed(); // 로그가 들어있는 배열은 objectList[1]부터이고, 처음 json에 들어있는 데이터가 시간 역순이라서, 시간 순서에 맞게 하기 위해 거꾸로 만들기
            
/////////////// attackPolicy 배열에 들어있는 IP가 있는 지 확인하는 부분 ///////////
            
            let matchingPolicy = []; // 맨 위에 정의했던 attackPolicy[]에 들어있는 '정책 번호 문자열'이 json에 있을 경우 저장할 용도인 배열

            policyArray = objectList[1].map(obj => obj.type).filter(type => type !== undefined); // 
		
///////////////// 중복된 요소들을 제거하기 위해 Set 자료형으로 바꿨다가 다시 배열로 바꾸기 ////////////////
            var set = new Set(policyArray); 
            policyArray = [...set];

/////// attackPolicy배열에 들어있는 값과 json에서 중복을 제거하고 가져온 policyArray배열을 비교 후, 같은 '정책 번호 문자열'
/////// 이 있으면 matchingPolicy 배열에 추가
            matchingPolicy.push(...policyArray.filter(value => attackPolicy.includes(value)));  

            for (let i =0; i <= matchingPolicy.length ; i++){
                if (typeof matchingPolicy[i] !== 'undefined' && !checkCookie("attack_정책_" + matchingPolicy[i])){
                    playMusicFromDB(1);
                    setCookie("attack_정책_" + matchingPolicy[i], matchingPolicy[i], COOKIE_TIME);
                    console.log("!!!!! 정책_" + matchingPolicy[i] + " 에 해당하는 공격이 감지 됐습니다. !!!");
                    console.log("attack_정책_" + matchingPolicy[i] + "쿠키를 만들었습니다.");


                    // 첫 공격 시간 출력
                    for (let i = 0; i <= objectList[1].length; i++){
                      if (objectList[1][i].type === matchingPolicy[i]){

                        eventLog = ` 첫 공격 시간은 : ${objectList[1][i].datetimeText} 입니다.
                        목적지 주소는 : ${objectList[1][i].dip} 입니다.
                        출발지 포트는 : ${objectList[1][i].sport} 입니다.
                        목적지 포트는 : ${objectList[1][i].dport} 입니다. 
                        
                        `;

                        console.log(eventLog);
                        // 공격 로그 indexedDB에 저장
                        saveAttackLogToDB(objectList[1][i]);

                        break;
                      }
                    }

                    turnBackgroundRed();
                  
                }
            }

////////////// sipArray배열과 json.sip 주소를 가져와 비교해 처리하는 부분 ///////////////////

            sipArray = objectList[i].map(obj => obj.sip).filter(sip => sip !== undefined);

            var set = new Set(sipArray);
            sipArray = [...set];

            let matchingIpArray = [];

            matchingIpArray = sipArray.filter(value => attackArray.includes(value));

////////////// dipArray배열과 json.dip 주소를 가져와 비교해 처리하는 부분 ///////////////////            
	    
	     
            dipArray = objectList[1].map(obj => obj.dip).filter(dip => dip !== undefined);

            var set = new Set(dipArray);
            dipArray = [...set];

            matchingIpArray.push(...dipArray.filter(value => attackArray.includes(value)));

////////////// attackArray 배열에 들어있는 주소가 있을 경우, 쿠키 생성 및 알람을 울리는 부분 ///////////////////                 

            for (let i =0; i <= matchingIpArray.length ; i++){
              
                if (typeof matchingIpArray[i] !== 'undefined' && !checkCookie("attack_" + matchingIpArray[i])){
                  
                    playMusicFromDB(1);
                  
                    setCookie("attack_" + matchingIpArray[i], matchingIpArray[i], COOKIE_TIME);
                  
                    console.log("!!!!! " + matchingIpArray[i] + " 에 해당하는 공격이 감지 됐습니다. !!!");
                    console.log("attack_" + matchingIpArray[i] + "쿠키를 만들었습니다.");

                    // 첫 공격 시간 출력
                    for (let i = 0; i <= objectList[1].length; i++){
                        if (objectList[1][i].type === matchingIpArray[i]){

                        eventLog = ` 첫 공격 시간은 : ${objectList[1][i].datetimeText} 입니다.
                        목적지 주소는 : ${objectList[1][i].dip} 입니다.
                        출발지 포트는 : ${objectList[1][i].sport} 입니다.
                        목적지 포트는 : ${objectList[1][i].dport} 입니다. 
                        
                        `;

                        console.log(eventLog);
                        // 공격 로그 indexedDB에 저장
                        saveAttackLogToDB(objectList[1][i]);

                        break;
                        }
                    }
                  turnBackgroundRed();
                }
            }
///////////////////////////////////// 네트워크 스캔 공격이 있는지 체크하는 부분///////////////////////////////
          
            checkNetworkScan(objectList[1], sipArray);



////////////////////////////////////// 포트 스캔 공격이 있는지 체크하는 부분 ////////////////////////////////                  

          
            checkPortScan(objectList[1], sipArray);

////////////////////////////////////// 메모리 샤용량 체크 /////////////////////////////////////

            checkMemory();
                  
            console.log(" alert_attack_on_UTM_based_on_json is on working ");
            
        })
        .catch(error => {
            console.error("getDataFromURL 실행중 에러가 발생: ", error);
        })
}

//###########################
/* 
id를 받아 indexedDB에 저장된 음악 데이터를 다시 가져와 실행하는 함수
*/
//###########################
function playMusicFromDB(id) {
  const transaction = db.transaction([MUSIC_STORE], 'readonly');
  const objectStore = transaction.objectStore(MUSIC_STORE);
  // id 인수에 담김 숫자로 오브젝트 스토어에서 데이터를 가져와 request에 넣음
  const request = objectStore.get(id);

  // request라는 변수에 오브젝트를 불러오는 코드가 실패해 onerror 이벤트 발생시 실행
  request.onerror = function(event) {
    console.error('음악 파일 불러오기 오류:', event.target.errorCode);
  };
  // request라는 변수에 오브젝트를 불러오는 코드가 성공해 onsuccess 이벤트 발생시 실행
  request.onsuccess = function(event) {
    const musicBlob = event.target.result.music;
    const audioURL = window.URL.createObjectURL(musicBlob);
    // window.URL.createObjectURL()도 지원 안 하는 브라우저일 경우 밑 코드 사용
    // const audioURL = window.webkitURL.createObjectURL(musicBlob);

    var audio = new Audio(audioURL);
    audio.controls = true;
    audio.play(); // 음악 파일 재생

    document.body.appendChild(audio);

    console.log('음악 파일 재생');
  };
};  


//#############################
/* 
노래를 업로드 시, 이진 파일로 변경하는 함수
*/
//#############################
function uploadMusic(event) {
  // 파일을 file에 저장
  const file = event.target.files[0];
  // FileReader() 객체를 만듦
  const reader = new FileReader();
  // FileReader 객체에 데이터가 담기는 이벤트 발생시, event.target.result(ArrayBuffer객체)를 blob객체로 만듦
  reader.onload = function(event) {
    const musicBlob = new Blob([event.target.result], { type: file.type });
    // 만든 Blob 객체를 
    saveMusicToDB(musicBlob);
  };
  // reader.onload 이벤트를 발생시키기 위한 코드로, 이진데이터 파일을 바이트 단위(ArrayBuffer)로 읽음
  reader.readAsArrayBuffer(file);
}

//############################
/* 
indexedDB에 음악 파일 저장
*/
//############################
function saveMusicToDB(musicBlob) {
  // music_store라는 오브젝트 스토어를 readwrite로 지정하고
  const transaction = db.transaction([MUSIC_STORE], 'readwrite');
  // 오브젝트 스토어를 열고
  const objectStore = transaction.objectStore(MUSIC_STORE);
  // 받은 musicBlob을 저장함
  const request = objectStore.put({ id:1,  music: musicBlob });

  request.onerror = function(event) {
    console.error('음악 파일 저장 오류:', event.target.errorCode);
  };

  request.onsuccess = function(event) {
    console.log('음악 파일 저장 완료');
    // ****** 노래 업로드 후 자동으로 10초마다 실행되는 부분 ******
    setInterval(getDataFromUrl, 10000);
    console.log('정상적으로 alert_attack_on_UTM_based_on_json이 실행되고 있습니다. ★ 스피커가 켜져있는 지 꼭 확인하세요! ★');
  };
}

//#################################
/*
json을 받으면, indexedDB에 저장하는 함수
*/
//#################################
function saveAttackLogToDB(json) {
  const transaction = db.transaction([ATTACK_LOG_STORE], 'readwrite');
  
  const objectStore = transaction.objectStore(ATTACK_LOG_STORE);
  
  const request = objectStore.add({
    time: json.datetimeText, sip: json.sip, sport: json.sport,
    dip: json.dip, dport: json.dport, service: json.service
  });

  request.onerror = function(event){
    console.error(`공격 로그를 indexedDB에 올리는 데 실패했습니다.
    F12 -> Application -> indexedDB를 초기화 후 다시 전체 코드를 실행해 주십쇼`, event.target.errorCode);
  };

  request.onsuccess = function(event) {
    console.log(`공격 로그를 indexedDB에 올리기 성공했습니다. `, event.target.result);
  };

};



//#################################
/*
쿠키 만드는 초 단위로 만드는 함수
*/
//#################################
function setCookie(cookie_name, cookie_value, seconds) {
  var d = new Date();
  d.setTime(d.getTime() + (seconds * 1000));
  var expires = "expires="+ d.toUTCString();
  document.cookie = cookie_name + "=" + cookie_value + ";" + expires + ";path=/";
}

//################################
/* 
이름으로 받은 쿠기가 있는지, 전체 쿠키들을 뒤져서 확인하는 함수
*/
//################################
function checkCookie(name) {
  var cookies = document.cookie.split(";");
  for (var i = 0; i < cookies.length; i++) {
    var cookie = cookies[i];
    while (cookie.charAt(0) == " ") {
      cookie = cookie.substring(1);
    }
    if (cookie.indexOf(name) == 0) {
      return true;
    }
  }
  return false;
}

//#########################################
/*
jsonArray와 jsonArray에서 중복을 제거한 sipArray를 인수로 받아, 10개 이상 다른 dip를 가진
sip를 공격이라 간주하고 노래를 트는 함수
*/
//#########################################
function checkNetworkScan(jsonArray, sipArray) {

  if (jsonArray.length >= CHECK_NUMBER) {


    // sipArray에 들어있는 sip들을 객체로 모두 초기화하기
    let attackObjs = sipArray.reduce((acc, cur) => ({...acc, [cur]: {}}), {});

    const attackCounts = {};


    for (const {sip, dip} of jsonArray) {

      if (attackObjs.hasOwnProperty(sip)) {
        attackCounts[sip] = (attackCounts[sip] || 0) + 1;

        if (!attackObjs[sip].hasOwnProperty(dip)) {
          attackObjs[sip][dip] = {};

          if (Object.keys(attackObjs[sip]).length >= NETWORK_SCAN_NUMBER && !checkCookie("attack_" + sip)) {
            playMusicFromDB(1);
            setCookie("attack_" + sip, sip, COOKIE_TIME);
            console.log(sip + " 주소로부터 네트워크 스캔 공격이 의심됩니다. ");

            // 첫 공격 시간 출력
            for (let i = 0; i <= jsonArray.length; i++) {
              if (jsonArray[i].type === matchingIpArray[i]) {

                eventLog = `
    			첫 공격 시간은 : ${jsonArray[i].datetimeText} 입니다.
              	목적지 주소는 : ${jsonArray[i].dip} 입니다.
              	출발지 포트는 : ${jsonArray[i].sport} 입니다.
                목적지 포트는 : ${jsonArray[i].dport} 입니다. 
                        
                `;

                console.log(eventLog);
                // 공격 로그 indexedDB에 저장
                saveAttackLogToDB(jsonArray[i]);

                break;
              }
            }

            turnBackgroundRed();
          }
        }
      }
    }
  }
};





//#########################################
/*
하나의 sip가 여러개의 dport를 가지는 체크해, 포트스캔 공격이 있는지 체크하는 함수
*/
//#########################################
function checkPortScan(jsonArray, sipArray) {

  if (jsonArray.length >= CHECK_NUMBER) {


    // sipArray에 들어있는 sip들을 객체로 모두 초기화하기
    let attackObjs = sipArray.reduce((acc, cur) => ({...acc, [cur]: {}}), {});

    const attackCounts = {};
    const dportCounts = {};

    for (const {sip,dport} of jsonArray) {

      if (attackObjs.hasOwnProperty(sip)) {
        attackCounts[sip] = (attackCounts[sip] || 0) + 1;
        dportCounts[sip] = dportCounts[sip] || new Set();
        dportCounts[sip].add(dport);


        if (!attackObjs[sip].hasOwnProperty(dport)) {
          attackObjs[sip][dport] = {};

          if (dportCounts[sip].size >= PORT_SCAN_NUMBER && !checkCookie("attack_" + sip)) {
            playMusicFromDB(1);
            setCookie("attack_" + sip, sip, COOKIE_TIME);
            console.log(sip + " 주소로부터 포트 스캔 공격이 의심됩니다. ");

            // 첫 공격 시간 출력
            for (let i = 0; i <= jsonArray.length; i++) {
              if (jsonArray[i].type === matchingIpArray[i]) {

                eventLog = ` 
				첫 공격 시간은 : ${jsonArray[i].datetimeText} 입니다.
              	목적지 주소는 : ${jsonArray[i].dip} 입니다.
              	출발지 포트는 : ${jsonArray[i].sport} 입니다.
              	목적지 포트는 : ${jsonArray[i].dport} 입니다. 
                        
                `;

                console.log(eventLog);
                // 공격 로그 indexedDB에 저장
                saveAttackLogToDB(jsonArray[i]);

                break;
              }
            }

            turnBackgroundRed();
          }
        }
      }
    }
  }
};




//#################################
/*
메모리를 체크해, 브라우저의 메모리 한계에 도달해 꺼지기 직전 알람을 울리는 함수
*/
//#################################
function checkMemory() {

  if (window.performance && window.performance.memory){
    let memoryInfo = window.performance.memory;
    let usedHeapOfTotal = (memoryInfo.usedJSHeapSize / memoryInfo.totalJSHeapSize) * 100;
    let usedHeapOfLimit = (memoryInfo.usedJSHeapSize / memoryInfo.jsHeapSizeLimit) * 100;

    if (usedHeapOfTotal > 99){
      console.log(`
      현재 페이지에서 사용중인 메모리가 한계치의 99%이상에 도달했습니다.
      현재 페이지 메모리 사용량 : ${usedHeapOfTotal.toFixed(2)}
      ★ 메모리 사용량이 100%가 되면 페이지가 갑자기 로그아웃 될 수 있습니다. 
      컴퓨터를 **재부팅** 하시길 추천드립니다. ★
      *** 경고의 의미로 노래를 재생합니다. ***
      `);
      playMusicFromDB(1);
    }

    if (usedHeapOfLimit > 50){
      console.log(`
      현재 브라우저에 할당된 메모리가 한계치의 50%이상에 도달했습니다.
      현재 브라우저 메모리 사용량 : ${usedHeapOfLimit.toFixed(2)}
      ★ 메모리 사용량이 100%가 되면 페이지가 갑자기 로그아웃 될 수 있습니다. 
      컴퓨터를 **재부팅** 하시길 추천드립니다. ★
      *** 경고의 의미로 노래를 재생합니다. ***
      `);
      playMusicFromDB(1);
    }
    
  } else{
    console.log(" window.performance는 현재 브라우저에서 지원하지 않는 기능입니다. ");
  }
  
}




//#################################
/*
공격이 올 시 화면을 붉게 만드는 함수
*/
//#################################
function turnBackgroundRed() {
  document.getElementById("warning").style.display = "black";
}



/*#######################################
            실행 코드들
########################################*/

// indexedDB 생성
const DB_NAME = 'music_db';
const DB_VERSION = 1;
const MUSIC_STORE = 'music_store';
const ATTACK_LOG_STORE = 'attack_log_store';

let db;

const request = window.indexedDB.open(DB_NAME, DB_VERSION);

request.onerror = function(event) {
  console.error('indexedDB 생성 오류:', event.target.errorCode);
};

request.onsuccess = function(event) {
  db = event.target.result;
  console.log('indexedDB 생성 완료');
};

request.onupgradeneeded = function(event) {
  const db = event.target.result;
  const objectStore = db.createObjectStore(MUSIC_STORE, { keyPath: 'id', autoIncrement: true });
  const attackLog = db.createObjectStore(ATTACK_LOG_STORE, {
    keyPath:'id', autoIncrement:true});
  
  console.log('object store 생성 및 업데이트 완료');
};

const fileInput = document.createElement('input');
fileInput.type = 'file';

// 파일 선택 창 띄우기
fileInput.click();

// 노래 업로드 시 이벤트 실행
fileInput.addEventListener('change', uploadMusic);