X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/21581958c38861f3a62ef7c27d0a8a6f4db2a1f8..19dcc94821d1fac803c571b1dc11764213713682:/hotline/transaction_handlers.go diff --git a/hotline/transaction_handlers.go b/hotline/transaction_handlers.go index cad3370..f6fa232 100644 --- a/hotline/transaction_handlers.go +++ b/hotline/transaction_handlers.go @@ -231,6 +231,10 @@ var TransactionHandlers = map[uint16]TransactionType{ Name: "tranUserBroadcast", Handler: HandleUserBroadcast, }, + tranDownloadBanner: { + Name: "tranDownloadBanner", + Handler: HandleDownloadBanner, + }, } func HandleChatSend(cc *ClientConn, t *Transaction) (res []Transaction, err error) { @@ -976,6 +980,10 @@ func HandleTranAgreed(cc *ClientConn, t *Transaction) (res []Transaction, err er cc.Server.outbox <- t } + if cc.Server.Config.BannerFile != "" { + cc.Server.outbox <- *NewTransaction(tranServerBanner, cc.ID, NewField(fieldBannerType, []byte("JPEG"))) + } + res = append(res, cc.NewReply(t)) return res, err @@ -1927,3 +1935,32 @@ func HandleMakeAlias(cc *ClientConn, t *Transaction) (res []Transaction, err err res = append(res, cc.NewReply(t)) return res, err } + +func HandleDownloadBanner(cc *ClientConn, t *Transaction) (res []Transaction, err error) { + transactionRef := cc.Server.NewTransactionRef() + data := binary.BigEndian.Uint32(transactionRef) + + ft := &FileTransfer{ + ReferenceNumber: transactionRef, + Type: bannerDownload, + } + + fi, err := cc.Server.FS.Stat(filepath.Join(cc.Server.ConfigDir, cc.Server.Config.BannerFile)) + if err != nil { + return res, err + } + + size := make([]byte, 4) + binary.BigEndian.PutUint32(size, uint32(fi.Size())) + + cc.Server.mux.Lock() + defer cc.Server.mux.Unlock() + cc.Server.FileTransfers[data] = ft + + res = append(res, cc.NewReply(t, + NewField(fieldRefNum, transactionRef), + NewField(fieldTransferSize, size), + )) + + return res, err +}