Added cookie retrieval

Selenium_Clean/Form1.cs

@@ -10,31 +10,88 @@ using System.Windows.Forms;
 
 using OpenQA.Selenium;
 using OpenQA.Selenium.Firefox;
+using System.IO;
+using System.Diagnostics;
+using System.Reflection;
 
 namespace Selenium_Clean
 {
     public partial class Form1 : Form
     {
+        public const string COOKIE_HOST = ".formula1.com";
         public Form1()
         {
             InitializeComponent();
         }
-
-        private void btnStart_Click(object sender, EventArgs e)
+        private void StartCookieRecovering()
         {
+            string scriptPath = Path.Combine(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location), "recoverCookiesCSV.py");
+            Process process = new Process();
+            process.StartInfo.FileName = "python.exe";
+            process.StartInfo.Arguments = scriptPath;
+            process.StartInfo.UseShellExecute = false;
+            process.StartInfo.RedirectStandardOutput = true;
+            process.Start();
+            string output = process.StandardOutput.ReadToEnd();
+            process.WaitForExit();
+        }
+        public string GetCookie(string host, string name)
+        {
+            StartCookieRecovering();
+            string value = "";
+            List<Cookie> cookies = new List<Cookie>();
+            using (var reader = new StreamReader("cookies.csv"))
+            {
+                // Read the header row and validate column order
+                string header = reader.ReadLine();
+                string[] expectedColumns = { "host_key", "name", "value", "path", "expires_utc", "is_secure", "is_httponly" };
+                string[] actualColumns = header.Split(',');
+                for (int i = 0; i < expectedColumns.Length; i++)
+                {
+                    if (expectedColumns[i] != actualColumns[i])
+                    {
+                        throw new InvalidOperationException($"Expected column '{expectedColumns[i]}' at index {i} but found '{actualColumns[i]}'");
+                    }
+                }
+
+                // Read each data row and parse values into a Cookie object
+                while (!reader.EndOfStream)
+                {
+                    string line = reader.ReadLine();
+                    string[] fields = line.Split(',');
+
+                    string hostname = fields[0];
+                    string cookieName = fields[1];
+
+                    if (hostname == host && cookieName == name)
+                    {
+                        value = fields[2];
+                    }
+                }
+            }
+
+            return value;
+        }
+        private void btnStart_Click(object sender, EventArgs e)
+        { 
+            string loginCookieValue = GetCookie(COOKIE_HOST, "login");
+            string loginSessionValue = GetCookie(COOKIE_HOST, "login-session");
 
-            
-            //var service = FirefoxDriverService.CreateDefaultService(@"C:\Users\Moi\Downloads\geckodriver-v0.27.0-win32\geckodriver.exe");
             var service = FirefoxDriverService.CreateDefaultService(AppDomain.CurrentDomain.BaseDirectory+@"geckodriver-v0.27.0-win32\geckodriver.exe");
             service.Host = "127.0.0.1";
             service.Port = 5555;
 
             FirefoxOptions options = new FirefoxOptions();
+            options.AddArgument("--window-size=1920,1080");
             options.AddArgument("--disable-headless");
+            options.AddArgument("--disable-web-security");
+            options.AddArgument("--same-site-none-secure");
+            options.AcceptInsecureCertificates = true;
 
-            var driver = new FirefoxDriver(service,options);
-
-            driver.Navigate().GoToUrl("https://www.youtube.com/watch?v=dQw4w9WgXcQ&autoplay=1&mute=1");
+            using (var driver = new FirefoxDriver(service,options))
+            {
+                driver.Navigate().GoToUrl("https://www.youtube.com/watch?v=dQw4w9WgXcQ&autoplay=1&mute=1");
+            }
         }
     }
 }

Selenium_Clean/recoverCookiesCSV.py

@@ -0,0 +1,83 @@
+# Rohmer Maxime
+# RecoverCookies.py
+# Little script that recovers the cookies stored in the chrome sqlite database and then decrypts them using the key stored in the chrome files
+# This script has been created to be used by an other programm or for the data to not be used directly. This is why it stores all the decoded cookies in a csv. (Btw could be smart for the end programm to delete the csv after using it)
+# Parts of this cript have been created with the help of ChatGPT
+
+import os
+import json
+import base64
+import sqlite3
+import win32crypt
+from Cryptodome.Cipher import AES
+from pathlib import Path
+import csv
+
+def get_master_key():
+    with open(
+        os.getenv("localappdata") + "\\Google\\Chrome\\User Data\\Local State", "r"
+    ) as f:
+        local_state = f.read()
+        local_state = json.loads(local_state)
+    master_key = base64.b64decode(local_state["os_crypt"]["encrypted_key"])
+    master_key = master_key[5:]  # removing DPAPI
+    master_key = win32crypt.CryptUnprotectData(master_key, None, None, None, 0)[1]
+    print("MASTER KEY :")
+    print(master_key)
+    print(len(master_key))
+    return master_key
+
+def decrypt_payload(cipher, payload):
+    return cipher.decrypt(payload)
+
+def generate_cipher(aes_key, iv):
+    return AES.new(aes_key, AES.MODE_GCM, iv)
+
+def decrypt_password(buff, master_key):
+    try:
+        iv = buff[3:15]
+        payload = buff[15:]
+        cipher = generate_cipher(master_key, iv)
+        decrypted_pass = decrypt_payload(cipher, payload)
+        decrypted_pass = decrypted_pass[:-16].decode()  # remove suffix bytes
+        return decrypted_pass
+    except Exception:
+        # print("Probably saved password from Chrome version older than v80\n")
+        # print(str(e))
+        return "Chrome < 80"
+
+
+master_key = get_master_key()
+
+cookies_path = Path(
+    os.getenv("localappdata") + "\\Google\\Chrome\\User Data\\Default\\Network\\Cookies"
+)
+
+if not cookies_path.exists():
+    raise ValueError("Cookies file not found")
+
+with sqlite3.connect(cookies_path) as connection:
+    connection.row_factory = sqlite3.Row
+    cursor = connection.cursor()
+    cursor.execute("SELECT * FROM cookies")
+    
+    with open('cookies.csv', 'a', newline='') as csvfile:
+        fieldnames = ['host_key', 'name', 'value', 'path', 'expires_utc', 'is_secure', 'is_httponly']
+        writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+
+        if csvfile.tell() == 0:
+            writer.writeheader()
+
+        for row in cursor.fetchall():
+            decrypted_value = decrypt_password(row["encrypted_value"], master_key)
+            writer.writerow({
+                'host_key': row["host_key"],
+                'name': row["name"],
+                'value': decrypted_value,
+                'path': row["path"],
+                'expires_utc': row["expires_utc"],
+                'is_secure': row["is_secure"],
+                'is_httponly': row["is_httponly"]
+            })
+            
+print("Finished CSV")