1
0
Fork 0
mirror of https://github.com/mautrix/signal.git synced 2026-05-15 05:36:53 -04:00
mautrix-signal/pkg/libsignalgo/aes256gcmsiv.go
2025-01-17 20:05:42 +02:00

89 lines
2.8 KiB
Go

// mautrix-signal - A Matrix-signal puppeting bridge.
// Copyright (C) 2023 Sumner Evans
// Copyright (C) 2025 Tulir Asokan
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package libsignalgo
/*
#include "./libsignal-ffi.h"
*/
import "C"
import "runtime"
type AES256_GCM_SIV struct {
nc noCopy
ptr *C.SignalAes256GcmSiv
}
func wrapAES256_GCM_SIV(ptr *C.SignalAes256GcmSiv) *AES256_GCM_SIV {
aes := &AES256_GCM_SIV{ptr: ptr}
runtime.SetFinalizer(aes, (*AES256_GCM_SIV).Destroy)
return aes
}
func NewAES256_GCM_SIV(key []byte) (*AES256_GCM_SIV, error) {
var aes C.SignalMutPointerAes256GcmSiv
signalFfiError := C.signal_aes256_gcm_siv_new(&aes, BytesToBuffer(key))
if signalFfiError != nil {
return nil, wrapError(signalFfiError)
}
return wrapAES256_GCM_SIV(aes.raw), nil
}
func (aes *AES256_GCM_SIV) mutPtr() C.SignalMutPointerAes256GcmSiv {
return C.SignalMutPointerAes256GcmSiv{aes.ptr}
}
func (aes *AES256_GCM_SIV) constPtr() C.SignalConstPointerAes256GcmSiv {
return C.SignalConstPointerAes256GcmSiv{aes.ptr}
}
func (aes *AES256_GCM_SIV) Destroy() error {
runtime.SetFinalizer(aes, nil)
return wrapError(C.signal_aes256_gcm_siv_destroy(C.SignalMutPointerAes256GcmSiv{raw: aes.ptr}))
}
func (aes *AES256_GCM_SIV) Encrypt(plaintext, nonce, associatedData []byte) ([]byte, error) {
var encrypted C.SignalOwnedBuffer = C.SignalOwnedBuffer{}
signalFfiError := C.signal_aes256_gcm_siv_encrypt(
&encrypted,
C.SignalConstPointerAes256GcmSiv{raw: aes.ptr},
BytesToBuffer(plaintext),
BytesToBuffer(nonce),
BytesToBuffer(associatedData),
)
runtime.KeepAlive(aes)
if signalFfiError != nil {
return nil, wrapError(signalFfiError)
}
return CopySignalOwnedBufferToBytes(encrypted), nil
}
func (aes *AES256_GCM_SIV) Decrypt(ciphertext, nonce, associatedData []byte) ([]byte, error) {
var decrypted C.SignalOwnedBuffer = C.SignalOwnedBuffer{}
signalFfiError := C.signal_aes256_gcm_siv_decrypt(
&decrypted,
C.SignalConstPointerAes256GcmSiv{raw: aes.ptr},
BytesToBuffer(ciphertext),
BytesToBuffer(nonce),
BytesToBuffer(associatedData),
)
if signalFfiError != nil {
return nil, wrapError(signalFfiError)
}
return CopySignalOwnedBufferToBytes(decrypted), nil
}