X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/df2735b25a21d839746be31b43edb9e348086b0a..945b98135dceeebc524fd7fc557204c80f09c21d:/hotline/transfer.go diff --git a/hotline/transfer.go b/hotline/transfer.go index fa431ce..2acb5d6 100644 --- a/hotline/transfer.go +++ b/hotline/transfer.go @@ -4,6 +4,9 @@ import ( "bytes" "encoding/binary" "errors" + "io" + "os" + "path/filepath" ) type transfer struct { @@ -26,3 +29,41 @@ func (tf *transfer) Write(b []byte) (int, error) { return len(b), nil } + +func receiveFile(r io.Reader, targetFile, resForkFile, infoFork, counterWriter io.Writer) error { + var ffo flattenedFileObject + if _, err := ffo.ReadFrom(r); err != nil { + return err + } + + // Write the information fork + _, err := infoFork.Write(ffo.FlatFileInformationFork.MarshalBinary()) + if err != nil { + return err + } + + if _, err = io.CopyN(targetFile, io.TeeReader(r, counterWriter), ffo.dataSize()); err != nil { + return err + } + + if ffo.FlatFileHeader.ForkCount == [2]byte{0, 3} { + if err := binary.Read(r, binary.BigEndian, &ffo.FlatFileResForkHeader); err != nil { + return err + } + + if _, err = io.CopyN(resForkFile, io.TeeReader(r, counterWriter), ffo.rsrcSize()); err != nil { + return err + } + } + return nil +} + +func (s *Server) bannerDownload(w io.Writer) error { + bannerBytes, err := os.ReadFile(filepath.Join(s.ConfigDir, s.Config.BannerFile)) + if err != nil { + return err + } + _, err = w.Write(bannerBytes) + + return err +}