44 lines
897 B
Go
44 lines
897 B
Go
package server
|
|
|
|
import (
|
|
"crypto/ecdh"
|
|
"encoding/json"
|
|
"reflect"
|
|
"time"
|
|
|
|
"github.com/AYM1607/ccclip/internal/db"
|
|
"github.com/AYM1607/ccclip/pkg/crypto"
|
|
)
|
|
|
|
type AuthenticatedPayload struct {
|
|
DeviceID string `json:"deviceID"`
|
|
Payload []byte `json:"payload"`
|
|
}
|
|
|
|
type FingerPrint struct {
|
|
Timestamp time.Time `json:"timestamp"`
|
|
}
|
|
|
|
func decryptAuthenticatedPayload[T any](p AuthenticatedPayload, d db.DB, pk *ecdh.PrivateKey) (T, error) {
|
|
var res T
|
|
var zero T
|
|
|
|
_T := reflect.TypeOf(res)
|
|
if _T.Kind() == reflect.Pointer {
|
|
res = reflect.New(_T.Elem()).Interface().(T)
|
|
}
|
|
|
|
device, err := d.GetDevice(p.DeviceID)
|
|
if err != nil {
|
|
return zero, err
|
|
}
|
|
|
|
key := crypto.NewSharedKey(pk, crypto.PublicKeyFromBytes(device.PublicKey), crypto.ReceiveDirection)
|
|
plain := crypto.Decrypt(key, p.Payload)
|
|
|
|
err = json.Unmarshal(plain, res)
|
|
if err != nil {
|
|
return zero, err
|
|
}
|
|
return res, nil
|
|
}
|