81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"ippserver/packages/ipp"
|
|
"ippserver/packages/mdnsserver"
|
|
"net/http"
|
|
"sync"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
loglevel string
|
|
location string
|
|
)
|
|
|
|
func init() {
|
|
flag.StringVar(&loglevel, "loglevel", "info", "The wanted loglevel error/info/debug")
|
|
}
|
|
|
|
func main() {
|
|
customFormatter := new(log.TextFormatter)
|
|
customFormatter.TimestampFormat = "2006-01-02 15:04:05"
|
|
log.SetFormatter(customFormatter)
|
|
customFormatter.FullTimestamp = true
|
|
log.SetLevel(log.InfoLevel)
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
go mdnsserver.Run(ctx)
|
|
|
|
http.HandleFunc("/ipp/print", handle)
|
|
|
|
log.Info("http server started on :1234")
|
|
err := http.ListenAndServe(":1234", nil)
|
|
if err != nil {
|
|
log.Fatal("ListenAndServe: " + err.Error())
|
|
}
|
|
}
|
|
|
|
var mut sync.Mutex
|
|
var requestID uint32
|
|
|
|
func handle(w http.ResponseWriter, r *http.Request) {
|
|
mut.Lock()
|
|
requestID++
|
|
mut.Unlock()
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, "Unsupported method", http.StatusMethodNotAllowed)
|
|
}
|
|
|
|
request := ipp.NewRequest(0, 0)
|
|
request.UnMarshal(r.Body)
|
|
log.Infof("Upstream Request: id: %v op: %v", request.RequestID(), request.Operation())
|
|
var response *ipp.Response
|
|
var err error
|
|
switch request.Operation() {
|
|
case ipp.GetPrinterAttributes:
|
|
response = handleGetPrinterAttributes(request)
|
|
case ipp.PrintJob:
|
|
response, err = handlePrintJob(request, r.Body, requestID)
|
|
case ipp.GetJobs:
|
|
response, err = handleGetJobs(request, requestID)
|
|
case ipp.ValidateJob:
|
|
response = handleValidateJob(request)
|
|
case ipp.GetJobAttributes:
|
|
response, err = handleGetJobAttributes(request, requestID)
|
|
default:
|
|
response = ipp.NewResponse(ipp.ClientErrorBadRequest, request.RequestID())
|
|
}
|
|
if err != nil {
|
|
log.Errorf("Failed to handle request: %v", err.Error())
|
|
response = ipp.NewResponse(ipp.ServerErrorServiceUnavailable, request.RequestID())
|
|
}
|
|
log.Infof("Upstream Response: %v", response.Header())
|
|
data := response.Marshal()
|
|
w.Write(data)
|
|
}
|