在 golang 中打开超时的 PE 文件

Question

我想尝试在 Go 中打开一个超时的 PE 文件。为了实现这一点，我在引导文件指针和错误时使用匿名函数。我使用带有超时情况的 select 子句来强制执行超时，如下所示。

go func() {
    f, e := pe.Open(filePath)
    file <- f
    err <- e
}()
select {
case <-fileOpenTimeout:
    fmt.Printf("ERROR: Opening PE file timed out")
    return
case fileError := <-err:
    if fileError == nil{...}
}

这段代码适用于我的用例。但是，如果文件打开时间过长，这可能会导致资源泄漏。我该如何防止这种情况？有没有更好的方法来强制打开 PE 文件时超时？

Answer 1

如果你有一个传递给匿名函数的 done 通道，你可以使用它来发送你提前结束的信号。

func asd() {
    fileOpenTimeout := time.After(5 * time.Second)
    type fileResponse struct {
        file *pe.File
        err error
    }
    response := make(chan fileResponse)
    done := make(chan struct{})
    go func(done <-chan struct{}) {
        f, e := pe.Open(filePath)
        r := fileResponse{
            file: f,
            err: e,
        }
        select {
        case response <- r:
            // do nothing, response sent
        case <-done:
            // clean up
            if f != nil {
                f.Close()
            }
        }
    }(done)
    select {
    case <-fileOpenTimeout:
        fmt.Printf("ERROR: Opening PE file timed out")
        close(done)
        return
    case r := <-response:
        if r.err != nil { ... }
    }
}

当完成通道关闭时，您将始终能够读取零值。所以你的匿名功能不会泄漏。还有一个结构文件响应，其范围仅限于函数，以简化从go例程传回多个值的过程