revsuit/internal/ipinfo/qqwry/download.go

package qqwry

import (
	"bytes"
	"compress/zlib"
	"crypto/tls"
	"encoding/binary"
	"errors"
	"io"
	"net/http"
	"os"
	"sync"
	"time"

	log "unknwon.dev/clog/v2"
)

const (
	CopyWriteUrl = "https://qqwry.mirror.noc.one/copywrite.rar"
	Url          = "https://qqwry.mirror.noc.one/qqwry.rar"
)

func get(url string) (b []byte, err error) {
	var resp *http.Response
	client := http.Client{
		Timeout: 90 * time.Second,
		Transport: &http.Transport{
			TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, // disable verify
		}}
	request, _ := http.NewRequest("GET", url, nil)
	request.Header.Add("User-Agent", "Nali/2.1.2 (Nali CLI, https://nali.skk.moe)")

	resp, err = client.Do(request)
	if err != nil {
		return
	}

	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
			log.Warn("%v", err)
		}
	}(resp.Body)

	return io.ReadAll(resp.Body)
}

func getKey(b []byte) (key uint32, err error) {
	if len(b) != 280 {
		return 0, errors.New("copywrite.rar is corrupt")
	}
	key = binary.LittleEndian.Uint32(b[20:])
	return key, err
}

func decrypt(b []byte, key uint32) (_ []byte, err error) {
	var rc io.ReadCloser
	for i := 0; i < 0x200; i++ {
		key *= uint32(0x805)
		key++
		key &= uint32(0xff)
		b[i] = b[i] ^ byte(key)
	}
	rc, err = zlib.NewReader(bytes.NewBuffer(b))
	if err != nil {
		return
	}

	defer func(rc io.ReadCloser) {
		err := rc.Close()
		if err != nil {
			log.Warn("%v", err)
		}
	}(rc)

	return io.ReadAll(rc)
}

func download() (err error) {
	var (
		copyWriteData, qqwryData []byte
		wg                       sync.WaitGroup
		key                      uint32
	)
	log.Info("Downloading qqwry.dat...")

	wg.Add(2)
	go func() {
		defer wg.Done()
		if copyWriteData, err = get(CopyWriteUrl); err != nil {
			return
		}
	}()

	go func() {
		defer wg.Done()
		if qqwryData, err = get(Url); err != nil {
			return
		}
	}()
	wg.Wait()

	if err != nil {
		return err
	}
	if key, err = getKey(copyWriteData); err != nil {
		return err
	}
	if b, err := decrypt(qqwryData, key); err != nil {
		return err
	} else {
		return os.WriteFile("qqwry.dat", b, 0644)
	}
}