http - 解密后的反向代理服务文件
问题描述
我需要将 *os.File 放入 request.Body 但我不知道如何处理文件描述符的关闭(以及临时文件的消除)。我不能使用 http.ServeFile 来提供文件。
我有 3 个软件:A)用户界面(我无法编辑) B)提供文件的后端 C)一个 http 反向代理,其作用是签署从 A 到 B 的请求,加密从 A 到 B 的文件,和从 B 到 A 的解密文件。
软件 A 预计会收到解密文件内容的主体。我知道使用 B 解密目录中的文件然后让 A 打开它会更容易,但不幸的是我无法编辑 A ,它需要请求正文中的文件。
软件的工作流程如下:软件 A 向 C 发送请求,C 签署请求并将其转发给 B,然后 C 继续解密文件并将解密后的内容在请求正文中提供给 A。
type transportPageServeFile struct {
http.RoundTripper
// vault handles encryption and decryption of files
vault *vault.Container
}
func (t *transportPageServeFile) RoundTrip(req *http.Request) (resp *http.Response, err error) {
resp, err = t.RoundTripper.RoundTrip(req)
if err != nil {
return nil, errors.Wrap(err, "transportPageServeFile")
}
// create temporary destination dir for the decrypted file
dstDir, err := utils.MkTmpDir()
if err != nil {
return nil, errors.Wrap(err, "transportPageServeFile")
}
defer os.RemoveAll(dstDir) // i'm deleting the directory where the temporary file is stored, deleting it before it even gets transmitted to software A
fileName := req.Header.Get(api.FileNameHeader)
decryptedFileDst := fmt.Sprintf("%s/%s", dstDir, fileName)
// DecryptFileFromReader takes a reader and decrypts the file, saving it in the decided destination
err = t.vault.DecryptFileFromReader(resp.Body, decryptedFileDst)
if err != nil {
return nil, errors.Wrap(err, "transportPageServeFile")
}
defer resp.Body.Close()
// now open the saved decrypted file
decryptedFile, err := os.Open(decryptedFileDst)
if err != nil {
return nil, errors.Wrap(err,"transportPageServeFile")
}
defer decryptedFile.Close() // close file before serving the content?
resp.Body = decryptedFile
return resp, nil
}
// pageServeFile is the handler of /serveFile
func (rpc *RPC) pageServeFile(c *gin.Context) {
transport := &transportPageServeFile{RoundTripper:http.DefaultTransport, vault:rpc.vault}
target, err := url.Parse(rpc.apiEndpoint)
if err != nil {
rpc.onError(c, errors.Wrapf(err, "pageServeFile"))
return
}
proxy := httputil.NewSingleHostReverseProxy(target)
proxy.Transport = transport
proxy.ServeHTTP(c.Writer, c.Request)
}
如何在请求正文中复制文件(使用阅读器)以及在转发响应后何时/如何关闭文件描述符?
解决方案
响应体是一个 io.ReadCloser,因此假设反向代理调用 Close 是合理的。确实如此,正如您通过运行以下程序所看到的那样:
package main
import (
"fmt"
"io"
"net/http"
"net/http/httptest"
"net/http/httputil"
"net/url"
"runtime/debug"
)
type dbg struct{ io.ReadCloser }
func (d dbg) Close() error {
fmt.Println("Close() called:")
debug.PrintStack()
return d.ReadCloser.Close()
}
func main() {
// srv simulates "software B"
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello, world"))
}))
target, _ := url.Parse(srv.URL)
p := httputil.NewSingleHostReverseProxy(target)
// To modify the response, use the aptly named ModifyResponse field; no
// need to implement a RoundTripper.
p.ModifyResponse = func(res *http.Response) error {
res.Body = dbg{res.Body}
return nil
}
req := httptest.NewRequest("GET", "http://example.com/hello", nil)
rec := httptest.NewRecorder()
p.ServeHTTP(rec, req)
}
// Output:
// Close() called:
// goroutine 1 [running]:
// runtime/debug.Stack(0x10, 0x0, 0x0)
// /usr/local/go/src/runtime/debug/stack.go:24 +0x9d
// runtime/debug.PrintStack()
// /usr/local/go/src/runtime/debug/stack.go:16 +0x22
// main.dbg.Close(0x75b780, 0xc00013e140, 0xc000152000, 0x7f61fc523088)
// /tmp/tmp.H9O7cizbkv/main.go:17 +0x7f
// net/http/httputil.(*ReverseProxy).ServeHTTP(0xc0000ac0f0, 0x75e460, 0xc000032440, 0xc00010a000)
// /usr/local/go/src/net/http/httputil/reverseproxy.go:311 +0x8a6
// main.main()
// /tmp/tmp.H9O7cizbkv/main.go:41 +0x152
推荐阅读
- excel - VBA - 将我返回到原始状态的代码(可能将字段类型从“文本”转换为“数字”)
- angular - 使用 Visual Studio 2015 在 .Net 上开发 Angular 7 应用程序
- java - 如何为在java中计算后缀表达式的RPN算法添加一元运算符
- java - A* 搜索算法自定义 f(n)
- javascript - 在javascript中删除二维数组中的给定单元格
- java - Java 8 从输入文件流式创建多个对象
- javascript - Jquery slimjet 事件委托不起作用
- c# - WPF 应用程序无法删除/保存文件
- python - 如何在 python 3.6.4 中为不和谐机器人创建商店?
- ruby-on-rails - Friendly_id 由于 before_action 阻止编辑/新页面:find_post