cmd/ungzip/main.go

   1 // tofuproxy -- flexible HTTP/HTTPS proxy, TLS terminator, X.509 TOFU
   2 //              manager, WARC/geminispace browser
   3 // Copyright (C) 2021-2024 Sergey Matveev <stargrave@stargrave.org>
   4 //
   5 // This program is free software: you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation, version 3 of the License.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 package main
  18
  19 import (
  20         "bufio"
  21         "compress/gzip"
  22         "fmt"
  23         "io"
  24         "log"
  25         "os"
  26 )
  27
  28 type CountableReader struct {
  29         *bufio.Reader
  30         n int64
  31 }
  32
  33 func (r *CountableReader) ReadByte() (b byte, err error) {
  34         b, err = r.Reader.ReadByte()
  35         if err == nil {
  36                 r.n++
  37         }
  38         return b, err
  39 }
  40
  41 func (r *CountableReader) Read(p []byte) (n int, err error) {
  42         n, err = r.Reader.Read(p)
  43         r.n += int64(n)
  44         return
  45 }
  46
  47 func main() {
  48         log.SetFlags(log.Lshortfile)
  49         fdOff := os.NewFile(3, "fdOff")
  50         br := bufio.NewReader(os.Stdin)
  51         cr := &CountableReader{Reader: br}
  52         bw := bufio.NewWriter(os.Stdout)
  53         z, err := gzip.NewReader(cr)
  54         if err != nil {
  55                 log.Fatalln(err)
  56         }
  57         z.Multistream(false)
  58         var offset, offsetPrev int64
  59         for {
  60                 written, err := io.Copy(bw, z)
  61                 if err != nil {
  62                         log.Fatalln(err)
  63                 }
  64                 offset = cr.n
  65                 if fdOff != nil {
  66                         fmt.Fprintf(fdOff, "%d\t%d\n", offset-offsetPrev, written)
  67                 }
  68                 offsetPrev = offset
  69                 err = z.Reset(cr)
  70                 if err != nil {
  71                         if err == io.EOF {
  72                                 break
  73                         }
  74                         log.Fatalln(err)
  75                 }
  76                 z.Multistream(false)
  77         }
  78 }