一. 简介

之前在介绍如何用C#创建VBA类库时,我们聊过怎么把.NET程序封装成COM组件。但有个关键问题——VBA没法直接引用.dll文件,必须先把它注册到系统里,VBA才能识别并调用。

由于VBA依赖Excel,而Excel有32位和64位两种版本,注册时就得分别处理:

RegAsm.exe是.NET框架自带的工具,一般路径都固定。为了脚本通用,最好同时注册32位和64位,这样无论用户用哪种版本的Excel都能正常工作。

注册成功后,RegAsm.exe会把.dll的相关信息写入注册表。效果如下图所示(图中可以看到ProgId和对应的GUID):

Powershell 注册.dll类库到系统的操作方法

二. 程序代码

2.1 管理员权限执行脚本

调用RegAsm.exe必须用管理员身份运行,否则会失败。所以脚本里需要加一段权限判断——如果当前用户不是管理员,就自动提权重新启动。

param(
    [string]$originalScriptFolderPath
)
# 校验当前用户是否有管理员权限
function Test-IsAdmin {
    $id = [System.Security.Principal.WindowsIdentity]::GetCurrent()
    $principal = New-Object System.Security.Principal.WindowsPrincipal($id)
    if ($principal.IsInRole([System.Security.Principal.WindowsBuiltInRole]::Administrator)) {
        return $true
    }
    return $false
}
# 若没有管理员权限
if (-not (Test-IsAdmin)) {
    Write-Host "未检测到管理员权限,正在尝试重新启动为管理员模式..." -ForegroundColor Yellow
    # 暂停1秒钟, 显示文字
    Start-Sleep -Seconds 1
    # 以管理员方式启动当前脚本
    Start-Process -FilePath "powershell.exe" `
                  -ArgumentList "-NoProfile -ExecutionPolicy RemoteSigned -File $($PSCommandPath) -originalScriptFolderPath $($PSScriptRoot)" `
                  -Verb RunAs
    # 退出当前脚本, 防止打开多个窗口
    exit
}
Write-Host "已成功以管理员权限运行PowerShell ...`n" -ForegroundColor Green
# 当以管理员权限运行脚本的时候, 作业目录默认会移动到【C:\WINDOWS\system32】下
# 我们手动将作业目录其移动到脚本所在的文件夹路径
if ($null -ne $originalScriptFolderPath) {
    Set-Location $originalScriptFolderPath
}

2.2 自定义函数

注册COM时需要两个核心参数,先理清楚它们的用途:

function Register-Com {
    param (
        [string]$dllFullPath,
        [string]$regAsmExeFullPath
    )
    $versionMsg = "32位"
    if ($regAsmExeFullPath.Contains('Framework64')) {
       $versionMsg = "64位"
    }
    $params = "`"$dllFullPath`" /codebase /tlb"
    $result = $null
    try {
        $result = Start-Process $regAsmExeFullPath -ArgumentList $params -Wait -PassThru
        if ($result.ExitCode -eq 0) {
            Write-Host "${versionMsg}的dll注册成功"
        }
    } 
    catch {
        Write-Host "${versionMsg}的dll注册失败: $($result.ExitCode)"
        Write-Host "${versionMsg}的dll注册失败: $($_.Exception.Message)"
    }
}
# 卸载Com
function Unregister-Com {
    param (
        [string]$dllFullPath,
        [string]$regAsmExeFullPath
    )
    $versionMsg = "32位"
    if ($regAsmExeFullPath.Contains('Framework64')) {
       $versionMsg = "64位"
    }
     # 卸载Com所需的参数
    $params = "`"$dllFullPath`" /unregister"
    $result = $null
    try {
        $result = Start-Process $regAsmExeFullPath -ArgumentList $params -Wait -PassThru
        if ($result.ExitCode -eq 0) {
            Write-Host "${versionMsg}的dll卸载成功"
        }
    } 
    catch {
        Write-Host "${versionMsg}的dll卸载失败: $($result.ExitCode)"
        Write-Host "${versionMsg}的dll卸载失败: $($_.Exception.Message)"
    }
}

验证COM是否存在时有个细节:32位PowerShell只能验证32位COM,64位PowerShell只能验证64位COM。现在大多数电脑都是64位系统,默认PowerShell也是64位的,所以验证64位直接调用即可;但验证32位时,需要手动指定32位PowerShell的路径。

function Test-32ComBoth {
    param (
        [string]$progId
    )
    # 32位Powershell的路径
    $powershell32 = "$env:WINDIR\SysWOW64\WindowsPowerShell\v1.0\powershell.exe"
    $script = @"
try {
    New-Object -ComObject '$progId' | Out-Null
    Write-Host '${progId}的32位COM存在'
} catch {
    Write-Host '${progId}的32位COM不存在'
}
"@
    # 测试32位com
    & $powershell32 -NoProfile -Command $script
}
function Test-64ComBoth {
    param (
        [string]$progId
    )
    # 测试64位com
    try {
        New-Object -ComObject $progId | Out-Null
        Write-Host "${progId}的64位COM存在"
    }
    catch {
        Write-Host "${progId}的64位COM不存在"
    }
}

2.3 注册代码

下面是完整的主逻辑:先定位.dll文件和RegAsm.exe路径,然后让用户选择安装还是卸载。两步都分别调用32位和64位的注册/卸载函数,最后验证COM是否存在,并检查注册表项。

# 要安装的.dll库的路径, 此处设定为和脚本同一级别的目录下
[string]$dllPath = ".\03-ToolLib.dll"
# .dll库的ProgId
$ProgId = "ToolLib.ToolCom"
# 获取32位和64位的RegAsm.exe的路径, 由于不知道具体版本, 所以使用了通配符
$regAsm32ExePath = "$env:WINDIR\Microsoft.NET\Framework\v4.0.*\RegAsm.exe"
$regAsm64ExePath = "$env:WINDIR\Microsoft.NET\Framework64\v4.0.*\RegAsm.exe"
# 获取RegAsm.exe的绝对路径
$regAsm32ExeFullPath = (Get-ChildItem "$regAsm32ExePath" -ErrorAction SilentlyContinue | Select-Object -First 1).FullName
$regAsm64ExeFullPath = (Get-ChildItem "$regAsm64ExePath" -ErrorAction SilentlyContinue | Select-Object -First 1).FullName
# 如果32位和64位的RegAsm.exe有任意一个不存在的话
if ((-not $regAsm32ExeFullPath) -or (-not $regAsm64ExeFullPath)) {
    Write-Host "找不到RegAsm.exe文件..."
    Pause
    exit 1
}
# 通过相对路径获取绝对路径
$fullPathObj = Resolve-Path $dllPath -ErrorAction SilentlyContinue
if (-not $fullPathObj) {
    Write-Host "找不到${dllPath}文件..."
    Pause
    exit 1
}
# 获取dll文件的绝对路径
$dllFullPath = $fullPathObj.Path
# 让用户选择要执行的操作
[string]$confirm = Read-Host "选择要执行的操作. 1. 安装依赖库 2. 卸载依赖库"
if ($confirm -eq "2") {
    # 卸载32位的com
    Unregister-Com -dllFullPath $dllFullPath -regAsmExeFullPath $regAsm32ExePath
    # 卸载64位的com
    Unregister-Com -dllFullPath $dllFullPath -regAsmExeFullPath $regAsm64ExePath
    Pause
    exit
} elseif ($confirm -ne "1") {
    Write-Host "请输入 1 或者 2"
    Pause
    exit
}
# 注册32位的com
Register-Com -dllFullPath $dllFullPath -regAsmExeFullPath $regAsm32ExePath
# 注册64位的com
Register-Com -dllFullPath $dllFullPath -regAsmExeFullPath $regAsm64ExePath
# .dll类库中的命名空间和方法名
Test-32ComBoth "$ProgId"
Test-64ComBoth "$ProgId"
# 根据 ProgId 查询注册表路径是否存在
$RegistryProgIdPath = "Registry::HKEY_CLASSES_ROOT\$ProgId\"
if (-not (Test-Path $RegistryProgIdPath)) {
    Write-Host "找不到${RegistryProgIdPath}注册表..."
    Pause
    exit 1
}
# 获取 .dll库的Guid
# {6143FB0B-9C17-4859-860C-6DA4A466ECD1}
$dllCLSID = (Get-Item "$RegistryProgIdPath\CLSID").GetValue("")
# 查询32位.dll对应的注册表
Get-ChildItem "Registry::HKLM\Software\Classes\WOW6432Node\CLSID\$dllCLSID" -ErrorAction SilentlyContinue
# 查询64位.dll对应的注册表
Get-ChildItem "Registry::HKEY_CLASSES_ROOT\CLSID\$dllCLSID" -ErrorAction SilentlyContinue
Pause

三. 注册成功

Powershell 注册.dll类库到系统的操作方法

本文转载于:https://www.jb51.net/jiaoben/362635pn0.htm 如有侵犯,请联系zhengruancom@outlook.com删除。
免责声明:正软商城发布此文仅为传递信息,不代表正软商城认同其观点或证实其描述。