package advanced_nego

import (
	"errors"
	"fmt"

	"gitee.com/wlhet/orago/network/security"
)

type encryptService struct {
	defaultService
	algoID int
}

func NewEncryptService(comm *AdvancedNegoComm) (*encryptService, error) {
	output := &encryptService{
		defaultService: defaultService{
			comm:        comm,
			serviceType: 2,
			version:     0xB200200,
			availableServiceNames: []string{"", "RC4_40", "RC4_56", "RC4_128", "RC4_256",
				"DES40C", "DES56C", "3DES112", "3DES168", "AES128", "AES192", "AES256"},
			availableServiceIDs: []int{0, 1, 8, 10, 6, 3, 2, 11, 12, 15, 16, 17},
		},
	}
	err := output.buildServiceList([]string{"DES56C", "AES128", "AES192", "AES256"}, true, true)
	//output.selectedServ, err = output.validate(strings.Split(str,","), true)
	if err != nil {
		return nil, err
	}
	return output, nil
}

func (serv *encryptService) readServiceData(subPacketnum int) error {
	var err error
	comm := serv.comm
	serv.version, err = comm.readVersion()
	if err != nil {
		return err
	}
	resp, err := comm.readUB1()
	if err != nil {
		return err
	}
	serv.algoID = int(resp)

	return nil
}
func (serv *encryptService) writeServiceData() error {
	serv.writeHeader(3)
	comm := serv.comm
	comm.writeVersion(serv.getVersion())
	selectedIndices := make([]byte, len(serv.selectedIndices))
	for i := 0; i < len(serv.selectedIndices); i++ {
		index := serv.selectedIndices[i]
		selectedIndices[i] = uint8(serv.availableServiceIDs[index])
	}
	comm.writeBytes(selectedIndices)
	// send selected driver
	comm.writeUB1(1)
	return nil
}

func (serv *encryptService) getServiceDataLength() int {
	return 17 + len(serv.selectedIndices)
}

func (serv *encryptService) activateAlgorithm() error {
	key := serv.comm.session.Context.AdvancedService.SessionKey
	//iv := make([]byte, 16)
	var algo security.OracleNetworkEncryption = nil
	var err error
	switch serv.algoID {
	case 0:
		return nil
	case 2:
		algo, err = security.NewOracleNetworkDESCryptor(key[:8], nil)
	case 15:
		algo, err = security.NewOracleNetworkCBCEncrypter(key[:16], nil)
	case 16:
		algo, err = security.NewOracleNetworkCBCEncrypter(key[:24], nil)
	case 17:
		algo, err = security.NewOracleNetworkCBCEncrypter(key[:32], nil)
	default:
		err = errors.New(fmt.Sprintf("advanced negotiation error: encryption service algorithm: %d still not supported", serv.algoID))
	}
	if err != nil {
		return err
	}
	serv.comm.session.Context.AdvancedService.CryptAlgo = algo
	return nil
}