powershell script to test if a USB is attached

Hi,

I found this under the Client API post.  Would anyone know how to convert the 'LIST' section into function that can reused?

Thanks

$pipeClient = new-object System.IO.Pipes.NamedPipeClientStream("vhclient")
$pipeClient.Connect()
$pipeClient.ReadMode = [System.IO.Pipes.PipeTransmissionMode]::Message
$writer = new-object System.IO.StreamWriter($pipeClient)
$writer.AutoFlush = $true
$writer.Write("LIST")
$reader = new-object System.IO.StreamReader($pipeClient)
$bytes = New-Object System.Collections.Generic.List[byte]
while ($reader.Peek() -ne -1)
{
  $bytes.Add($reader.Read())
}
$result = [System.Text.Encoding]::UTF8.GetString($bytes)

This is what I eventually got to work.

# Possible scenarios
# Token JC (myusbdrive.XX) - USB token available
# Token JC (myusbdrive.XX) (In-use by you) - it's currently mine
# Token JC (myusbdrive.XX) (In-use by:SYSTEM (SYSTEM) at 192.168.0.34) - someone else has it
#

# Report USB drive's attached
#
function eToken_List () {
    $pipeClient = new-object System.IO.Pipes.NamedPipeClientStream("vhclient")
    $pipeClient.Connect()
    $pipeClient.ReadMode = [System.IO.Pipes.PipeTransmissionMode]::Message

    $writer = new-object System.IO.StreamWriter($pipeClient)
    $writer.AutoFlush = $true

    $reader = new-object System.IO.StreamReader($pipeClient)

    $bytes = New-Object System.Collections.Generic.List[byte]

    $writer.Write("LIST")

    while ($reader.Peek() -ne -1)
    {
      $bytes.Add($reader.Read())

    }

    $pipeClient.Dispose()

    $result = [System.Text.Encoding]::UTF8.GetString($bytes)
    $result
    return
}

# Disconnect a USB drive
#
function eToken_Stop ( $token ) {
    $pipeClient = new-object System.IO.Pipes.NamedPipeClientStream("vhclient")
    $pipeClient.Connect()
    
    $writer = new-object System.IO.StreamWriter($pipeClient)
    $writer.AutoFlush = $true

    $token = 'STOP USING,' + $token
    Write-Host $token
    $writer.Write($token)
    Start-Sleep -m 1000

    $pipeClient.Dispose()
}

# Connect a USB drive
#
function eToken_Use ( $token ) {
    $pipeClient = new-object System.IO.Pipes.NamedPipeClientStream("vhclient")
    $pipeClient.Connect()
    
    $writer = new-object System.IO.StreamWriter($pipeClient)
    $writer.AutoFlush = $true

    $token = 'USE,' + $token
    Write-Host $token
    $writer.Write($token)
    Start-Sleep -m 1000

    $pipeClient.Dispose()
}

 

# token details we want to mount

$etoken = 'myusbdrive.XX'
$mount = 'failed'

# file to stash the VirtualHere info
# NB: Couldn't get the result to pipe even with Out-String -Stream | Select-String!

$filename = Get-Random
$filename = 'C:\' + $filename + '.txt'

eToken_List | Out-File -FilePath $filename -Encoding ASCII
$mytoken = Select-String -Pattern ('Token JC \(' + $etoken + '\) \(In-use by you') -Path $filename -Quiet
if ($mytoken -eq $false ) {

    # Is the token being used by someone else?

    $mytoken = Select-String -Pattern ('Token JC \(' + $etoken + '\) \(In-use by') -Path $filename -Quiet
    if ($mytoken -eq $true ) {
        eToken_Stop $etoken

        # recheck that token is now free

        eToken_List | Out-File -FilePath $filename -Encoding ASCII
        $mytoken = Select-String -Pattern ('Token JC \(' + $etoken + '\) \(In-use by') -Path $filename -Quiet
        if ($mytoken -eq $false ) {
            $mount = 'yes'
        }
    }
    else {
    
        # No one has the token
        
        $mount = 'yes'
    }
}
else {
    
    # We already have the token
    
    $mount = 'no'
}

# Housekeeping

Remove-Item $filename

 

# mount our token and exit

if ( $mount -eq 'yes' ) {
    eToken_Use $etoken
    exit 0
}
else {
    if ( $mount = 'no' ) {
        exit 0
    }
    else {
        exit 1
    }
}