void udp_copy_exit()
{
int i;
fprintf(stderr, "exit udpcopy\n");
if (-1 != raw_sock){
close(raw_sock);
raw_sock = -1;
}
send_close();
#if (UDPCOPY_OFFLINE)
if (pcap != NULL) {
pcap_close(pcap);
}
#endif
log_end();
if (clt_settings.transfer.mappings != NULL){
for (i = 0; i < clt_settings.transfer.num; i++){
free(clt_settings.transfer.mappings[i]);
}
free(clt_settings.transfer.mappings);
clt_settings.transfer.mappings = NULL;
}
exit(EXIT_SUCCESS);
}
void tcp_copy_exit()
{
int i;
fprintf(stderr, "exit tcpcopy\n");
destroy_for_sessions();
if(-1 != raw_sock){
close(raw_sock);
raw_sock = -1;
}
send_close();
address_close_sock();
log_end();
#ifdef TCPCOPY_MYSQL_ADVANCED
release_mysql_user_pwd_info();
#endif
if(clt_settings.transfer.mappings != NULL){
for(i = 0; i < clt_settings.transfer.num; i++){
free(clt_settings.transfer.mappings[i]);
}
free(clt_settings.transfer.mappings);
clt_settings.transfer.mappings = NULL;
}
exit(EXIT_SUCCESS);
}
void read_cb(struct rpc_context *rpc, int status, void *data, void *private_data)
{
struct client *client = private_data;
COMPOUND4res *res = data;
if (status != RPC_STATUS_SUCCESS) {
fprintf(stderr, "Failed to read file on server %s\n",
client->server);
exit(10);
}
if (res->status != NFS4_OK) {
fprintf(stderr, "Failed to read file on server %s\n",
client->server);
exit(10);
}
write(1, res->resarray.resarray_val[1].nfs_resop4_u.opread.READ4res_u.resok4.data.data_val, res->resarray.resarray_val[1].nfs_resop4_u.opread.READ4res_u.resok4.data.data_len);
/*
* Are we at end-of-file? If so we can close the file and exit.
*/
if (res->resarray.resarray_val[1].nfs_resop4_u.opread.READ4res_u.resok4.eof) {
send_close(rpc, client->fh, close_cb, client);
return;
}
/*
* We still have more data to read.
*/
client->offset += res->resarray.resarray_val[1].nfs_resop4_u.opread.READ4res_u.resok4.data.data_len;
send_read(rpc, client->fh, client->offset, 4096, read_cb, client);
}
void on_T308_2(void *user) /* WAIT_REL or REL_REQ */
{
SOCKET *sock = user;
diag(COMPONENT,DIAG_WARN,"Trouble: T308_2 has expired");
if (sock->state != ss_wait_rel && sock->state != ss_rel_req)
complain(sock,"T308_2");
sock->conn_timer = NULL;
if (sock->state == ss_rel_req) send_close(sock);
free_sock(sock);
}
static void
open_file_msg_cb (GObject *source_object,
GAsyncResult *res,
gpointer user_data)
{
FilechooserPortalData *data = user_data;
GtkFileChooserNative *self = data->self;
GDBusMessage *reply;
GError *error = NULL;
reply = g_dbus_connection_send_message_with_reply_finish (data->connection, res, &error);
if (reply && g_dbus_message_to_gerror (reply, &error))
g_clear_object (&reply);
if (reply == NULL)
{
if (!data->hidden)
_gtk_native_dialog_emit_response (GTK_NATIVE_DIALOG (self), GTK_RESPONSE_DELETE_EVENT);
g_warning ("Can't open portal file chooser: %s", error->message);
g_error_free (error);
filechooser_portal_data_free (data);
self->mode_data = NULL;
return;
}
g_variant_get_child (g_dbus_message_get_body (reply), 0, "o",
&data->portal_handle);
if (data->hidden)
{
/* The dialog was hidden before we got the handle, close it now */
send_close (data);
filechooser_portal_data_free (data);
self->mode_data = NULL;
}
else
{
data->portal_response_signal_id =
g_dbus_connection_signal_subscribe (data->connection,
"org.freedesktop.portal.Desktop",
"org.freedesktop.portal.Request",
"Response",
data->portal_handle,
NULL,
G_DBUS_SIGNAL_FLAGS_NO_MATCH_RULE,
response_cb,
self, NULL);
}
g_object_unref (reply);
}
int httpd_main(int argc, char *argv[])
{
char *p = NULL, *r;
char *buffer;
int number = 0, ret;
struct netbuf *buf;
send_accept();
buffer = kz_kmalloc(DEFAULT_NETBUF_SIZE);
while (1) {
kz_recv(MSGBOX_ID_HTTPD, NULL, (void *)&buf);
switch (buf->cmd) {
case TCP_CMD_ESTAB:
number = buf->option.tcp.establish.number;
p = buffer;
break;
case TCP_CMD_CLOSE:
number = 0;
send_accept();
break;
case TCP_CMD_RECV:
memcpy(p, buf->top, buf->size);
p += buf->size;
*p = '\0';
r = strchr(buffer, '\n');
if (r) {
*r = '\0';
r++;
ret = parse(number, buffer);
memmove(buffer, r, p - r + 1);
p -= (r - buffer);
if (ret) send_close(number);
}
break;
}
kz_kmfree(buf);
}
return 0;
}
void
gtk_file_chooser_native_portal_hide (GtkFileChooserNative *self)
{
FilechooserPortalData *data = self->mode_data;
/* This is always set while dialog visible */
g_assert (data != NULL);
data->hidden = TRUE;
if (data->portal_handle)
{
send_close (data);
filechooser_portal_data_free (data);
}
self->mode_data = NULL;
}
static void tcp_copy_over(const int sig){
logInfo(LOG_WARN,"sig %d received",sig);
int total=0;
while(!isReadCompletely)
{
logInfo(LOG_WARN,"sleep one second");
sleep(1);
total++;
if(total>30)
{
break;
}
}
if(-1!=raw_sock)
{
close(raw_sock);
}
send_close();
endLogInfo();
exit(0);
}
void
tcp_copy_exit()
{
int i;
tc_event_loop_finish(&event_loop);
destroy_for_sessions();
if (-1 != raw_sock) {
close(raw_sock);
raw_sock = -1;
}
send_close();
address_close_sock();
#if (TCPCOPY_OFFLINE)
if (pcap != NULL) {
pcap_close(pcap);
}
#endif
tc_log_end();
#ifdef TCPCOPY_MYSQL_ADVANCED
release_mysql_user_pwd_info();
#endif
if (clt_settings.transfer.mappings != NULL) {
for (i = 0; i < clt_settings.transfer.num; i++) {
free(clt_settings.transfer.mappings[i]);
}
free(clt_settings.transfer.mappings);
clt_settings.transfer.mappings = NULL;
}
exit(EXIT_SUCCESS);
}
void handle_packet(apacket *p, atransport *t)
{
asocket *s;
D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
((char*) (&(p->msg.command)))[1],
((char*) (&(p->msg.command)))[2],
((char*) (&(p->msg.command)))[3]);
print_packet("recv", p);
switch(p->msg.command){
case A_SYNC:
if(p->msg.arg0){
send_packet(p, t);
if(HOST) send_connect(t);
} else {
t->connection_state = CS_OFFLINE;
handle_offline(t);
send_packet(p, t);
}
return;
case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
/* XXX verify version, etc */
if(t->connection_state != CS_OFFLINE) {
t->connection_state = CS_OFFLINE;
handle_offline(t);
}
parse_banner((char*) p->data, t);
handle_online();
if(!HOST) send_connect(t);
break;
case A_OPEN: /* OPEN(local-id, 0, "destination") */
if(t->connection_state != CS_OFFLINE) {
char *name = (char*) p->data;
name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
s = create_local_service_socket(name);
if(s == 0) {
send_close(0, p->msg.arg0, t);
} else {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
send_ready(s->id, s->peer->id, t);
s->ready(s);
}
}
break;
case A_OKAY: /* READY(local-id, remote-id, "") */
if(t->connection_state != CS_OFFLINE) {
if((s = find_local_socket(p->msg.arg1))) {
if(s->peer == 0) {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
}
s->ready(s);
}
}
break;
case A_CLSE: /* CLOSE(local-id, remote-id, "") */
if(t->connection_state != CS_OFFLINE) {
if((s = find_local_socket(p->msg.arg1))) {
s->close(s);
}
}
break;
case A_WRTE:
if(t->connection_state != CS_OFFLINE) {
if((s = find_local_socket(p->msg.arg1))) {
unsigned rid = p->msg.arg0;
p->len = p->msg.data_length;
if(s->enqueue(s, p) == 0) {
D("Enqueue the socket\n");
send_ready(s->id, rid, t);
}
return;
}
}
break;
default:
printf("handle_packet: what is %08x?!\n", p->msg.command);
}
put_apacket(p);
}
static void uni_call(SOCKET *sock,unsigned char mid)
{
char buffer[MAX_ATM_ADDR_LEN+1];
int error;
switch (mid) {
case ATM_MSG_STATUS: /* 5.5.6.12 */
{
CALL_STATE state;
/*
* NOTE: T322 isn't implemented yet, but when it is, make sure
* to only stop it on STATUS iff the cause is
* ATM_CV_RESP_STAT_ENQ. Supplementary services break if
* you stop on any STATUS.
*/
state = q_fetch(&in_dsc,QF_call_state);
if (state == cs_null) break; /* clear call */
if (sock->call_state == cs_rel_req || sock->call_state ==
cs_rel_ind) return;
if (state != sock->call_state)
diag(COMPONENT,DIAG_WARN,"STATUS %s received in state %s",
cs_name[state],cs_name[sock->call_state]);
}
return;
default:
;
}
switch (mid) {
case ATM_MSG_CALL_PROC: /* CONNECTING, WAIT_REL, REL_REQ */
if (sock->state == ss_wait_rel || sock->state == ss_rel_req) {
send_status(sock->sig,sock,0,ATM_CV_INCOMP_MSG,
ATM_MSG_CALL_PROC);
return;
}
if (sock->state != ss_connecting) break;
/* check for 2nd CALL_PROC @@@ */
STOP_TIMER(sock);
if (q_present(&in_dsc,QG_conn_id)) {
int vpci;
vpci = q_fetch(&in_dsc,QF_vpi);
sock->pvc.sap_family = AF_ATMPVC;
sock->pvc.sap_addr.itf = get_itf(sock->sig,&vpci);
sock->pvc.sap_addr.vpi = vpci;
sock->pvc.sap_addr.vci = q_fetch(&in_dsc,QF_vci);
diag(COMPONENT,DIAG_DEBUG,"ITF.VPI.VCI: %d.%d.%d",
sock->pvc.sap_addr.itf,sock->pvc.sap_addr.vpi,
sock->pvc.sap_addr.vci);
}
START_TIMER(sock,T310);
sock->call_state = cs_out_proc;
return;
case ATM_MSG_CONNECT: /* CONNECTING, REL_REQ */
if (sock->state == ss_rel_req) {
send_status(sock->sig,sock,0,ATM_CV_INCOMP_MSG,ATM_MSG_CONNECT);
return;
}
if (sock->state != ss_connecting) break;
STOP_TIMER(sock);
if (q_present(&in_dsc,QG_conn_id)) {
int vpci;
vpci = q_fetch(&in_dsc,QF_vpi);
sock->pvc.sap_family = AF_ATMPVC;
sock->pvc.sap_addr.itf = get_itf(sock->sig,&vpci);
sock->pvc.sap_addr.vpi = vpci;
sock->pvc.sap_addr.vci = q_fetch(&in_dsc,QF_vci);
diag(COMPONENT,DIAG_DEBUG,"ITF.VPI.VCI: %d/%d.%d",
sock->pvc.sap_addr.itf,sock->pvc.sap_addr.vpi,
sock->pvc.sap_addr.vci);
}
error = 0;
if (!sock->pvc.sap_addr.vpi && !sock->pvc.sap_addr.vci)
error = -EPROTO;
/* more problems */
if (error) {
set_error(sock,error);
send_release(sock,0); /* @@@ cause follows reason ??? */
START_TIMER(sock,T308_1);
new_state(sock,ss_rel_req);
return;
}
send_connect_ack(sock);
/* @@@ fill in sock->remote */
/* @@@ fill in traffic parameters */
send_kernel(sock->id,kptr_null,as_okay,0,&sock->pvc,NULL,
&sock->local,&sock->sap,&sock->qos);
new_state(sock,ss_connected);
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
sock->owner = 1;
#endif
if (atm2text(buffer,MAX_ATM_ADDR_LEN+1,(struct sockaddr *)
&sock->remote,0) < 0) strcpy(buffer,"<invalid>");
diag(COMPONENT,DIAG_INFO,"Active open succeeded (CR 0x%06X, "
"ID %s, to %s)",sock->call_ref,kptr_print(&sock->id),buffer);
return;
case ATM_MSG_CONN_ACK: /* ACCEPTING, WAIT_REL, REL_REQ */
diag(COMPONENT,DIAG_DEBUG,"CA vpi.vci=%d.%d",
sock->pvc.sap_addr.vpi,sock->pvc.sap_addr.vci);
if (sock->state == ss_wait_rel || sock->state == ss_rel_req) {
send_status(sock->sig,sock,0,ATM_CV_INCOMP_MSG,
ATM_MSG_CONN_ACK);
return;
}
if (sock->state != ss_accepting) break;
STOP_TIMER(sock);
send_kernel(sock->id,kptr_null,as_okay,0,NULL,NULL,&sock->local,
&sock->sap,NULL);
new_state(sock,ss_connected);
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
sock->owner = 0;
#endif
if (atm2text(buffer,MAX_ATM_ADDR_LEN+1, (struct sockaddr *)
&sock->remote,0) < 0) strcpy(buffer,"<invalid>");
diag(COMPONENT,DIAG_INFO,"Passive open succeeded (CR 0x%06X, "
"ID %s, from %s)",sock->call_ref,kptr_print(&sock->id),buffer);
return;
case ATM_MSG_RELEASE: /* all states */
{
unsigned char cause;
cause = q_fetch(&in_dsc,QF_cause);
diag(COMPONENT,DIAG_DEBUG,"Cause %d (%s)",cause,cause > 127 ?
"invalid cause" : cause_text[cause]);
}
switch (sock->state) {
case ss_connecting:
set_error(sock,-ECONNREFUSED);
/* fall through */
case ss_accepting:
set_error(sock,-ECONNRESET); /* ERESTARTSYS ? */
send_release_complete(sock->sig,sock->call_ref,0);
SEND_ERROR(sock->id,sock->error);
STOP_TIMER(sock);
free_sock(sock);
return;
case ss_rel_req:
send_close(sock);
/* fall through */
case ss_wait_rel:
STOP_TIMER(sock);
free_sock(sock);
return;
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
case ss_mod_req:
#endif
STOP_TIMER(sock);
/* fall through */
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
case ss_mod_lcl:
case ss_mod_rcv:
case ss_mod_fin_ok:
case ss_mod_fin_fail:
case ss_mod_fin_ack:
#endif
case ss_connected:
diag(COMPONENT,DIAG_INFO,"Passive close (CR 0x%06X)",
sock->call_ref);
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
if (timer_handler(sock->conn_timer) == on_T361)
STOP_TIMER(sock);
#endif
send_close(sock);
new_state(sock,ss_rel_ind);
return;
case ss_indicated:
/* fall through */
case ss_proceeding:
send_release_complete(sock->sig,sock->call_ref,0);
new_state(sock,ss_zombie);
/* fall through */
case ss_rel_ind:
return;
default:
send_release_complete(sock->sig,sock->call_ref,0);
/* @@@ should be ATM_CV_INCOMP_MSG */
break;
}
break;
case ATM_MSG_RESTART:
set_error(sock,-ENETRESET);
/* fall through */
case ATM_MSG_STATUS: /* fall through when clearing */
case ATM_MSG_REL_COMP: /* basically any state (except LISTENING and
ZOMBIE) */
{
unsigned char cause;
if (mid != ATM_MSG_REL_COMP || !q_present(&in_dsc,QF_cause))
cause = 0;
else {
cause = q_fetch(&in_dsc,QF_cause);
diag(COMPONENT,DIAG_DEBUG,"Cause %d (%s)",cause,
cause > 127 ? "invalid cause" : cause_text[cause]);
}
switch (sock->state) {
case ss_connecting:
set_error(sock,cause == ATM_CV_UNALLOC ?
-EADDRNOTAVAIL : cause == ATM_CV_RES_UNAVAIL ||
#if defined(UNI31) || defined(UNI40) || defined(DYNAMIC_UNI)
cause == ATM_CV_UCR_UNAVAIL_NEW ||
#endif
cause == ATM_CV_NO_ROUTE_DEST ? -EHOSTUNREACH :
cause == ATM_CV_NUM_CHANGED ? -EREMCHG :
cause == ATM_CV_DEST_OOO ? -EHOSTDOWN :
-ECONNREFUSED);
/* fall through */
case ss_accepting:
set_error(sock,-ECONNRESET); /* ERESTARTSYS ? */
SEND_ERROR(sock->id,sock->error);
STOP_TIMER(sock);
free_sock(sock);
return;
case ss_rel_req:
send_close(sock);
/* fall through */
case ss_wait_rel:
STOP_TIMER(sock);
free_sock(sock);
return;
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
case ss_mod_req:
#endif
STOP_TIMER(sock);
/* fall through */
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
case ss_mod_lcl:
case ss_mod_rcv:
case ss_mod_fin_ok:
case ss_mod_fin_fail:
case ss_mod_fin_ack:
#endif
case ss_connected:
diag(COMPONENT,DIAG_INFO,"Passive close (CR 0x%06X)",
sock->call_ref);
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
if (timer_handler(sock->conn_timer) == on_T361)
STOP_TIMER(sock);
#endif
send_close(sock);
/* fall through */
case ss_rel_ind:
new_state(sock,ss_wait_close);
return;
case ss_indicated:
/* fall through */
case ss_proceeding:
new_state(sock,ss_zombie);
return;
default:
break;
}
break; /* fail */
}
case ATM_MSG_ALERTING:
/*
* We basically ignore this junk message, except for the connection
* identifier it may carry.
*/
if (q_present(&in_dsc,QG_conn_id)) {
int vpci;
vpci = q_fetch(&in_dsc,QF_vpi);
sock->pvc.sap_family = AF_ATMPVC;
sock->pvc.sap_addr.itf = get_itf(sock->sig,&vpci);
sock->pvc.sap_addr.vpi = vpci;
sock->pvc.sap_addr.vci = q_fetch(&in_dsc,QF_vci);
diag(COMPONENT,DIAG_DEBUG,"ITF.VPI.VCI: %d.%d.%d",
sock->pvc.sap_addr.itf,sock->pvc.sap_addr.vpi,
sock->pvc.sap_addr.vci);
}
return;
case ATM_MSG_NOTIFY:
/* silently ignore this junk */
return;
#if defined(Q2963_1) || defined(DYNAMIC_UNI)
/*
* Buglet ahead: should actually test "call_state"
*/
case ATM_MSG_MODIFY_REQ:
if (!(sock->sig->uni & S_Q2963_1)) goto _default;
if (sock->state != ss_connected || sock->owner) break;
sock->new_qos = sock->qos;
if (q_present(&in_dsc,QF_fw_pcr_01))
sock->new_qos.rxtp.max_pcr = q_fetch(&in_dsc,QF_fw_pcr_01);
if (q_present(&in_dsc,QF_bw_pcr_01))
sock->new_qos.txtp.max_pcr = q_fetch(&in_dsc,QF_bw_pcr_01);
send_kernel(sock->id,kptr_null,as_modify,
ATM_MF_INC_RSV | ATM_MF_DEC_RSV | ATM_MF_DEC_SHP,
NULL,NULL,NULL,NULL,&sock->new_qos);
new_state(sock,ss_mod_rcv);
return;
case ATM_MSG_MODIFY_ACK:
if (!(sock->sig->uni & S_Q2963_1)) goto _default;
if (sock->state != ss_mod_req) break;
STOP_TIMER(sock);
sock->qos = sock->new_qos;
if (q_present(&in_dsc,QG_bbrt)) send_conn_avail(sock);
send_kernel(sock->id,kptr_null,as_modify,ATM_MF_SET,NULL,NULL,NULL,
NULL,&sock->qos);
new_state(sock,ss_mod_fin_ok);
return;
case ATM_MSG_MODIFY_REJ:
if (!(sock->sig->uni & S_Q2963_1)) goto _default;
if (sock->state != ss_mod_req) break;
STOP_TIMER(sock);
sock->error = -EAGAIN;
send_kernel(sock->id,kptr_null,as_modify,ATM_MF_SET,NULL,NULL,NULL,
NULL,&sock->qos);
new_state(sock,ss_mod_fin_fail);
return;
case ATM_MSG_CONN_AVAIL:
if (!(sock->sig->uni & S_Q2963_1)) goto _default;
if (sock->state != ss_connected || sock->owner) break;
STOP_TIMER(sock);
send_kernel(sock->id,kptr_null,as_modify,ATM_MF_SET,NULL,NULL,NULL,
NULL,&sock->qos);
new_state(sock,ss_mod_fin_ack);
return;
_default: /* jump here if we don't want to understand a message */
#endif
default:
diag(COMPONENT,DIAG_WARN,"Bad signaling message %d",mid);
send_status(sock->sig,sock,0,ATM_CV_UNKNOWN_MSG_TYPE,mid);
return;
}
diag(COMPONENT,DIAG_WARN,
"Signaling message %s is incompatible with state %s/%s (%d?%d)",
mid2name(mid),state_name[sock->state],cs_name[sock->call_state],
(int) sock->state,(int) sock->call_state);
send_status(sock->sig,sock,0,ATM_CV_INCOMP_MSG,mid);
}
static void exit_tcp_copy(){
close(raw_sock);
raw_sock=-1;
send_close();
exit(0);
}
void handle_packet(apacket *p, atransport *t)
{
asocket *s;
D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
((char*) (&(p->msg.command)))[1],
((char*) (&(p->msg.command)))[2],
((char*) (&(p->msg.command)))[3]);
print_packet("recv", p);
switch(p->msg.command){
case A_SYNC:
if(p->msg.arg0){
send_packet(p, t);
if(HOST) send_connect(t);
} else {
t->connection_state = CS_OFFLINE;
handle_offline(t);
send_packet(p, t);
}
return;
case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
/* XXX verify version, etc */
if(t->connection_state != CS_OFFLINE) {
t->connection_state = CS_OFFLINE;
handle_offline(t);
}
parse_banner(reinterpret_cast<const char*>(p->data), t);
if (HOST || !auth_required) {
handle_online(t);
if (!HOST) send_connect(t);
} else {
send_auth_request(t);
}
break;
case A_AUTH:
if (p->msg.arg0 == ADB_AUTH_TOKEN) {
t->connection_state = CS_UNAUTHORIZED;
t->key = adb_auth_nextkey(t->key);
if (t->key) {
send_auth_response(p->data, p->msg.data_length, t);
} else {
/* No more private keys to try, send the public key */
send_auth_publickey(t);
}
} else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
adb_auth_verified(t);
t->failed_auth_attempts = 0;
} else {
if (t->failed_auth_attempts++ > 10)
adb_sleep_ms(1000);
send_auth_request(t);
}
} else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
adb_auth_confirm_key(p->data, p->msg.data_length, t);
}
break;
case A_OPEN: /* OPEN(local-id, 0, "destination") */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
char *name = (char*) p->data;
name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
s = create_local_service_socket(name);
if(s == 0) {
send_close(0, p->msg.arg0, t);
} else {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
send_ready(s->id, s->peer->id, t);
s->ready(s);
}
}
break;
case A_OKAY: /* READY(local-id, remote-id, "") */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
if((s = find_local_socket(p->msg.arg1, 0))) {
if(s->peer == 0) {
/* On first READY message, create the connection. */
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
s->ready(s);
} else if (s->peer->id == p->msg.arg0) {
/* Other READY messages must use the same local-id */
s->ready(s);
} else {
D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s\n",
p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
}
}
}
break;
case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
if (t->online && p->msg.arg1 != 0) {
if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
/* According to protocol.txt, p->msg.arg0 might be 0 to indicate
* a failed OPEN only. However, due to a bug in previous ADB
* versions, CLOSE(0, remote-id, "") was also used for normal
* CLOSE() operations.
*
* This is bad because it means a compromised adbd could
* send packets to close connections between the host and
* other devices. To avoid this, only allow this if the local
* socket has a peer on the same transport.
*/
if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s\n",
p->msg.arg1, t->serial, s->peer->transport->serial);
} else {
s->close(s);
}
}
}
break;
case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
unsigned rid = p->msg.arg0;
p->len = p->msg.data_length;
if(s->enqueue(s, p) == 0) {
D("Enqueue the socket\n");
send_ready(s->id, rid, t);
}
return;
}
}
break;
default:
printf("handle_packet: what is %08x?!\n", p->msg.command);
}
put_apacket(p);
}
void handle_packet(apacket *p, atransport *t)
{
asocket *s;
D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
((char*) (&(p->msg.command)))[1],
((char*) (&(p->msg.command)))[2],
((char*) (&(p->msg.command)))[3]);
print_packet("recv", p);
switch(p->msg.command){
case A_SYNC:
if(p->msg.arg0){
send_packet(p, t);
if(HOST) send_connect(t);
} else {
t->connection_state = CS_OFFLINE;
handle_offline(t);
send_packet(p, t);
}
return;
case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
/* XXX verify version, etc */
if(t->connection_state != CS_OFFLINE) {
t->connection_state = CS_OFFLINE;
handle_offline(t);
}
parse_banner((char*) p->data, t);
if (HOST || !auth_enabled) {
handle_online(t);
if(!HOST) send_connect(t);
} else {
#ifndef NO_AUTH
send_auth_request(t);
#endif
}
break;
#ifndef NO_AUTH
case A_AUTH:
if (p->msg.arg0 == ADB_AUTH_TOKEN) {
t->key = adb_auth_nextkey(t->key);
if (t->key) {
send_auth_response(p->data, p->msg.data_length, t);
} else {
/* No more private keys to try, send the public key */
send_auth_publickey(t);
}
} else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
adb_auth_verified(t);
t->failed_auth_attempts = 0;
} else {
if (t->failed_auth_attempts++ > 10)
adb_sleep_ms(1000);
send_auth_request(t);
}
} else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
adb_auth_confirm_key(p->data, p->msg.data_length, t);
}
break;
#endif
case A_OPEN: /* OPEN(local-id, 0, "destination") */
if (t->online) {
char *name = (char*) p->data;
name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
s = create_local_service_socket(name);
if(s == 0) {
send_close(0, p->msg.arg0, t);
} else {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
send_ready(s->id, s->peer->id, t);
s->ready(s);
}
}
break;
case A_OKAY: /* READY(local-id, remote-id, "") */
if (t->online) {
if((s = find_local_socket(p->msg.arg1))) {
if(s->peer == 0) {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
}
s->ready(s);
}
}
break;
case A_CLSE: /* CLOSE(local-id, remote-id, "") */
if (t->online) {
D("CLOSE(%d, %d, \"\")\n", p->msg.arg0, p->msg.arg1);
if((s = find_local_socket(p->msg.arg1))) {
s->close(s);
}
}
break;
case A_WRTE:
if (t->online) {
if((s = find_local_socket(p->msg.arg1))) {
unsigned rid = p->msg.arg0;
p->len = p->msg.data_length;
if(s->enqueue(s, p) == 0) {
D("Enqueue the socket\n");
send_ready(s->id, rid, t);
}
return;
}
}
break;
default:
printf("handle_packet: what is %08x?!\n", p->msg.command);
}
put_apacket(p);
}
void handle_packet(apacket *p, atransport *t)
{
D("handle_packet() %c%c%c%c", ((char*) (&(p->msg.command)))[0],
((char*) (&(p->msg.command)))[1],
((char*) (&(p->msg.command)))[2],
((char*) (&(p->msg.command)))[3]);
print_packet("recv", p);
switch(p->msg.command){
case A_SYNC:
if (p->msg.arg0){
send_packet(p, t);
#if ADB_HOST
send_connect(t);
#endif
} else {
t->connection_state = kCsOffline;
handle_offline(t);
send_packet(p, t);
}
return;
case A_CNXN: // CONNECT(version, maxdata, "system-id-string")
handle_new_connection(t, p);
break;
case A_AUTH:
if (p->msg.arg0 == ADB_AUTH_TOKEN) {
t->connection_state = kCsUnauthorized;
send_auth_response(p->data, p->msg.data_length, t);
} else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
if (adb_auth_verify(t->token, sizeof(t->token), p->data, p->msg.data_length)) {
adb_auth_verified(t);
t->failed_auth_attempts = 0;
} else {
if (t->failed_auth_attempts++ > 256) adb_sleep_ms(1000);
send_auth_request(t);
}
} else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
adb_auth_confirm_key(p->data, p->msg.data_length, t);
}
break;
case A_OPEN: /* OPEN(local-id, 0, "destination") */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
char *name = (char*) p->data;
name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
asocket* s = create_local_service_socket(name, t);
if (s == nullptr) {
send_close(0, p->msg.arg0, t);
} else {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
send_ready(s->id, s->peer->id, t);
s->ready(s);
}
}
break;
case A_OKAY: /* READY(local-id, remote-id, "") */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
asocket* s = find_local_socket(p->msg.arg1, 0);
if (s) {
if(s->peer == 0) {
/* On first READY message, create the connection. */
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
s->ready(s);
} else if (s->peer->id == p->msg.arg0) {
/* Other READY messages must use the same local-id */
s->ready(s);
} else {
D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s",
p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
}
} else {
// When receiving A_OKAY from device for A_OPEN request, the host server may
// have closed the local socket because of client disconnection. Then we need
// to send A_CLSE back to device to close the service on device.
send_close(p->msg.arg1, p->msg.arg0, t);
}
}
break;
case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
if (t->online && p->msg.arg1 != 0) {
asocket* s = find_local_socket(p->msg.arg1, p->msg.arg0);
if (s) {
/* According to protocol.txt, p->msg.arg0 might be 0 to indicate
* a failed OPEN only. However, due to a bug in previous ADB
* versions, CLOSE(0, remote-id, "") was also used for normal
* CLOSE() operations.
*
* This is bad because it means a compromised adbd could
* send packets to close connections between the host and
* other devices. To avoid this, only allow this if the local
* socket has a peer on the same transport.
*/
if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s",
p->msg.arg1, t->serial, s->peer->transport->serial);
} else {
s->close(s);
}
}
}
break;
case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
asocket* s = find_local_socket(p->msg.arg1, p->msg.arg0);
if (s) {
unsigned rid = p->msg.arg0;
p->len = p->msg.data_length;
if (s->enqueue(s, p) == 0) {
D("Enqueue the socket");
send_ready(s->id, rid, t);
}
return;
}
}
break;
default:
printf("handle_packet: what is %08x?!\n", p->msg.command);
}
put_apacket(p);
}
void handle_packet(apacket *p, atransport *t)
{
asocket *s;
D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
((char*) (&(p->msg.command)))[1],
((char*) (&(p->msg.command)))[2],
((char*) (&(p->msg.command)))[3]);
print_packet("recv", p);
switch(p->msg.command){
case A_SYNC:
if(p->msg.arg0){
send_packet(p, t);
if(HOST) send_connect(t);
} else {
t->connection_state = CS_OFFLINE;
handle_offline(t);
send_packet(p, t);
}
return;
case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
/* XXX verify version, etc */
if(t->connection_state != CS_OFFLINE) {
t->connection_state = CS_OFFLINE;
handle_offline(t);
}
parse_banner((char*) p->data, t);
handle_online(t);
send_connect(t);
break;
case A_OPEN: /* OPEN(local-id, 0, "destination") */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
char *name = (char*) p->data;
name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
s = create_local_service_socket(name);
if(s == 0) {
send_close(0, p->msg.arg0, t);
} else {
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
send_ready(s->id, s->peer->id, t);
s->ready(s);
}
}
break;
case A_OKAY: /* READY(local-id, remote-id, "") */
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
if((s = find_local_socket(p->msg.arg1, 0))) {
if(s->peer == 0) {
/* On first READY message, create the connection. */
s->peer = create_remote_socket(p->msg.arg0, t);
s->peer->peer = s;
s->ready(s);
} else if (s->peer->id == p->msg.arg0) {
/* Other READY messages must use the same local-id */
s->ready(s);
} else {
D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s\n",
p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
}
}
}
break;
case A_CLSE: /* CLOSE(local-id, remote-id, "") */
if (t->online && p->msg.arg1 != 0) {
if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
/* According to protocol.txt, p->msg.arg0 might be 0 to indicate
* a failed OPEN only. However, due to a bug in previous ADB
* versions, CLOSE(0, remote-id, "") was also used for normal
* CLOSE() operations.
*
* This is bad because it means a compromised adbd could
* send packets to close connections between the host and
* other devices. To avoid this, only allow this if the local
* socket has a peer on the same transport.
*/
if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s\n",
p->msg.arg1, t->serial, s->peer->transport->serial);
} else {
s->close(s);
}
}
}
break;
case A_WRTE:
if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
unsigned rid = p->msg.arg0;
p->len = p->msg.data_length;
if(s->enqueue(s, p) == 0) {
D("Enqueue the socket\n");
send_ready(s->id, rid, t);
}
return;
}
}
break;
default:
printf("handle_packet: what is %08x?!\n", p->msg.command);
}
put_apacket(p);
}
