go:封装io.Reader 实现灵活可扩展的io程序

package rc4

import (
    "io"
)

type RC4Reader struct {
    baseReader io.Reader // 特别要注意这个baseReader，这里屏蔽了底层的操作，专注于加密这个用户逻辑
    password   string
    sbox       []byte
    i, j       int
}

func NewReader(password string, reader io.Reader) *RC4Reader {
    key := []byte(password)
    keylength := len(key)

    r := new(RC4Reader)
    r.password = password
    r.baseReader = reader

    sbox := make([]byte, 256)
    for i := 0; i <= 255; i++ {
        sbox[i] = byte(i)
    }

    j := 0
    for i := 0; i < 256; i++ {
        j = (j + int(sbox[i]) + int(key[i%keylength])) % 256
        sbox[i], sbox[j] = sbox[j], sbox[i]
    }
    r.sbox = sbox
    return r
}

func (r *RC4Reader) Read(p []byte) (n int, err error) {
    tp := make([]byte, len(p), len(p))
    n, err = r.baseReader.Read(tp) // 从baseReader读取数据，基于对接口的信任，这一定能取到需要加密的数据，而不需要管这个数据是从string还是文件或者网络流中来。这就是接口的妙用。
    // 同时，处理baseReader的错误，进行封装即可
    if err != nil {
        return n, err
    }
    tp = tp[:n]

    // 下面是加密编码操作，不需要理解
    ret := make([]byte, n, n)
    for i := 0; i < n; i++ {
        r.i = (r.i + 1) % 256                //a
        r.j = (r.j + int(r.sbox[r.i])) % 256 //b
        r.sbox[r.i], r.sbox[r.j] = r.sbox[r.j], r.sbox[r.i]
        ret[i] = tp[i] ^ r.sbox[(int(r.sbox[r.i])+int(r.sbox[r.j]))%256]
    }

    copy(p, ret)
    return n, nil
}

func main() {
    // 使用strings.Reader作为数据源，直接加密string的数据
    r := rc4.NewReader("yzh", strings.NewReader("yzhyzhyzhyzhyzhyzhyzhyzhyzhyzhyzhyzh"))
    var enc bytes.Buffer
    for {
        ret := make([]byte, 4, 4)
        n, err := r.Read(ret)
        if err != nil {
            break
        }
        enc.Write(ret[:n])
    }
    fmt.Println("encrypt is", enc.Bytes())

    // 使用bytes.Reader作为数据源，解码bytes数据
    rdec := rc4.NewReader("yzh", bytes.NewReader(enc.Bytes()))
    var dec bytes.Buffer
    for {
        ret := make([]byte, 4, 4)
        n, err := rdec.Read(ret)
        if err != nil {
            break
        }
        dec.Write(ret[:n])
    }
    fmt.Println("decrypt is", dec.Bytes(), dec.String())
}