cacert-gosigner/cmd/clientsim/main.go

/*
Copyright 2022 CAcert Inc.
SPDX-License-Identifier: Apache-2.0

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

// client simulator

package main

import (
	"context"
	"fmt"
	"os"
	"sync"
	"time"

	"github.com/justincpresley/go-cobs"
	"github.com/shamaton/msgpackgen/msgpack"
	"github.com/sirupsen/logrus"

	"git.cacert.org/cacert-gosigner/pkg/messages"
)

const cobsDelimiter = 0x00

var cobsConfig = cobs.Config{SpecialByte: cobsDelimiter, Delimiter: true, EndingSave: true}

func main() {
	logger := logrus.New()
	logger.SetOutput(os.Stdout)
	logger.SetLevel(logrus.InfoLevel)

	sim := &clientSimulator{
		logger: logger,
	}

	err := sim.Run()
	if err != nil {
		logger.Errorf("simulator returned an error: %v", err)
	}
}

type clientSimulator struct {
	logger    *logrus.Logger
	commands  chan messages.Command
	responses chan []byte
}

func (c *clientSimulator) writeTestCommands(ctx context.Context) error {
	messages.RegisterGeneratedResolver()

	const healthInterval = 10 * time.Second

	timer := time.NewTimer(healthInterval)

	for {
		select {
		case <-ctx.Done():
			_ = timer.Stop()

			return nil
		case <-timer.C:
			c.commands <- messages.Command{
				Code:      messages.CmdHealth,
				TimeStamp: time.Now().UTC(),
			}

			timer.Reset(healthInterval)
		}
	}
}

func (c *clientSimulator) handleInput(ctx context.Context) error {
	const (
		bufferSize   = 1024 * 1024
		readInterval = 50 * time.Millisecond
	)

	buf := make([]byte, bufferSize)

	for {
		select {
		case <-ctx.Done():
			return nil
		default:
			count, err := os.Stdin.Read(buf)
			if err != nil {
				return fmt.Errorf("reading input failed: %w", err)
			}

			if count == 0 {
				time.Sleep(readInterval)

				continue
			}

			data := buf[:count]

			err = cobs.Verify(data, cobsConfig)
			if err != nil {
				return fmt.Errorf("frame verification failed: %w", err)
			}

			c.responses <- cobs.Decode(data, cobsConfig)
		}
	}
}

func (c *clientSimulator) handleCommands(ctx context.Context) error {
	for {
		select {
		case command := <-c.commands:
			commandBytes, err := msgpack.Marshal(command)
			if err != nil {
				return fmt.Errorf("could not marshal command bytes: %w", err)
			}

			_, err = os.Stdout.Write(cobs.Encode(commandBytes, cobsConfig))
			if err != nil {
				return fmt.Errorf("write failed: %w", err)
			}

			responseBytes := <-c.responses

			var response messages.Response

			err = msgpack.Unmarshal(responseBytes, &response)
			if err != nil {
				return fmt.Errorf("could not unmarshal msgpack data: %w", err)
			}

			c.logger.Errorf("received response: %+v", response)
		case <-ctx.Done():
			return nil
		}
	}
}

func (c *clientSimulator) Run() error {
	ctx, cancel := context.WithCancel(context.Background())

	c.commands = make(chan messages.Command)
	c.responses = make(chan []byte)

	wg := sync.WaitGroup{}
	wg.Add(2)

	var inputError, commandError error

	go func(inputErr error) {
		_ = c.handleInput(ctx)

		cancel()

		wg.Done()
	}(inputError)

	go func(commandErr error) {
		_ = c.handleCommands(ctx)

		cancel()

		wg.Done()
	}(commandError)

	var result error

	if err := c.writeTestCommands(ctx); err != nil {
		c.logger.Errorf("test commands failed: %v", err)
	}

	cancel()
	wg.Wait()

	if inputError != nil {
		c.logger.Errorf("reading input failed: %v", inputError)
	}

	if commandError != nil {
		c.logger.Errorf("sending commands failed: %v", commandError)
	}

	return result
}