mirrors/rdpgw
2
0
Fork 0
mirror of https://github.com/bolkedebruin/rdpgw.git synced 2026-02-10 18:07:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Re-org returning content

Browse Source
This commit is contained in:
Andrew Heberle 2025-08-30 16:03:03 +08:00
parent c79484091d
commit 53adff7326
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
cmd/rdpgw/web/web.go
View File

@ -242,21 +242,23 @@ func (h *Handler) HandleDownload(w http.ResponseWriter, r *http.Request) {
d.Settings.GatewayCredentialMethod = 1
d.Settings.GatewayUsageMethod = 1
if h.rdpSigner != nil {
// get rdp content
rdpContent := d.String()
signedContent, err := h.rdpSigner.SignRdp(rdpContent)
if err != nil {
log.Printf("Could not sign RDP file due to %s", err)
http.Error(w, errors.New("could not sign RDP file").Error(), http.StatusInternalServerError)
return
}
// return signd rdp file
http.ServeContent(w, r, fn, time.Now(), bytes.NewReader(signedContent))
// no rdp siging so return as-is
if h.rdpSigner == nil {
http.ServeContent(w, r, fn, time.Now(), strings.NewReader(d.String()))
return
}
http.ServeContent(w, r, fn, time.Now(), strings.NewReader(d.String()))
// get rdp content
rdpContent := d.String()
// sign rdp content
signedContent, err := h.rdpSigner.SignRdp(rdpContent)
if err != nil {
log.Printf("Could not sign RDP file due to %s", err)
http.Error(w, errors.New("could not sign RDP file").Error(), http.StatusInternalServerError)
return
}
// return signd rdp file
http.ServeContent(w, r, fn, time.Now(), bytes.NewReader(signedContent))
}