You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							123 lines
						
					
					
						
							3.9 KiB
						
					
					
				
			
		
		
	
	
							123 lines
						
					
					
						
							3.9 KiB
						
					
					
				#include "selfdrive/ui/qt/api.h"
 | 
						|
 | 
						|
#include <openssl/bio.h>
 | 
						|
#include <openssl/pem.h>
 | 
						|
#include <openssl/rsa.h>
 | 
						|
 | 
						|
#include <QCryptographicHash>
 | 
						|
#include <QDateTime>
 | 
						|
#include <QDebug>
 | 
						|
#include <QFile>
 | 
						|
#include <QJsonDocument>
 | 
						|
#include <QNetworkRequest>
 | 
						|
 | 
						|
#include "selfdrive/common/params.h"
 | 
						|
#include "selfdrive/common/util.h"
 | 
						|
#include "selfdrive/hardware/hw.h"
 | 
						|
 | 
						|
namespace CommaApi {
 | 
						|
 | 
						|
const std::string private_key_path =
 | 
						|
    Hardware::PC() ? util::getenv_default("HOME", "/.comma/persist/comma/id_rsa", "/persist/comma/id_rsa")
 | 
						|
                   : "/persist/comma/id_rsa";
 | 
						|
 | 
						|
QByteArray rsa_sign(const QByteArray &data) {
 | 
						|
  auto file = QFile(private_key_path.c_str());
 | 
						|
  if (!file.open(QIODevice::ReadOnly)) {
 | 
						|
    qDebug() << "No RSA private key found, please run manager.py or registration.py";
 | 
						|
    return QByteArray();
 | 
						|
  }
 | 
						|
  auto key = file.readAll();
 | 
						|
  file.close();
 | 
						|
  file.deleteLater();
 | 
						|
  BIO* mem = BIO_new_mem_buf(key.data(), key.size());
 | 
						|
  assert(mem);
 | 
						|
  RSA* rsa_private = PEM_read_bio_RSAPrivateKey(mem, NULL, NULL, NULL);
 | 
						|
  assert(rsa_private);
 | 
						|
  auto sig = QByteArray();
 | 
						|
  sig.resize(RSA_size(rsa_private));
 | 
						|
  unsigned int sig_len;
 | 
						|
  int ret = RSA_sign(NID_sha256, (unsigned char*)data.data(), data.size(), (unsigned char*)sig.data(), &sig_len, rsa_private);
 | 
						|
  assert(ret == 1);
 | 
						|
  assert(sig_len == sig.size());
 | 
						|
  BIO_free(mem);
 | 
						|
  RSA_free(rsa_private);
 | 
						|
  return sig;
 | 
						|
}
 | 
						|
 | 
						|
QString create_jwt(const QJsonObject &payloads, int expiry) {
 | 
						|
  QJsonObject header = {{"alg", "RS256"}};
 | 
						|
 | 
						|
  QString dongle_id = QString::fromStdString(Params().get("DongleId"));
 | 
						|
  auto t = QDateTime::currentSecsSinceEpoch();
 | 
						|
  QJsonObject payload = {{"identity", dongle_id}, {"nbf", t}, {"iat", t}, {"exp", t + expiry}};
 | 
						|
  for (auto it = payloads.begin(); it != payloads.end(); ++it) {
 | 
						|
    payload.insert(it.key(), it.value());
 | 
						|
  }
 | 
						|
 | 
						|
  auto b64_opts = QByteArray::Base64UrlEncoding | QByteArray::OmitTrailingEquals;
 | 
						|
  QString jwt = QJsonDocument(header).toJson(QJsonDocument::Compact).toBase64(b64_opts) + '.' +
 | 
						|
                QJsonDocument(payload).toJson(QJsonDocument::Compact).toBase64(b64_opts);
 | 
						|
 | 
						|
  auto hash = QCryptographicHash::hash(jwt.toUtf8(), QCryptographicHash::Sha256);
 | 
						|
  auto sig = rsa_sign(hash);
 | 
						|
  jwt += '.' + sig.toBase64(b64_opts);
 | 
						|
  return jwt;
 | 
						|
}
 | 
						|
 | 
						|
}  // namespace CommaApi
 | 
						|
 | 
						|
HttpRequest::HttpRequest(QObject *parent, const QString &requestURL, bool create_jwt_) : create_jwt(create_jwt_), QObject(parent) {
 | 
						|
  networkAccessManager = new QNetworkAccessManager(this);
 | 
						|
  reply = NULL;
 | 
						|
 | 
						|
  networkTimer = new QTimer(this);
 | 
						|
  networkTimer->setSingleShot(true);
 | 
						|
  networkTimer->setInterval(20000);
 | 
						|
  connect(networkTimer, &QTimer::timeout, this, &HttpRequest::requestTimeout);
 | 
						|
 | 
						|
  sendRequest(requestURL);
 | 
						|
}
 | 
						|
 | 
						|
void HttpRequest::sendRequest(const QString &requestURL) {
 | 
						|
  QString token;
 | 
						|
  if(create_jwt) {
 | 
						|
    token = CommaApi::create_jwt();
 | 
						|
  } else {
 | 
						|
    QString token_json = QString::fromStdString(util::read_file(util::getenv_default("HOME", "/.comma/auth.json", "/.comma/auth.json")));
 | 
						|
    QJsonDocument json_d = QJsonDocument::fromJson(token_json.toUtf8());
 | 
						|
    token = json_d["access_token"].toString();
 | 
						|
  }
 | 
						|
 | 
						|
  QNetworkRequest request;
 | 
						|
  request.setUrl(QUrl(requestURL));
 | 
						|
  request.setRawHeader(QByteArray("Authorization"), ("JWT " + token).toUtf8());
 | 
						|
 | 
						|
  reply = networkAccessManager->get(request);
 | 
						|
 | 
						|
  networkTimer->start();
 | 
						|
  connect(reply, &QNetworkReply::finished, this, &HttpRequest::requestFinished);
 | 
						|
}
 | 
						|
 | 
						|
void HttpRequest::requestTimeout() {
 | 
						|
  reply->abort();
 | 
						|
}
 | 
						|
 | 
						|
// This function should always emit something
 | 
						|
void HttpRequest::requestFinished() {
 | 
						|
  if (reply->error() != QNetworkReply::OperationCanceledError) {
 | 
						|
    networkTimer->stop();
 | 
						|
    QString response = reply->readAll();
 | 
						|
 | 
						|
    if (reply->error() == QNetworkReply::NoError) {
 | 
						|
      emit receivedResponse(response);
 | 
						|
    } else {
 | 
						|
      qDebug() << reply->errorString();
 | 
						|
      emit failedResponse(reply->errorString());
 | 
						|
    }
 | 
						|
  } else {
 | 
						|
    emit timeoutResponse("timeout");
 | 
						|
  }
 | 
						|
  reply->deleteLater();
 | 
						|
  reply = NULL;
 | 
						|
}
 | 
						|
 |