跳到主要內容

Windows Batch - 修改環境變數

Introduction

操作環境變數是常遇到的議題。像安裝軟體時,為了要讓使用者可以在command line直接執行我們的程式,我們會將程式資料夾路徑加入到PATH變數中;移除軟體時就會把它從PATH中拿掉。或者我們有某些如軟體安裝目錄的資訊,希望能夠讓我們不同的程式得知,而將這個值建立到環境變數中以方便操作。接下來將針對操作環境變數的不同方法做介紹。

Methods

registry

輸入regedit打開註冊檔編輯器。修改目前使用者環境變數:

HKEY_CURRENT_USER\Environment

修改系統環境變數:

KEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\Session Manager\Environment

Command line:

REM 設定環境變數TEST為%PATH%的內容
reg add "HKEY_CURRENT_USER/Environment" /v TEST /t REG_EXPAND_SZ /d "%PATH%" /f
REM 設定環境變數TEST為%PATH%
reg add "HKEY_CURRENT_USER/Environment" /v TEST /t REG_EXPAND_SZ /d ^%PATH^% /f
REM 刪除環境變數TEST
reg delete "HKEY_CURRENT_USER/Environment" /v TEST /f

修改後不會立即生效。

setx.exe

微軟提供修改環境變數的程式。可以直接參考命令說明: link。修改完後效果與直接修改系統>進階系統設定>系統環境變數的效果相同。

wmic

操作WMI介面的Command line工具。根據網路上人家所說使用這個方法可以讓各個正在執行的程式立即取到最新的環境變數,但經過我測試僅WinXP可以,其餘是不行的,除非也是透過wmi去取得環境變數內容。(測試過XP, 2003, 2008, Win7, 2012)

Create

建立變數%VAR%值%VALUE%到系統環境變數中:

wmic ENVIRONMENT create name="%VAR%",username="<system>",VariableValue="%VALUE%" 

建立變數到目前使用者環境變數中:

wmic ENVIRONMENT create name="%VAR%",username="%username%",VariableValue="%VALUE%" 

如果使用者是有網域的情況,username要改為:

username="%userdomain%\%username%"

Update

將所有使用者名稱為%VAR%的環境變數修改為%VALUE%:

wmic ENVIRONMENT where "name='%VAR%'" set VariableValue="%VALUE%"

Delete

刪除所有使用者名稱為%VAR%的環境變數:

wmic ENVIRONMENT where "name='%VAR%'" delete

刪除系統名稱為%VAR%的環境變數:

wmic ENVIRONMENT where "name='%VAR%' and username='<system>'" delete

Query

取得名稱為%VAR%的環境變數,顯示所有使用者與所有欄位:

wmic ENVIRONMENT where "name='%VAR%'"

取得系統名稱為%VAR%的環境變數,僅顯示VariableValue欄位:

wmic ENVIRONMENT where "name='%VAR%' and username='<system>'" get VariableValue

Changed Notification

由於registry與wmic的方法並不像setx的效果,會去通知其它程式環境變數的改變。因此可以透過底下Windows API達到通知的效果,當然程式也是要重啟的。

#include <windows.h>
#include <iostream>
using namespace std;
 
int main(int argc, char* argv[])
{
	LPDWORD MsgResult = 0;
	SendMessageTimeout(HWND_BROADCAST, WM_SETTINGCHANGE, 0,
		LPARAM("Environment"), SMTO_ABORTIFHUNG, 5000, MsgResult);
	cout  << MsgResult << endl;
	return 0;
}

別人也有提供taskkill explorer.exe在重新啟動的方法,但這樣做似乎…僅能用於測試上。

Reference

留言

這個網誌中的熱門文章

解決RobotFramework從3.1.2升級到3.2.2之後,Choose File突然會整個Hand住的問題

考慮到自動測試環境的維護,我們很久以前就使用java去執行robot framework。前陣子開始處理從3.1.2升級到3.2.2的事情,主要先把明確的runtime語法錯誤與deprecate item處理好,這部分內容可以參考: link 。 直到最近才發現,透過SeleniumLibrary執行Choose File去上傳檔案的動作,會導致測試案例timeout。本篇文章主要分享心路歷程與解決方法,我也送了一條issue給robot framework: link 。 我的環境如下: RobotFramework: 3.2.2 Selenium: 3.141.0 SeleniumLibrary: 3.3.1 Remote Selenium Version: selenium-server-standalone-3.141.59 首先並非所有Choose File的動作都會hang住,有些測試案例是可以執行的,但是上傳一個作業系統ISO檔案一定會發生問題。後來我透過wireshark去比對新舊版本的上傳動作,因為我使用 Remote Selenium ,所以Selenium會先把檔案透過REST API發送到Remote Selenium Server上。從下圖我們可以發現,在3.2.2的最後一個TCP封包,比3.1.2大概少了500個bytes。 於是就開始了我trace code之路。包含SeleniumLibrary產生要送給Remote Selenium Server的request內容,還有HTTP Content-Length的計算,我都確認過沒有問題。 最後發現問題是出在socket API的使用上,就是下圖的這支code: 最後發現可能因為開始使用nio的方式送資料,但沒處理到尚未送完的資料內容,而導致發生問題。加一個loop去做計算就可以解決了。 最後我有把解法提供給robot framework官方,在他們出新的版本之前,我是將改完的_socket.py放在我們自己的Lib底下,好讓我們測試可以正常進行。(shutil.py應該也是為了解某個bug而產生的樣子..)

第一次寫MIB就上手

SNMP(Simple Network Management Protocol)是用來管理網路設備的一種Protocol,我對它的認識也是從工作接觸開始。雖說是管理網路設備,但是主機、電源供應器、RAID等也都可以透過它來做管理。如果你做了一個應用程式,當然所有的操作也都可以透過SNMP來完成,不過可能會很痛苦。前陣子遇到一個學弟,它告訴我說:「我可能不會想寫程式。」為什麼? 因為這是他痛苦的根源。 在這篇文章中,不是要告訴你SNMP是什麼,會看這篇文章的大哥們,應該已經對SNMP有些認識了。 是的!主題是MIB(Management information base)! 對於一個3th-party的SNMP oid,有MIB可以幫助你去了解它所提供的資訊是什麼,且可以對它做什麼操作。最近我運氣很好剛好做到關於修改MIB的工作,也讓我順便了解一下它的語法,接下來我要交給大家MIB的基礎認識。 smidump 我並非使用什麼高強的Editor去編寫MIB,我僅透過Nodepad++編輯和smidump編譯而已。smidump是Kay教我使用的一個將MIB module轉成樹狀結構或oid列表的工具,唯一的缺點是不會告訴你哪一行打錯。當然有錢直接買編輯樹狀結構的工具就可以不需要了解語法了! 安裝 在Ubuntu上可先輸入smidump確認是否安裝,如果沒安裝可透過apt-get install libsmi2ldbl安裝。(CentOS可以透過yum install libsmi) root@tonylin:~/multi-boot-server# smidump The program 'smidump' is currently not installed. You can install it by typing: apt-get install libsmi2ldbl 使用 透過下面兩行指令,就可以將mib file產生出對應的tree與oid列表的檔案。也可以透過這個結果確認MIB是不是你想要的。 smidump -f tree example1.mib > xtree.txt smidump -f identifiers example1.mib > xiden.txt 如果有參考其它檔案要加上p的參數: smidum...

Windows DLL - 32-bit dll with 64-bit driver

前言 在64-bit作業系統上所使用的驅動程式,一定是64-bit,然而應用程式卻可能是32或64-bit。當32-bit應用程式傳值給驅動程式時是有可能會發生溢位的。主要原因是32與64-bit指標所佔用記憶體長度的不同。接下來我將透過Reference 1中的程式PhyMem,來告訴大家問題在哪與如何修改。(最後改完的程式碼恕我不提供) PhyMem介紹 這是一個中國人寫的程式。作用與WinIO相同,可以存取windows的io port與physical memory。這個程式包含pmdll、driver與test三個專案,分別產生dll、sys與exe。作者僅提供32-bit的版本,但只要修改編譯設定就可以讓它產生出64-bit的artifact。但如同我前言所說,如果你是32-bit的dll要存取64的sys該怎麼辦? 可以規定User在64-bit的OS用64-bit的應用程式就好了阿! 但是將一個32-bit應用程式改為64-bit的有這麼簡單嗎? 除此之外,中間傳遞的資料型態也會影響到正常功能。 設定driver專案 我開發環境是VC2008與DDK6000,下載Reference1的專案是無法直接編譯的,可以參考我的設定去修改編譯、連結參數。輸出檔可以看個人需求,根據32或64命名,或者是用同一個檔案名稱。 32-bit C/C++ > 一般 > 其它Include目錄($(DDKROOT)為設定於環境變數的DDK安裝目錄): $(DDKROOT)\inc\ddk";"$(DDKROOT)\inc\api";"$(DDKROOT)\inc\crt";"$(DDKROOT)\inc\crt\gl";"$(DDKROOT)\inc\crt\sys" 連結器 > 一般 > 其它程式庫目錄: $(DDKROOT)\lib\wnet\i386。 連結器 > 資訊清單檔: 將產生資訊清單與UAC選擇否,因為這不適用於driver。 連結器 > 進階 > 隨機化的基底位置: 選擇預設。 64-bit Reference1僅提供32-bit設定。64-bit可在建置>組態管理員中,新增x64平台,而設定檔可從win32複製過來修...