mirror of
https://github.com/jambonz/freeswitch-modules.git
synced 2025-12-19 08:47:44 +00:00
8a3c001b59
* wip * #103 wip * wip * wip * support both grpc (legacy) and websockets api for aws transcribe * renaming
324 lines
11 KiB
C++
324 lines
11 KiB
C++
#include "transcribe_manager.hpp"
|
|
#include "crc.h"
|
|
|
|
#include <switch.h>
|
|
|
|
#include <openssl/sha.h>
|
|
#include <openssl/hmac.h>
|
|
#include <iomanip>
|
|
#include <regex>
|
|
#include <iostream>
|
|
#include <cstring>
|
|
#include <netinet/in.h>
|
|
|
|
using namespace std;
|
|
|
|
namespace {
|
|
std::string uri_encode(const std::string &value) {
|
|
std::string encoded;
|
|
char hex[4];
|
|
for (char c : value) {
|
|
if (isalnum(c) || c == '-' || c == '_' || c == '.' || c == '~') {
|
|
encoded += c;
|
|
} else {
|
|
sprintf(hex, "%%%02X", c);
|
|
encoded.append(hex);
|
|
}
|
|
}
|
|
return encoded;
|
|
}
|
|
}
|
|
// see
|
|
// https://docs.aws.amazon.com/transcribe/latest/dg/websocket.html#websocket-url
|
|
// https://docs.aws.amazon.com/transcribe/latest/dg/event-stream.html
|
|
|
|
void TranscribeManager::getSignedWebsocketUrl(string& host, string& path, const string& accessKey,
|
|
const string& secretKey, const string& securityToken, const string& region, const std::string& lang,
|
|
const char* vocabularyName, const char* vocabularyFilterName, const char* vocabularyFilterMethod,
|
|
const char* piiEntities, int shouldIdentifyPiiEntities, const char* languageModelName) {
|
|
string method = "GET";
|
|
string service = "transcribe";
|
|
string endpoint = "wss://transcribestreaming." + region + ".amazonaws.com";
|
|
host = "transcribestreaming." + region + ".amazonaws.com";
|
|
|
|
time_t now = time(0);
|
|
tm *gmtm = gmtime(&now);
|
|
|
|
char amzDate[21];
|
|
snprintf (amzDate, 21, "%04d%02d%02dT%02d%02d%02dZ",
|
|
1900 + gmtm->tm_year, 1 + gmtm->tm_mon, gmtm->tm_mday,
|
|
gmtm->tm_hour, gmtm->tm_min, gmtm->tm_sec);
|
|
|
|
char datestamp[9];
|
|
snprintf (datestamp, 9, "%04d%02d%02d", 1900 + gmtm->tm_year, 1 + gmtm->tm_mon, gmtm->tm_mday);
|
|
|
|
string canonical_uri = "/stream-transcription-websocket";
|
|
string canonical_headers = "host:" + host + "\n";
|
|
string signed_headers = "host";
|
|
string algorithm = "AWS4-HMAC-SHA256";
|
|
string credential_scope = string(datestamp) + "%2F" + region + "%2F" + service + "%2F" + "aws4_request";
|
|
|
|
// N.B.: The order of all of these query args are important!
|
|
// Otherwise, the signature will be invalid.
|
|
string canonical_querystring = "X-Amz-Algorithm=" + algorithm;
|
|
canonical_querystring += "&X-Amz-Credential=" + accessKey + "%2F" + credential_scope;
|
|
canonical_querystring += "&X-Amz-Date=" + string(amzDate);
|
|
canonical_querystring += "&X-Amz-Expires=300";
|
|
canonical_querystring += "&X-Amz-Security-Token=" + uri_encode(securityToken);
|
|
canonical_querystring += "&X-Amz-SignedHeaders=" + signed_headers;
|
|
|
|
if (piiEntities && shouldIdentifyPiiEntities) {
|
|
canonical_querystring += "&content-redaction-type=PII";
|
|
}
|
|
|
|
canonical_querystring += "&language-code=" + lang;
|
|
|
|
if (languageModelName) {
|
|
std::string str(languageModelName);
|
|
canonical_querystring += "&language-model-name=" + uri_encode(str);
|
|
}
|
|
|
|
canonical_querystring += "&media-encoding=pcm";
|
|
|
|
if (piiEntities) {
|
|
std::string str(piiEntities);
|
|
canonical_querystring += "&pii-entitytypes=" + uri_encode(str);
|
|
}
|
|
|
|
canonical_querystring += "&sample-rate=8000";
|
|
|
|
// custom vocabulary and filter
|
|
if (vocabularyFilterMethod) {
|
|
std::string str(vocabularyFilterMethod);
|
|
canonical_querystring += "&vocabulary-filter-method=" + str;
|
|
}
|
|
if (vocabularyFilterName) {
|
|
std::string str(vocabularyFilterName);
|
|
canonical_querystring += "&vocabulary-filter-name=" + str;
|
|
}
|
|
if (vocabularyName) {
|
|
std::string str(vocabularyName);
|
|
canonical_querystring += "&vocabulary-name=" + str;
|
|
}
|
|
|
|
string payload_hash = getSha256("");
|
|
|
|
string canonical_request = method + '\n'
|
|
+ canonical_uri + '\n'
|
|
+ canonical_querystring + '\n'
|
|
+ canonical_headers + '\n'
|
|
+ signed_headers + '\n'
|
|
+ payload_hash;
|
|
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR,"TranscribeManager::getSignedWebsocketUrl canonical_request: %s\n", canonical_request.c_str());
|
|
string string_to_sign = algorithm + "\n"
|
|
+ amzDate + "\n"
|
|
+ regex_replace(credential_scope, regex("%2F"), "/") + "\n"
|
|
+ getSha256(canonical_request);
|
|
|
|
unsigned char signing_key[SHA256_DIGEST_LENGTH];
|
|
getSignatureKey(signing_key, secretKey, datestamp, region, service);
|
|
|
|
unsigned char signatureBinary[SHA256_DIGEST_LENGTH];
|
|
getHMAC(signatureBinary, signing_key, SHA256_DIGEST_LENGTH, string_to_sign);
|
|
string signature = toHex(signatureBinary);
|
|
|
|
canonical_querystring += "&X-Amz-Signature=" + signature;
|
|
string request_url = endpoint + canonical_uri + "?" + canonical_querystring;
|
|
path = canonical_uri + "?" + canonical_querystring;
|
|
|
|
return;
|
|
}
|
|
|
|
string TranscribeManager::getSha256(string str) {
|
|
SHA256_CTX ctx;
|
|
SHA256_Init(&ctx);
|
|
SHA256_Update(&ctx, str.c_str(), str.length());
|
|
unsigned char hash[SHA256_DIGEST_LENGTH] = { 0 };
|
|
SHA256_Final(hash, &ctx);
|
|
|
|
ostringstream os;
|
|
os << hex << setfill('0');
|
|
for (int i = 0; i < SHA256_DIGEST_LENGTH; ++i) {
|
|
os << setw(2) << static_cast<unsigned int>(hash[i]);
|
|
}
|
|
|
|
return os.str();
|
|
}
|
|
|
|
void TranscribeManager::getSignatureKey(unsigned char *signatureKey, const string& secretKey,
|
|
const string& datestamp, const string& region, const string& service) {
|
|
string key = string("AWS4") + secretKey;
|
|
unsigned char kDate[SHA256_DIGEST_LENGTH];
|
|
unsigned char kRegion[SHA256_DIGEST_LENGTH];
|
|
unsigned char kService[SHA256_DIGEST_LENGTH];
|
|
unsigned char kSigning[SHA256_DIGEST_LENGTH];
|
|
getHMAC(kDate, (unsigned char *)key.c_str(), key.length(), datestamp);
|
|
getHMAC(kRegion, kDate, SHA256_DIGEST_LENGTH, region);
|
|
getHMAC(kService, kRegion, SHA256_DIGEST_LENGTH, service);
|
|
getHMAC(kSigning, kService, SHA256_DIGEST_LENGTH, "aws4_request");
|
|
|
|
memcpy(signatureKey, kSigning, SHA256_DIGEST_LENGTH);
|
|
}
|
|
|
|
void TranscribeManager::getHMAC(unsigned char *hmac, unsigned char *key, int keyLen, const string& str) {
|
|
unsigned char *data = (unsigned char*)str.c_str();
|
|
unsigned char *result = HMAC(EVP_sha256(), key, keyLen, data, strlen((char *)data), NULL, NULL);
|
|
memcpy(hmac, result, SHA256_DIGEST_LENGTH);
|
|
}
|
|
|
|
string TranscribeManager::toHex(unsigned char *hmac) {
|
|
ostringstream os;
|
|
os << hex << setfill('0');
|
|
for (int i = 0; i < SHA256_DIGEST_LENGTH; ++i) {
|
|
os << setw(2) << static_cast<unsigned int>(hmac[i]);
|
|
}
|
|
|
|
return os.str();
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
bool TranscribeManager::parseResponse(const string& response, string& payload, bool& isError, bool verbose) {
|
|
const char* buffer = response.c_str();
|
|
|
|
uint32_t totalLen;
|
|
memcpy(&totalLen, &buffer[0], sizeof(uint32_t));
|
|
totalLen = ntohl(totalLen);
|
|
|
|
uint32_t headerLen;
|
|
memcpy(&headerLen, &buffer[4], sizeof(uint32_t));
|
|
headerLen = ntohl(headerLen);
|
|
|
|
if (!verifyCRC(buffer, totalLen)) {
|
|
return false;
|
|
}
|
|
|
|
buffer += 12; // bytes 0 - 11 are prelude
|
|
|
|
const int numberOfHeaders = 3;
|
|
for (int i = 0; i < numberOfHeaders; i++) {
|
|
parseHeader(&buffer, isError, verbose);
|
|
}
|
|
|
|
payload = string(buffer, totalLen - headerLen - 4*4);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool TranscribeManager::verifyCRC(const char* buffer, const uint32_t totalLength) {
|
|
uint32_t preludeCRC;
|
|
memcpy(&preludeCRC, &buffer[8], 4);
|
|
preludeCRC = ntohl(preludeCRC);
|
|
|
|
uint32_t calculatedPreludeCRC = CRC::Calculate(&buffer[0], 8, CRC::CRC_32());
|
|
if (calculatedPreludeCRC != preludeCRC) {
|
|
cout << "Prelude CRC didn't match!" << endl;
|
|
return false;
|
|
}
|
|
|
|
uint32_t messageCRC;
|
|
memcpy(&messageCRC, &buffer[totalLength - 4], 4);
|
|
messageCRC = ntohl(messageCRC);
|
|
|
|
uint32_t calculatedMessageCRC = CRC::Calculate(buffer, totalLength - 4, CRC::CRC_32());
|
|
|
|
if (calculatedMessageCRC != messageCRC) {
|
|
cout << "Message CRC didn't match!" << endl;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void TranscribeManager::parseHeader(const char** buffer, bool& isError, bool verbose) {
|
|
uint8_t headerNameLen;
|
|
memcpy(&headerNameLen, *buffer, sizeof(uint8_t));
|
|
(*buffer)++;
|
|
|
|
string headerName(*buffer, headerNameLen);
|
|
*buffer += headerNameLen;
|
|
|
|
uint8_t headerType;
|
|
memcpy(&headerType, *buffer, sizeof(uint8_t));
|
|
(*buffer)++;
|
|
|
|
uint16_t headerValLen;
|
|
memcpy(&headerValLen, *buffer, sizeof(uint16_t));
|
|
headerValLen = ntohs(headerValLen);
|
|
*buffer += 2;
|
|
|
|
string headerVal(*buffer, headerValLen);
|
|
*buffer += headerValLen;
|
|
|
|
if (headerVal == "exception") {
|
|
isError = true;
|
|
}
|
|
if (verbose) {
|
|
cout << headerName << "(" << (int)headerType << "): " << headerVal << endl;
|
|
}
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
bool TranscribeManager::makeRequest(string& request, const vector<uint8_t>& data) {
|
|
char preludeAndCrcBuffer[4*3];
|
|
char headerBuffer[88];
|
|
char messageCrcBuffer[4];
|
|
|
|
// prelude
|
|
uint32_t totalLen = sizeof(preludeAndCrcBuffer) + sizeof(headerBuffer) + data.size() + sizeof(messageCrcBuffer);
|
|
uint32_t headerLen = sizeof(headerBuffer);
|
|
totalLen = htonl(totalLen);
|
|
headerLen = htonl(headerLen);
|
|
|
|
memcpy(&preludeAndCrcBuffer[0], &totalLen, sizeof(uint32_t));
|
|
memcpy(&preludeAndCrcBuffer[4], &headerLen, sizeof(uint32_t));
|
|
|
|
uint32_t preludeCRC = CRC::Calculate(&preludeAndCrcBuffer[0], 8, CRC::CRC_32());
|
|
preludeCRC = htonl(preludeCRC);
|
|
memcpy(&preludeAndCrcBuffer[8], &preludeCRC, sizeof(uint32_t));
|
|
|
|
// header
|
|
char* buffer = headerBuffer;
|
|
writeHeader(&buffer, ":content-type", "application/octet-stream");
|
|
writeHeader(&buffer, ":event-type", "AudioEvent");
|
|
writeHeader(&buffer, ":message-type", "event");
|
|
|
|
// write everything to response string except for the message CRC
|
|
request.append(preludeAndCrcBuffer, sizeof(preludeAndCrcBuffer));
|
|
request.append(headerBuffer, sizeof(headerBuffer));
|
|
request.append(data.begin(), data.end());
|
|
|
|
// message CRC
|
|
uint32_t messageCRC = CRC::Calculate(request.c_str(), request.length(), CRC::CRC_32());
|
|
messageCRC = htonl(messageCRC);
|
|
memcpy(messageCrcBuffer, &messageCRC, sizeof(uint32_t));
|
|
|
|
// write message CRC to response string
|
|
request.append(messageCrcBuffer, sizeof(messageCrcBuffer));
|
|
|
|
return true;
|
|
}
|
|
|
|
void TranscribeManager::writeHeader(char** buffer, const char* key, const char* val) {
|
|
uint8_t keyLen = strlen(key);
|
|
uint16_t valueLen = strlen(val);
|
|
|
|
memcpy(*buffer, &keyLen, sizeof(uint8_t));
|
|
(*buffer)++;
|
|
|
|
memcpy(*buffer, key, keyLen);
|
|
(*buffer) += keyLen;
|
|
|
|
uint8_t valueType = 7;
|
|
memcpy(*buffer, &valueType, sizeof(uint8_t));
|
|
(*buffer)++;
|
|
|
|
uint16_t valLen = htons(valueLen);
|
|
memcpy(*buffer, &valLen, sizeof(uint16_t));
|
|
(*buffer) += 2;
|
|
|
|
memcpy(*buffer, val, valueLen);
|
|
(*buffer) += valueLen;
|
|
} |