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

Commit da811959 authored by Josh Gao's avatar Josh Gao
Browse files

adb: check for an error response from adbd between each write. 

When sending a file, do a 0-timeout poll to check to see if an error has
occurred, so that we can immediately report failure.

Bug: http://b/26816782
Change-Id: I4a8aa8408a36940bfda7b0ecfa5d13755f4aa14d
(cherry picked from commit afcdcd70)
parent 4d74811c

adb/file_sync_client.cpp

+28 −3
Original line number Diff line number Diff line
@@ -88,7 +88,8 @@ class SyncConnection { 
            : total_bytes_(0),

              start_time_ms_(CurrentTimeMs()),

              expected_total_bytes_(0),

              expect_multiple_files_(false) {

              expect_multiple_files_(false),

              expect_done_(false) {

        max = SYNC_DATA_MAX; // TODO: decide at runtime.



        std::string error;
@@ -117,6 +118,16 @@ class SyncConnection { 


    bool IsValid() { return fd >= 0; }



    bool ReceivedError(const char* from, const char* to) {

        adb_pollfd pfd = {.fd = fd, .events = POLLIN};

        int rc = adb_poll(&pfd, 1, 0);

        if (rc < 0) {

            Error("failed to poll: %s", strerror(errno));

            return true;

        }

        return rc != 0;

    }



    bool SendRequest(int id, const char* path_and_mode) {

        size_t path_length = strlen(path_and_mode);

        if (path_length > 1024) {
@@ -175,6 +186,7 @@ class SyncConnection { 
        p += sizeof(SyncRequest);



        WriteOrDie(lpath, rpath, &buf[0], (p - &buf[0]));

        expect_done_ = true;

        total_bytes_ += data_length;

        return true;

    }
@@ -220,6 +232,11 @@ class SyncConnection { 
            total_bytes_ += bytes_read;

            bytes_copied += bytes_read;



            // Check to see if we've received an error from the other side.

            if (ReceivedError(lpath, rpath)) {

                break;

            }



            ReportProgress(rpath, bytes_copied, total_size);

        }
@@ -228,17 +245,24 @@ class SyncConnection { 
        syncmsg msg;

        msg.data.id = ID_DONE;

        msg.data.size = mtime;

        expect_done_ = true;

        return WriteOrDie(lpath, rpath, &msg.data, sizeof(msg.data));

    }



    bool CopyDone(const char* from, const char* to) {

        syncmsg msg;

        if (!ReadFdExactly(fd, &msg.status, sizeof(msg.status))) {

            Error("failed to copy '%s' to '%s': no ID_DONE: %s", from, to, strerror(errno));

            Error("failed to copy '%s' to '%s': couldn't read from device", from, to);

            return false;

        }

        if (msg.status.id == ID_OKAY) {

            if (expect_done_) {

                expect_done_ = false;

                return true;

            } else {

                Error("failed to copy '%s' to '%s': received premature success", from, to);

                return true;

            }

        }

        if (msg.status.id != ID_FAIL) {

            Error("failed to copy '%s' to '%s': unknown reason %d", from, to, msg.status.id);
@@ -357,6 +381,7 @@ class SyncConnection { 


    uint64_t expected_total_bytes_;

    bool expect_multiple_files_;

    bool expect_done_;



    LinePrinter line_printer_;

adb/test_device.py

+16 −0
Original line number Diff line number Diff line
@@ -766,6 +766,22 @@ class FileOperationsTest(DeviceTest): 
            if host_dir is not None:

                shutil.rmtree(host_dir)



    @requires_non_root

    def test_push_error_reporting(self):

        """Make sure that errors that occur while pushing a file get reported



        Bug: http://b/26816782

        """

        with tempfile.NamedTemporaryFile() as tmp_file:

            tmp_file.write('\0' * 1024 * 1024)

            tmp_file.flush()

            try:

                self.device.push(local=tmp_file.name, remote='/system/')

                self.fail("push should not have succeeded")

            except subprocess.CalledProcessError as e:

                output = e.output



            self.assertIn("Permission denied", output)



    def _test_pull(self, remote_file, checksum):

        tmp_write = tempfile.NamedTemporaryFile(mode='wb', delete=False)