Add new operations: SetClusterVersion, UpdateFirmware, SetLockPin, SetErasePin (c2c495cd) · Commits · kinetic-archive / kinetic-go

blockconnection.go

+60 −0

Original line number	Diff line number	Diff line
		@@ -157,6 +157,66 @@ func (conn *BlockConnection) UnlockDevice(pin []byte) (Status, error) {
		return conn.pinop(pin, kproto.Command_PinOperation_UNLOCK_PINOP)
		}

		func (conn *BlockConnection) UpdateFirmware(code []byte) (Status, error) {
		callback := &GenericCallback{}
		h := NewMessageHandler(callback)
		err := conn.nbc.UpdateFirmware(code, h)
		if err != nil {
		return callback.Status(), err
		}

		for callback.Done() == false {
		conn.nbc.Run()
		}

		return callback.Status(), nil
		}

		func (conn *BlockConnection) SetClusterVersion(version int64) (Status, error) {
		callback := &GenericCallback{}
		h := NewMessageHandler(callback)
		err := conn.nbc.SetClusterVersion(version, h)
		if err != nil {
		return callback.Status(), err
		}

		for callback.Done() == false {
		conn.nbc.Run()
		}

		return callback.Status(), nil
		}

		func (conn *BlockConnection) SetLockPin(currentPin []byte, newPin []byte) (Status, error) {
		callback := &GenericCallback{}
		h := NewMessageHandler(callback)
		err := conn.nbc.SetLockPin(currentPin, newPin, h)
		if err != nil {
		return callback.Status(), err
		}

		for callback.Done() == false {
		conn.nbc.Run()
		}

		return callback.Status(), nil
		}

		func (conn *BlockConnection) SetErasePin(currentPin []byte, newPin []byte) (Status, error) {
		callback := &GenericCallback{}
		h := NewMessageHandler(callback)
		err := conn.nbc.SetErasePin(currentPin, newPin, h)
		if err != nil {
		return callback.Status(), err
		}

		for callback.Done() == false {
		conn.nbc.Run()
		}

		return callback.Status(), nil
		}

		func (conn *BlockConnection) Close() {
		conn.nbc.Close()
		}

nonblockconnection.go

+61 −12

Original line number	Diff line number	Diff line
		@@ -26,8 +26,7 @@ func (conn NonBlockConnection) NoOp(h MessageHandler) error {

		cmd := newCommand(kproto.Command_NOOP)

		err := conn.service.submit(msg, cmd, nil, h)
		return err
		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) get(key []byte, getType kproto.Command_MessageType, h MessageHandler) error {
		@@ -40,8 +39,7 @@ func (conn *NonBlockConnection) get(key []byte, getType kproto.Command_MessageTy
		},
		}

		err := conn.service.submit(msg, cmd, nil, h)
		return err
		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) Get(key []byte, h MessageHandler) error {
		@@ -71,8 +69,7 @@ func (conn NonBlockConnection) GetKeyRange(r KeyRange, h *MessageHandler) erro
		},
		}

		err := conn.service.submit(msg, cmd, nil, h)
		return err
		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) Delete(entry Record, h *MessageHandler) error {
		@@ -90,8 +87,7 @@ func (conn NonBlockConnection) Delete(entry Record, h *MessageHandler) error {
		},
		}

		err := conn.service.submit(msg, cmd, nil, h)
		return err
		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) Put(entry Record, h *MessageHandler) error {
		@@ -110,8 +106,7 @@ func (conn NonBlockConnection) Put(entry Record, h *MessageHandler) error {
		},
		}

		err := conn.service.submit(msg, cmd, entry.Value, h)
		return err
		return conn.service.submit(msg, cmd, entry.Value, h)
		}

		func (conn NonBlockConnection) pinop(pin []byte, op kproto.Command_PinOperation_PinOpType, h MessageHandler) error {
		@@ -128,8 +123,7 @@ func (conn *NonBlockConnection) pinop(pin []byte, op kproto.Command_PinOperation
		},
		}

		err := conn.service.submit(msg, cmd, nil, h)
		return err
		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) SecureErase(pin []byte, h MessageHandler) error {
		@@ -149,6 +143,61 @@ func (conn NonBlockConnection) UnlockDevice(pin []byte, h MessageHandler) erro
		return conn.pinop(pin, kproto.Command_PinOperation_UNLOCK_PINOP, h)
		}

		func (conn NonBlockConnection) UpdateFirmware(code []byte, h MessageHandler) error {
		msg := newMessage(kproto.Message_HMACAUTH)
		cmd := newCommand(kproto.Command_SETUP)

		var download bool = true
		cmd.Body = &kproto.Command_Body{
		Setup: &kproto.Command_Setup{
		FirmwareDownload: &download,
		},
		}

		return conn.service.submit(msg, cmd, code, h)
		}

		func (conn NonBlockConnection) SetClusterVersion(version int64, h MessageHandler) error {
		msg := newMessage(kproto.Message_HMACAUTH)
		cmd := newCommand(kproto.Command_SETUP)

		cmd.Body = &kproto.Command_Body{
		Setup: &kproto.Command_Setup{
		NewClusterVersion: &version,
		},
		}

		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) SetLockPin(currentPin []byte, newPin []byte, h MessageHandler) error {
		msg := newMessage(kproto.Message_HMACAUTH)
		cmd := newCommand(kproto.Command_SECURITY)

		cmd.Body = &kproto.Command_Body{
		Security: &kproto.Command_Security{
		OldLockPIN: currentPin,
		NewLockPIN: newPin,
		},
		}

		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn NonBlockConnection) SetErasePin(currentPin []byte, newPin []byte, h MessageHandler) error {
		msg := newMessage(kproto.Message_HMACAUTH)
		cmd := newCommand(kproto.Command_SECURITY)

		cmd.Body = &kproto.Command_Body{
		Security: &kproto.Command_Security{
		OldErasePIN: currentPin,
		NewErasePIN: newPin,
		},
		}

		return conn.service.submit(msg, cmd, nil, h)
		}

		func (conn *NonBlockConnection) Run() error {
		return conn.service.listen()
		}