cmd/ungzip/main.go

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