package main

import (
    "fmt"
    "syscall"
    "unicode/utf16"
    "unsafe"
)

// 定義DLL句柄和函數(shù)地址
var (
    WinSCard, _                  = syscall.LoadLibrary(`C:\windows\system32\WinSCard.dll`)
    procSCardEstablishContext, _ = syscall.GetProcAddress(WinSCard, "SCardEstablishContext")
    procSCardReleaseContext, _   = syscall.GetProcAddress(WinSCard, "SCardReleaseContext")
    // 注意：這里使用 "SCardListReadersW" 而不是 "SCardListReaders"
    procSCardListReaders, _      = syscall.GetProcAddress(WinSCard, "SCardListReadersW")
)

// 定義SCard API常量
const (
    SCARD_SCOPE_USER   = 0 // 用戶上下文
    SCARD_SCOPE_SYSTEM = 2 // 系統(tǒng)上下文

    SCARD_ALL_READERS     = "SCard$AllReaders"     // 所有讀卡器組
    SCARD_DEFAULT_READERS = "SCard$DefaultReaders" // 默認(rèn)讀卡器組
)

// SCardListReadersW 的 Go 封裝
// hContext: 智能卡資源管理器的上下文句柄
// mszGroups: 讀卡器組名稱，UTF-16編碼，多字符串列表
// mszReaders: 輸出緩沖區(qū)，用于接收讀卡器名稱，UTF-16編碼，多字符串列表
// pcchReaders: 輸入/輸出參數(shù)，指定/返回 mszReaders 緩沖區(qū)的大?。ㄗ址麛?shù)）
func SCardListReaders(hContext syscall.Handle, mszGroups *uint16, mszReaders *uint16, pcchReaders *uint32) (retval error) {
    r0, _, _ := syscall.Syscall6(
        uintptr(procSCardListReaders),
        4, // 參數(shù)數(shù)量
        uintptr(hContext),
        uintptr(unsafe.Pointer(mszGroups)),
        uintptr(unsafe.Pointer(mszReaders)),
        uintptr(unsafe.Pointer(pcchReaders)),
        0,
        0,
    )
    if r0 != 0 {
        retval = syscall.Errno(r0)
    }
    return
}

// SCardReleaseContext 的 Go 封裝
// hContext: 智能卡資源管理器的上下文句柄
func SCardReleaseContext(hContext syscall.Handle) (retval error) {
    r0, _, _ := syscall.Syscall(
        uintptr(procSCardReleaseContext),
        1, // 參數(shù)數(shù)量
        uintptr(hContext),
        0,
        0,
    )
    if r0 != 0 {
        retval = syscall.Errno(r0)
    }
    return
}

// SCardEstablishContext 的 Go 封裝
// dwScope: 上下文范圍
// pvReserved1, pvReserved2: 保留參數(shù)，通常為0
// phContext: 輸出參數(shù)，接收智能卡資源管理器的上下文句柄
func SCardEstablishContext(dwScope uint32, pvReserved1 uintptr, pvReserved2 uintptr, phContext *syscall.Handle) (retval error) {
    r0, _, _ := syscall.Syscall6(
        uintptr(procSCardEstablishContext),
        4, // 參數(shù)數(shù)量
        uintptr(dwScope),
        uintptr(pvReserved1),
        uintptr(pvReserved2),
        uintptr(unsafe.Pointer(phContext)), // 傳遞 phContext 變量的地址
        0,
        0,
    )
    if r0 != 0 {
        retval = syscall.Errno(r0)
    }
    return
}

// 將錯(cuò)誤轉(zhuǎn)換為 uint32 類型的錯(cuò)誤碼
func ReturnValue(err error) uint32 {
    rv, ok := err.(syscall.Errno)
    if !ok {
        rv = 0 // 如果不是 syscall.Errno 類型，則返回0
    }
    return uint32(rv)
}

// 將 UTF-16 編碼的多字符串列表 (以雙空字符結(jié)束) 轉(zhuǎn)換為 Go 的 []string
func UTF16ToStrings(ls []uint16) []string {
    var ss []string
    if len(ls) == 0 {
        return ss
    }
    // 確保切片以雙空字符結(jié)束，以便正確解析
    if ls[len(ls)-1] != 0 {
        ls = append(ls, 0)
    }
    i := 0
    for j, cu := range ls {
        if cu == 0 { // 遇到空字符，表示一個(gè)字符串結(jié)束
            if j >= 1 && ls[j-1] == 0 { // 遇到雙空字符，表示列表結(jié)束
                break
            }
            if j-i > 0 { // 如果當(dāng)前字符串非空，則解碼并添加
                ss = append(ss, string(utf16.Decode(ls[i:j])))
            }
            i = j + 1 // 移動到下一個(gè)字符串的起始位置
            continue
        }
    }
    return ss
}

func main() {
    var (
        context  syscall.Handle // 智能卡上下文句柄
        scope    uint32         // 上下文范圍
        groups   *uint16        // 讀卡器組名稱指針
        cReaders uint32         // 讀卡器名稱緩沖區(qū)大小
    )

    // 確保在程序退出時(shí)釋放DLL
    defer syscall.FreeLibrary(WinSCard)

    // --- 嘗試列出讀卡器（在建立上下文之前，某些系統(tǒng)可能無法列出所有讀卡器） ---
    // 初始化 context 為0，表示不使用已建立的上下文
    context = 0
    // 將 Go 字符串轉(zhuǎn)換為 UTF-16 指針
    groups, err := syscall.UTF16PtrFromString(SCARD_ALL_READERS)
    if err != nil {
        fmt.Println("Reader Group conversion error: ", err)
        return
    }

    // 第一次調(diào)用 SCardListReaders 獲取所需緩沖區(qū)大小
    // mszReaders 傳入 nil，pcchReaders 接收所需大小
    err = SCardListReaders(context, groups, nil, &cReaders)
    if err != nil {
        // 如果返回 SCARD_E_NO_READERS_FOUND (0x80100002) 或 SCARD_E_NO_SMARTCARD (0x80100003)，表示沒有讀卡器
        // 或者 SCARD_E_SERVICE_STOPPED (0x8010001D) 表示智能卡服務(wù)未運(yùn)行
        fmt.Printf("SCardListReaders (initial call) failed: 0x%X %s\n", ReturnValue(err), err)
        // 如果錯(cuò)誤是 SCARD_E_NO_READERS_FOUND，不認(rèn)為是致命錯(cuò)誤，可以繼續(xù)
        if ReturnValue(err) == 0x80100002 {
            fmt.Println("No smart card readers found.")
        } else {
            return
        }
    }

    // 如果有讀卡器，分配緩沖區(qū)并再次調(diào)用 SCardListReaders 獲取實(shí)際數(shù)據(jù)
    if cReaders > 0 {
        r := make([]uint16, cReaders) // 分配足夠大的 UTF-16 緩沖區(qū)
        err = SCardListReaders(context, groups, &r[0], &cReaders) // 傳入緩沖區(qū)地址
        if err != nil {
            fmt.Printf("SCardListReaders (data retrieval) failed: 0x%X %s\n", ReturnValue(err), err)
            return
        }
        // 將 UTF-16 編碼的讀卡器列表轉(zhuǎn)換為 Go 的 []string
        readers := UTF16ToStrings(r[:cReaders])
        fmt.Println("Readers:", len(readers), readers)
    } else {
        fmt.Println("No readers found after initial check.")
    }


    // --- 建立智能卡上下文 ---
    scope = SCARD_SCOPE_SYSTEM // 設(shè)置上下文范圍為系統(tǒng)級別
    // 調(diào)用 SCardEstablishContext，phContext 傳入 context 變量的地址
    err = SCardEstablishContext(scope, 0, 0, &context)
    if err != nil {
        fmt.Printf("SCardEstablishContext failed: 0x%X %s\n", ReturnValue(err), err)
        // 常見錯(cuò)誤：0x8010001D (SCARD_E_SERVICE_STOPPED) - 智能卡資源管理器服務(wù)未運(yùn)行
        return
    }
    // 確保在函數(shù)退出時(shí)釋放上下文
    defer SCardReleaseContext(context)
    fmt.Printf("Context established: %X\n", context)

    // 可以在這里進(jìn)行其他智能卡操作...
}