mirror of
https://github.com/bettercap/bettercap
synced 2025-08-20 21:43:18 -07:00
misc: small fix or general refactoring i did not bother commenting
This commit is contained in:
Simone Margaritelli
parent
6af2de6de9
commit
76e094f687
8 changed files with 733 additions and 679 deletions
220
modules/zerogod/zerogod_ipp_primitives.go
Normal file
220
modules/zerogod/zerogod_ipp_primitives.go
Normal file
|
|
@ -0,0 +1,220 @@
|
|||
package zerogod
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"slices"
|
||||
"strconv"
|
||||
|
||||
"github.com/evilsocket/islazy/str"
|
||||
"github.com/phin1x/go-ipp"
|
||||
)
|
||||
|
||||
const IPP_CHUNK_MAX_LINE_SIZE = 1024
|
||||
|
||||
var IPP_REQUEST_NAMES = map[int16]string{
|
||||
// https://tools.ietf.org/html/rfc2911#section-4.4.15
|
||||
0x0002: "Print-Job",
|
||||
0x0003: "Print-URI",
|
||||
0x0004: "Validate-Job",
|
||||
0x0005: "Create-Job",
|
||||
0x0006: "Send-Document",
|
||||
0x0007: "Send-URI",
|
||||
0x0008: "Cancel-Job",
|
||||
0x0009: "Get-Job-Attributes",
|
||||
0x000A: "Get-Jobs",
|
||||
0x000B: "Get-Printer-Attributes",
|
||||
0x000C: "Hold-Job",
|
||||
0x000D: "Release-Job",
|
||||
0x000E: "Restart-Job",
|
||||
0x0010: "Pause-Printer",
|
||||
0x0011: "Resume-Printer",
|
||||
0x0012: "Purge-Jobs",
|
||||
// https://web.archive.org/web/20061024184939/http://uw714doc.sco.com/en/cups/ipp.html
|
||||
0x4001: "CUPS-Get-Default",
|
||||
0x4002: "CUPS-Get-Printers",
|
||||
0x4003: "CUPS-Add-Modify-Printer",
|
||||
0x4004: "CUPS-Delete-Printer",
|
||||
0x4005: "CUPS-Get-Classes",
|
||||
0x4006: "CUPS-Add-Modify-Class",
|
||||
0x4007: "CUPS-Delete-Class",
|
||||
0x4008: "CUPS-Accept-Jobs",
|
||||
0x4009: "CUPS-Reject-Jobs",
|
||||
0x400A: "CUPS-Set-Default",
|
||||
0x400B: "CUPS-Get-Devices",
|
||||
0x400C: "CUPS-Get-PPDs",
|
||||
0x400D: "CUPS-Move-Job",
|
||||
}
|
||||
|
||||
var IPP_USER_ATTRIBUTES = map[string]string{
|
||||
"printer-name": "PRINTER_NAME",
|
||||
"printer-info": "PRINTER_INFO",
|
||||
"printer-make-and-model": "PRINTER_MAKE PRINTER_MODEL",
|
||||
"printer-location": "PRINTER_LOCATION",
|
||||
"printer-privacy-policy-uri": "https://www.bettercap.org/",
|
||||
"ppd-name": "everywhere",
|
||||
}
|
||||
|
||||
func init() {
|
||||
ipp.AttributeTagMapping["printer-uri-supported"] = ipp.TagUri
|
||||
ipp.AttributeTagMapping["uri-authentication-supported"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["uri-security-supported"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["printer-name"] = ipp.TagName
|
||||
ipp.AttributeTagMapping["printer-info"] = ipp.TagText
|
||||
ipp.AttributeTagMapping["printer-make-and-model"] = ipp.TagText
|
||||
ipp.AttributeTagMapping["printer-state"] = ipp.TagEnum
|
||||
ipp.AttributeTagMapping["printer-state-reasons"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["ipp-versions-supported"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["operations-supported"] = ipp.TagEnum
|
||||
ipp.AttributeTagMapping["multiple-document-jobs-supported"] = ipp.TagBoolean
|
||||
ipp.AttributeTagMapping["charset-configured"] = ipp.TagCharset
|
||||
ipp.AttributeTagMapping["charset-supported"] = ipp.TagCharset
|
||||
ipp.AttributeTagMapping["natural-language-configured"] = ipp.TagLanguage
|
||||
ipp.AttributeTagMapping["generated-natural-language-supported"] = ipp.TagLanguage
|
||||
ipp.AttributeTagMapping["document-format-default"] = ipp.TagMimeType
|
||||
ipp.AttributeTagMapping["document-format-supported"] = ipp.TagMimeType
|
||||
ipp.AttributeTagMapping["printer-is-accepting-jobs"] = ipp.TagBoolean
|
||||
ipp.AttributeTagMapping["queued-job-count"] = ipp.TagInteger
|
||||
ipp.AttributeTagMapping["pdl-override-supported"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["printer-up-time"] = ipp.TagInteger
|
||||
ipp.AttributeTagMapping["compression-supported"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["printer-privacy-policy-uri"] = ipp.TagUri
|
||||
ipp.AttributeTagMapping["printer-location"] = ipp.TagText
|
||||
ipp.AttributeTagMapping["ppd-name"] = ipp.TagName
|
||||
ipp.AttributeTagMapping["job-state-reasons"] = ipp.TagKeyword
|
||||
ipp.AttributeTagMapping["job-state"] = ipp.TagEnum
|
||||
ipp.AttributeTagMapping["job-uri"] = ipp.TagUri
|
||||
ipp.AttributeTagMapping["job-id"] = ipp.TagInteger
|
||||
ipp.AttributeTagMapping["job-printer-uri"] = ipp.TagUri
|
||||
ipp.AttributeTagMapping["job-name"] = ipp.TagName
|
||||
ipp.AttributeTagMapping["job-originating-user-name"] = ipp.TagName
|
||||
ipp.AttributeTagMapping["time-at-creation"] = ipp.TagInteger
|
||||
ipp.AttributeTagMapping["time-at-completed"] = ipp.TagInteger
|
||||
ipp.AttributeTagMapping["job-printer-up-time"] = ipp.TagInteger
|
||||
}
|
||||
|
||||
func ippReadChunkSizeHex(ctx *HandlerContext) string {
|
||||
var buf []byte
|
||||
|
||||
for b := make([]byte, 1); ; {
|
||||
if n, err := ctx.client.Read(b); err != nil {
|
||||
ctx.mod.Error("could not read chunked byte: %v", err)
|
||||
} else if n == 0 {
|
||||
break
|
||||
} else if b[0] == '\n' {
|
||||
break
|
||||
} else {
|
||||
// ctx.mod.Info("buf += 0x%x (%c)", b[0], b[0])
|
||||
buf = append(buf, b[0])
|
||||
}
|
||||
|
||||
if len(buf) >= IPP_CHUNK_MAX_LINE_SIZE {
|
||||
ctx.mod.Warning("buffer size exceeded %d bytes when reading chunk size", IPP_CHUNK_MAX_LINE_SIZE)
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return str.Trim(string(buf))
|
||||
}
|
||||
|
||||
func ippReadChunkSize(ctx *HandlerContext) (uint64, error) {
|
||||
if chunkSizeHex := ippReadChunkSizeHex(ctx); chunkSizeHex != "" {
|
||||
ctx.mod.Debug("got chunk size: 0x%s", chunkSizeHex)
|
||||
return strconv.ParseUint(chunkSizeHex, 16, 64)
|
||||
}
|
||||
return 0, nil
|
||||
}
|
||||
|
||||
func ippReadChunkedBody(ctx *HandlerContext) ([]byte, error) {
|
||||
var chunkedBody []byte
|
||||
// read chunked loop
|
||||
for {
|
||||
// read the next chunk size
|
||||
if chunkSize, err := ippReadChunkSize(ctx); err != nil {
|
||||
return nil, fmt.Errorf("error reading next chunk size: %v", err)
|
||||
} else if chunkSize == 0 {
|
||||
break
|
||||
} else {
|
||||
chunk := make([]byte, chunkSize)
|
||||
if n, err := ctx.client.Read(chunk); err != nil {
|
||||
return nil, fmt.Errorf("error while reading chunk of %d bytes: %v", chunkSize, err)
|
||||
} else if n != int(chunkSize) {
|
||||
return nil, fmt.Errorf("expected chunk of size %d, got %d bytes", chunkSize, n)
|
||||
} else {
|
||||
chunkedBody = append(chunkedBody, chunk...)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return chunkedBody, nil
|
||||
}
|
||||
|
||||
func ippReadRequestBody(ctx *HandlerContext, http_req *http.Request) (io.ReadCloser, error) {
|
||||
ipp_body := http_req.Body
|
||||
|
||||
// check for an Expect 100-continue
|
||||
if http_req.Header.Get("Expect") == "100-continue" {
|
||||
buf := make([]byte, 4096)
|
||||
|
||||
// inform the client we're ready to read the request body
|
||||
ctx.client.Write([]byte("HTTP/1.1 100 Continue\r\n\r\n"))
|
||||
|
||||
if slices.Contains(http_req.TransferEncoding, "chunked") {
|
||||
ctx.mod.Debug("detected chunked encoding")
|
||||
if body, err := ippReadChunkedBody(ctx); err != nil {
|
||||
return nil, err
|
||||
} else {
|
||||
ipp_body = io.NopCloser(bytes.NewReader(body))
|
||||
}
|
||||
} else {
|
||||
// read the body in a single step
|
||||
read, err := ctx.client.Read(buf)
|
||||
if err != nil {
|
||||
if err == io.EOF {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, fmt.Errorf("error while reading ipp body from %v: %v", ctx.client.RemoteAddr(), err)
|
||||
} else if read == 0 {
|
||||
return nil, fmt.Errorf("error while reading ipp body from %v: no data", ctx.client.RemoteAddr())
|
||||
}
|
||||
|
||||
ipp_body = io.NopCloser(bytes.NewReader(buf[0:read]))
|
||||
}
|
||||
}
|
||||
|
||||
return ipp_body, nil
|
||||
}
|
||||
|
||||
func ippSendResponse(ctx *HandlerContext, response *ipp.Response) {
|
||||
ctx.mod.Debug("SENDING %++v", *response)
|
||||
|
||||
resp_data, err := response.Encode()
|
||||
if err != nil {
|
||||
ctx.mod.Error("error while encoding ipp response: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
headers := [][]byte{
|
||||
[]byte("HTTP/1.1 200 OK\r\n"),
|
||||
[]byte("Content-Type: application/ipp\r\n"),
|
||||
[]byte(fmt.Sprintf("Content-Length: %d\r\n", len(resp_data))),
|
||||
[]byte("Connection: close\r\n"),
|
||||
[]byte("\r\n"),
|
||||
}
|
||||
|
||||
for _, header := range headers {
|
||||
if _, err := ctx.client.Write(header); err != nil {
|
||||
ctx.mod.Error("error while writing header: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if _, err = ctx.client.Write(resp_data); err != nil {
|
||||
ctx.mod.Error("error while writing ipp response data: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
ctx.mod.Debug("sent %d of ipp response to %v", len(resp_data), ctx.client.RemoteAddr())
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue