add: png detector example

This commit is contained in:
Inanc Gumus
2019-11-05 10:48:27 +03:00
parent acbec476c3
commit ff25bcdcb9
2 changed files with 84 additions and 25 deletions

View File

@ -9,38 +9,44 @@
package main package main
import ( import (
"bytes"
"fmt"
"io" "io"
"log" "log"
"os" "os"
) )
// you can download the rosie.unknown image in the link:
// https://inancgumus.github.com/x/rosie.unknown
// then feed the file to the standard input of this program:
// go run . < rosie.unknown
func main() { func main() {
const pngSign = "\x89PNG\r\n\x1a\n"
const pngSignLen = 8
// create an in-memory buffer (bytes.Buffer is an io.Reader and an io.Writer).
memory := bytes.Buffer{}
// stream from the standard input to the in-memory buffer in 32KB data chunks. // stream from the standard input to the in-memory buffer in 32KB data chunks.
// os.Stdin.Read(...) -> memory.Write(...) // os.Stdin.Read(...) -> os.Stdout.Write(...)
if _, err := io.Copy(&memory, os.Stdin); err != nil { if _, err := io.Copy(os.Stdout, os.Stdin); err != nil {
log.Fatal(err) log.Fatal(err)
} }
}
// get the accumulated bytes from the in-memory buffer.
buf := memory.Bytes() // ioCopy streams from a file to another file.
// we use it to stream from the standard input to ouput.
// print the first eight bytes. func ioCopy(dst, src *os.File) error {
fmt.Printf("% x\n", buf[:pngSignLen]) // Use a fixed-length buffer to efficiently read from src stream in chunks.
buf := make([]byte, 32768)
// compare it with the png signature.
fmt.Printf("% x\n", pngSign) // read over and over again to read all the data.
for {
// read can read only up to the buffer length.
nr, er := src.Read(buf)
// only write data if there is something to write.
if nr > 0 {
_, ew := dst.Write(buf[:nr])
if ew != nil {
return ew
}
}
// io.EOF = there is nothing left to read—close the loop.
if er == io.EOF {
return nil
}
// Only return an error if the reading really fails.
if er != nil {
return er
}
}
return nil
} }

View File

@ -0,0 +1,53 @@
// Copyright © 2018 Inanc Gumus
// Learn Go Programming Course
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
//
// For more tutorials : https://learngoprogramming.com
// In-person training : https://www.linkedin.com/in/inancgumus/
// Follow me on twitter: https://twitter.com/inancgumus
package main
import (
"bytes"
"fmt"
"io"
"log"
"net/http"
)
// you can download the rosie.unknown image in the link:
// https://inancgumus.github.com/x/rosie.unknown
// then feed the file to the standard input of this program:
// go run . < rosie.unknown
func main() {
const pngSign = "\x89PNG\r\n\x1a\n"
const pngSignLen = 8
// create an in-memory buffer (bytes.Buffer is an io.Reader and an io.Writer).
memory := bytes.Buffer{}
// get it from the web server
resp, err := http.Get("https://inancgumus.github.com/x/rosie.unknown")
if err != nil {
log.Fatal(err)
}
defer resp.Body.Close()
// stream from the standard input to the in-memory buffer in 32KB data chunks.
// os.Stdin.Read(...) -> memory.Write(...)
if _, err := io.Copy(&memory, resp.Body); err != nil {
log.Fatal(err)
}
// get the accumulated bytes from the in-memory buffer.
buf := memory.Bytes()
// print the first eight bytes.
fmt.Printf("% x\n", buf[:pngSignLen])
// compare it with the png signature.
fmt.Printf("% x\n", pngSign)
}