快捷導(dǎo)航

基于Python制作一個(gè)解壓的內(nèi)存加速球

更新時(shí)間：2023年10月07日 15:00:11 作者：銀空飛羽

安全管家助手什么的上總會(huì)帶一個(gè)內(nèi)存加速球,有關(guān)掉進(jìn)程以及內(nèi)存清理的功能,本文就來(lái)利用Python制作一個(gè)解壓的內(nèi)存加速球,有需要的小伙伴可以參考下

寫(xiě)在前面

安全管家助手什么的上總會(huì)帶一個(gè)內(nèi)存加速球，有關(guān)掉進(jìn)程以及內(nèi)存清理的功能。雖然本質(zhì)上，內(nèi)存清理只是個(gè)幌子實(shí)際沒(méi)啥作用，但是當(dāng)你深度清理之后，肉眼可見(jiàn)的內(nèi)存下降確實(shí)很解壓。不過(guò)每次都得打開(kāi)安全軟件才能享受解壓的快感實(shí)在是有些買(mǎi)櫝還珠了，所以就想嘗試一下自己寫(xiě)個(gè)玩玩。

內(nèi)存清理

這里我使用python來(lái)編寫(xiě)。

搜索了一下，我確定了兩個(gè)方案，一個(gè)是通過(guò)windows自帶的內(nèi)存回收機(jī)制清理，另一個(gè)是通過(guò)windows提供的一個(gè)APIEmptyWorkingSet清理，下面我們就來(lái)對(duì)比一下。

windows自帶的內(nèi)存回收機(jī)制

通過(guò)申請(qǐng)大量的內(nèi)存來(lái)觸發(fā)windows的內(nèi)存回收機(jī)制，從而清理內(nèi)存。

import psutil
import ctypes
mem = psutil.virtual_memory()
size = mem.total // 2
buffer = ctypes.create_string_buffer(size)
del buffer

這里通過(guò)psutil獲取到內(nèi)存總數(shù)，并且通過(guò)ctypes庫(kù)來(lái)創(chuàng)建了一個(gè)一半總數(shù)的緩沖區(qū)，隨后立馬刪掉。

但是，這樣會(huì)造成臨時(shí)的卡頓，并且肉眼可見(jiàn)的內(nèi)存上漲。而且清理的不干凈。果斷放棄這個(gè)方案，一點(diǎn)都不順暢。

EmptyWorkingSet

EmptyWorkingSet是windows的一個(gè)API，它用來(lái)清空指定進(jìn)程的內(nèi)存中的工作集。將該程序所用的物理內(nèi)存定義為臟頁(yè)并轉(zhuǎn)移到磁盤(pán)上的內(nèi)存交換文件上，清空該進(jìn)程占用的物理內(nèi)存，當(dāng)進(jìn)程再次需要時(shí)會(huì)再次從外存進(jìn)行加載。那么加載的過(guò)程中可能會(huì)導(dǎo)致臨時(shí)的卡頓。

市面上的加速球大多用的都應(yīng)該是這個(gè)技術(shù)

通過(guò)psutil庫(kù)，以及搭配win32相關(guān)的庫(kù)，實(shí)現(xiàn)起來(lái)也很方便

import win32api
import win32con
import win32process
import psutil
for pid in psutil.pids():
    try:
        handle =  win32api.OpenProcess(win32con.PROCESS_ALL_ACCESS, False, pid)
        win32process.SetProcessWorkingSetSize(handle, -1, -1)
        win32api.EmptyWorkingSet(handle)
    except Exception:
        continue

而經(jīng)過(guò)與某管家進(jìn)行對(duì)比，發(fā)現(xiàn)效果大差不大，他們可能做了一些內(nèi)存整理的優(yōu)化。

進(jìn)程終止

殺死進(jìn)程就更簡(jiǎn)單了，這里為了避免出問(wèn)題，我只保留他們的父進(jìn)程，依舊是通過(guò)psutil來(lái)實(shí)現(xiàn)

import psutil
pid_set = set()
processes = []
# 只遍歷正在運(yùn)行的進(jìn)程
for pid in psutil.pids():
    if pid in pid_set:
        continue
    try:
        proc = psutil.Process(pid)
    except psutil.NoSuchProcess:
        continue
    # 只處理父進(jìn)程
    proc_p = proc.parent()
    if proc_p is not None:
        proc = proc_p
        if proc.pid in pid_set:
            continue
    pid_set.add(proc.pid)
    # 生成進(jìn)程信息
    process_info = proc.as_dict(attrs=['pid', 'name', 'username', 'status', 'memory_info'])
    process.append(process_info)

而殺死進(jìn)程則是通過(guò)如下代碼實(shí)現(xiàn)，pid就是指定進(jìn)程的pid