# -*- coding: utf-8 -*-

# Filename      main.py

import psutil
from ctypes import *
from ctypes.wintypes import *
from struct import *

OpenProcess = windll.kernel32.OpenProcess
ReadProcessMemory = windll.kernel32.ReadProcessMemory
GetLastError = windll.kernel32.GetLastError
CloseHandle = windll.kernel32.CloseHandle
PROCESS_ALL_ACCESS = 0x1F0FFF

class TrackManiaTimer(object):
    def __init__(self):
        self.pid = None
        self.process_name = "ManiaPlanet.exe"
        self.offsets = [0x37C, 0x3C, None]
        self.time = None
        self.timelast = None
        self.address = 0x0163B4D8

    def get_pid(self):
        for proc in psutil.process_iter():
            try:
                if (proc.name() == self.process_name):
                    self.pid = proc.pid
            except (PermissionError, psutil.AccessDenied):
                pass

    def read_process_memory(self):
        processHandle = OpenProcess(PROCESS_ALL_ACCESS, False, self.pid)
        length = 4
        data = ""
        read_buff = create_string_buffer(length)
        count = c_ulong(0)
        current_address = self.address
        for offset in self.offsets:
            if not ReadProcessMemory(processHandle, current_address, cast(read_buff, LPVOID), length, byref(count)):
                #print("[*] Error Code: 0x%08x" % GetLastError())
                #print(read_buff.raw)
                return -1
            else:
                val = read_buff.value
                result = int.from_bytes(val, byteorder='little')
                if(offset != None):
                    current_address = result+offset
                else:
                    return result



def main():
    t = TrackManiaTimer()
    t.get_pid()
    old_timer = 0
    while(True):
        new_timer = t.read_process_memory()
        if new_timer != old_timer:
            print(new_timer)
            old_timer = new_timer



if __name__ == '__main__':
    main()