99 lines
2.3 KiB
C++
99 lines
2.3 KiB
C++
/*
|
|
* Copyright (C) 2015 The Android Open Source Project
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#define TRACE_TAG ADB
|
|
|
|
#include "sysdeps.h"
|
|
#include "adb_auth.h"
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "adb.h"
|
|
#include "transport.h"
|
|
|
|
bool auth_required = true;
|
|
|
|
void send_auth_request(atransport *t)
|
|
{
|
|
D("Calling send_auth_request");
|
|
apacket *p;
|
|
int ret;
|
|
|
|
ret = adb_auth_generate_token(t->token, sizeof(t->token));
|
|
if (ret != sizeof(t->token)) {
|
|
D("Error generating token ret=%d", ret);
|
|
return;
|
|
}
|
|
|
|
p = get_apacket();
|
|
memcpy(p->data, t->token, ret);
|
|
p->msg.command = A_AUTH;
|
|
p->msg.arg0 = ADB_AUTH_TOKEN;
|
|
p->msg.data_length = ret;
|
|
send_packet(p, t);
|
|
}
|
|
|
|
void send_auth_response(uint8_t *token, size_t token_size, atransport *t)
|
|
{
|
|
D("Calling send_auth_response");
|
|
apacket *p = get_apacket();
|
|
int ret;
|
|
|
|
ret = adb_auth_sign(t->key, token, token_size, p->data);
|
|
if (!ret) {
|
|
D("Error signing the token");
|
|
put_apacket(p);
|
|
return;
|
|
}
|
|
|
|
p->msg.command = A_AUTH;
|
|
p->msg.arg0 = ADB_AUTH_SIGNATURE;
|
|
p->msg.data_length = ret;
|
|
send_packet(p, t);
|
|
}
|
|
|
|
void send_auth_publickey(atransport *t)
|
|
{
|
|
D("Calling send_auth_publickey");
|
|
std::string key = adb_auth_get_userkey();
|
|
if (key.empty()) {
|
|
D("Failed to get user public key");
|
|
return;
|
|
}
|
|
|
|
if (key.size() >= MAX_PAYLOAD_V1) {
|
|
D("User public key too large (%zu B)", key.size());
|
|
return;
|
|
}
|
|
|
|
apacket* p = get_apacket();
|
|
memcpy(p->data, key.c_str(), key.size() + 1);
|
|
|
|
p->msg.command = A_AUTH;
|
|
p->msg.arg0 = ADB_AUTH_RSAPUBLICKEY;
|
|
p->msg.data_length = key.size();
|
|
send_packet(p, t);
|
|
}
|
|
|
|
void adb_auth_verified(atransport *t)
|
|
{
|
|
handle_online(t);
|
|
send_connect(t);
|
|
}
|