<?php

[0] Fix | Delete

[1] Fix | Delete

namespace CookieAdmin;

[2] Fix | Delete

[3] Fix | Delete

if(!defined('COOKIEADMIN_VERSION') || !defined('ABSPATH')){

[4] Fix | Delete

die('Hacking Attempt');

[5] Fix | Delete

}

[6] Fix | Delete

[7] Fix | Delete

class CookieCategorizer {

[8] Fix | Delete

[9] Fix | Delete

/**

[10] Fix | Delete

* Takes the raw output from the scanner and enriches it with data from the knowledge base.

[11] Fix | Delete

*

[12] Fix | Delete

* @param array $found_cookies The array of cookies from the Scanner class.

[13] Fix | Delete

* @return array The enriched list of cookies.

[14] Fix | Delete

*/

[15] Fix | Delete

static function categorize_cookies($found_cookies, $categorized_cookies) {

[16] Fix | Delete

if (empty($found_cookies)) {

[17] Fix | Delete

return [];

[18] Fix | Delete

}

[19] Fix | Delete

[20] Fix | Delete

$prefixes = self::generate_query_prefixes(array_keys($found_cookies));

[21] Fix | Delete

[22] Fix | Delete

if (empty($prefixes)) {

[23] Fix | Delete

return $found_cookies;

[24] Fix | Delete

}

[25] Fix | Delete

[26] Fix | Delete

$db_candidates = self::fetch_candidates_from_db($prefixes);

[27] Fix | Delete

[28] Fix | Delete

if (empty($db_candidates)) {

[29] Fix | Delete

return $found_cookies;

[30] Fix | Delete

}

[31] Fix | Delete

[32] Fix | Delete

return self::match_and_enrich_cookies($found_cookies, $db_candidates, $categorized_cookies);

[33] Fix | Delete

}

[34] Fix | Delete

[35] Fix | Delete

[36] Fix | Delete

static function generate_query_prefixes($cookie_names){

[37] Fix | Delete

$prefixes = [];

[38] Fix | Delete

foreach ($cookie_names as $name) {

[39] Fix | Delete

[40] Fix | Delete

$parts = explode('_', $name, 2);

[41] Fix | Delete

$prefix = $parts[0];

[42] Fix | Delete

[43] Fix | Delete

// Add a simple rule to avoid tiny, useless prefixes like "a" or "b"

[44] Fix | Delete

if (strlen($prefix) >= 3) {

[45] Fix | Delete

$prefixes[] = $prefix;

[46] Fix | Delete

} else {

[47] Fix | Delete

// If prefix is too short, use the full name up to a limit

[48] Fix | Delete

$prefixes[] = substr($name, 0, 3);

[49] Fix | Delete

}

[50] Fix | Delete

}

[51] Fix | Delete

return array_unique($prefixes);

[52] Fix | Delete

}

[53] Fix | Delete

[54] Fix | Delete

[55] Fix | Delete

static function fetch_candidates_from_db($prefixes){

[56] Fix | Delete

[57] Fix | Delete

if (empty($prefixes)) {

[58] Fix | Delete

return [];

[59] Fix | Delete

}

[60] Fix | Delete

[61] Fix | Delete

$cookies_info = cookieadmin_load_cookies_csv($prefixes, 1);

[62] Fix | Delete

[63] Fix | Delete

if(empty($cookies_info) || is_wp_error($cookies_info)){

[64] Fix | Delete

return [];

[65] Fix | Delete

}

[66] Fix | Delete

[67] Fix | Delete

return $cookies_info;

[68] Fix | Delete

}

[69] Fix | Delete

[70] Fix | Delete

[71] Fix | Delete

static function match_and_enrich_cookies($found_cookies, $db_candidates, $old_categorized_cookies){

[72] Fix | Delete

$categorized_cookies = [];

[73] Fix | Delete

$remove_cookies = [];

[74] Fix | Delete

[75] Fix | Delete

foreach ($found_cookies as $scanned_name => $cookie_data) {

[76] Fix | Delete

$best_match = null;

[77] Fix | Delete

$longest_match_length = 0;

[78] Fix | Delete

[79] Fix | Delete

foreach ($db_candidates as $candidate) {

[80] Fix | Delete

[81] Fix | Delete

$candidate_name = $candidate['cookie_name'];

[82] Fix | Delete

[83] Fix | Delete

// Rule 1: Wildcard Match

[84] Fix | Delete

if ($candidate['wildcard'] == 1 && strpos($scanned_name, $candidate_name) === 0) {

[85] Fix | Delete

// "Best" is defined as the longest matching prefix.

[86] Fix | Delete

if (strlen($candidate_name) > $longest_match_length) {

[87] Fix | Delete

$longest_match_length = strlen($candidate_name);

[88] Fix | Delete

$best_match = $candidate;

[89] Fix | Delete

}

[90] Fix | Delete

}

[91] Fix | Delete

// Rule 2: Exact Match

[92] Fix | Delete

else if ($candidate['wildcard'] == 0 && $scanned_name === $candidate_name) {

[93] Fix | Delete

if (strlen($candidate_name) > $longest_match_length) {

[94] Fix | Delete

$longest_match_length = strlen($candidate_name);

[95] Fix | Delete

$best_match = $candidate;

[96] Fix | Delete

break;

[97] Fix | Delete

}

[98] Fix | Delete

}

[99] Fix | Delete

}

[100] Fix | Delete

[101] Fix | Delete

if (!empty($best_match)) {

[102] Fix | Delete

[103] Fix | Delete

$cookie_name = $best_match['cookie_name'];

[104] Fix | Delete

[105] Fix | Delete

if(!empty($old_categorized_cookies) && array_key_exists($cookie_name, $old_categorized_cookies)){

[106] Fix | Delete

$remove_cookies[] = $old_categorized_cookies[$cookie_name];

[107] Fix | Delete

}

[108] Fix | Delete

[109] Fix | Delete

$categorized_cookies[$cookie_name]['cookie_name'] = $cookie_name;

[110] Fix | Delete

$categorized_cookies[$cookie_name]['raw_name'] = $scanned_name;

[111] Fix | Delete

$categorized_cookies[$cookie_name]['category'] = $best_match['category'];

[112] Fix | Delete

$categorized_cookies[$cookie_name]['description'] = $best_match['description'];

[113] Fix | Delete

$categorized_cookies[$cookie_name]['patterns'] = $best_match['patterns'];

[114] Fix | Delete

[115] Fix | Delete

}

[116] Fix | Delete

}

[117] Fix | Delete

$categorized_cookies['remove_cookies'] = $remove_cookies;

[118] Fix | Delete

return $categorized_cookies;

[119] Fix | Delete

}

[120] Fix | Delete

}

[121] Fix | Delete