Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 6a29c516 authored by Luke Huang's avatar Luke Huang Committed by Automerger Merge Worker
Browse files

Merge "Fix the wrong condition statement in tokio select for doh_handler and... 

Merge "Fix the wrong condition statement in tokio select for doh_handler and correct some return code in doh_query" am: 2575a72d am: 5d725143 am: 558e8ede am: 8ad00fc9

Original change: https://android-review.googlesource.com/c/platform/packages/modules/DnsResolver/+/1815038

Change-Id: Iad585228477da9028be360cd34372afb31cb73ad
parents 95bbb16b 8ad00fc9

doh.rs

+17 −6
Original line number Diff line number Diff line
@@ -357,7 +357,6 @@ impl DohConnection { 
                    }

                }

            }

            // TODO: clean up the expired queries.

            self.recv_rx().await?;

            self.flush_tx().await?;

            if let Ok((stream_id, buf)) = self.recv_query() {
@@ -679,6 +678,19 @@ async fn handle_query_cmd( 
        let _ = resp.send(Response::Error { error: QueryError::ServerNotReady });

    }

}

fn need_process_queries(doh_conn_map: &HashMap<u32, (ServerInfo, Option<DohConnection>)>) -> bool {

    if doh_conn_map.is_empty() {

        return false;

    }

    for (_, doh_conn) in doh_conn_map.values() {

        if let Some(doh_conn) = doh_conn {

            if !doh_conn.query_map.is_empty() || !doh_conn.pending_queries.is_empty() {

                return true;

            }

        }

    }

    false

}



async fn doh_handler(

    mut cmd_rx: CmdReceiver,
@@ -703,7 +715,7 @@ async fn doh_handler( 
                    }

                }

                join_all(futures).await

            } , if !doh_conn_map.is_empty() => {},

            }, if need_process_queries(&doh_conn_map) => {},

            Some(result) = probe_futures.next() => {

                let runtime_clone = runtime.clone();

                handle_probe_result(result, &mut doh_conn_map, runtime_clone, validation_fn);
@@ -1015,12 +1027,11 @@ pub unsafe extern "C" fn doh_query( 
                        response.copy_from_slice(&answer);

                        answer.len() as ssize_t

                    }

                    Response::Error { error: QueryError::ServerNotReady } => RESULT_CAN_NOT_SEND,

                    _ => RESULT_INTERNAL_ERROR,

                    _ => RESULT_CAN_NOT_SEND,

                },

                Err(e) => {

                    error!("no result {}", e);

                    RESULT_INTERNAL_ERROR

                    RESULT_CAN_NOT_SEND

                }

            },

            Err(e) => {
@@ -1029,7 +1040,7 @@ pub unsafe extern "C" fn doh_query( 
            }

        }

    } else {

        RESULT_INTERNAL_ERROR

        RESULT_CAN_NOT_SEND

    }

}

tests/resolv_private_dns_test.cpp

+3 −11
Original line number Diff line number Diff line
@@ -285,7 +285,7 @@ TEST_P(TransportParameterizedTest, GetAddrInfo) { 
    if (testParamHasDoh()) {

        EXPECT_NO_FAILURE(expectQueries(0 /* dns */, 0 /* dot */, 2 /* doh */));

    } else {

        EXPECT_NO_FAILURE(expectQueries(2 /* dns */, 0 /* dot */, 0 /* doh */));

        EXPECT_NO_FAILURE(expectQueries(0 /* dns */, 2 /* dot */, 0 /* doh */));

    }



    // Stop the private DNS servers. Since we are in opportunistic mode, queries will
@@ -298,7 +298,7 @@ TEST_P(TransportParameterizedTest, GetAddrInfo) { 
    if (testParamHasDoh()) {

        EXPECT_NO_FAILURE(expectQueries(2 /* dns */, 0 /* dot */, 2 /* doh */));

    } else {

        EXPECT_NO_FAILURE(expectQueries(4 /* dns */, 0 /* dot */, 0 /* doh */));

        EXPECT_NO_FAILURE(expectQueries(2 /* dns */, 2 /* dot */, 0 /* doh */));

    }

}
@@ -357,8 +357,7 @@ TEST_F(PrivateDnsDohTest, ValidationFail) { 


// Tests that DoH query fails and fallback happens.

//   - Fallback to UDP if DoH query times out

//   - Fallback to DoT if DoH validation is in progress.

//   - Fallback to UDP if DoH validation has failed.

//   - Fallback to DoT if DoH validation is in progress or has failed.

TEST_F(PrivateDnsDohTest, QueryFailover) {

    const auto parcel = DnsResponderClient::GetDefaultResolverParamsParcel();

    ASSERT_TRUE(mDnsClient.SetResolversFromParcel(parcel));
@@ -391,13 +390,6 @@ TEST_F(PrivateDnsDohTest, QueryFailover) { 


    EXPECT_EQ(dot.queries(), 2);

    EXPECT_EQ(dns.queries().size(), 0U);

    waitForDohValidationTimeout();

    flushCache();



    // Expect that this query fall back to UDP.

    EXPECT_NO_FAILURE(sendQueryAndCheckResult());

    EXPECT_EQ(dot.queries(), 2);

    EXPECT_EQ(dns.queries().size(), 2U);

}



// Tests that the DnsResolver prioritizes IPv6 DoH servers over IPv4 DoH servers.