// Copyright 2014 The Android Open Source Project
//
// This software is licensed under the terms of the GNU General Public
// License version 2, as published by the Free Software Foundation, and
// may be copied, distributed, and modified under those terms.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

#include "android/base/async/AsyncReader.h"

#include "android/base/sockets/SocketErrors.h"
#include "android/base/sockets/SocketUtils.h"

namespace android {
namespace base {

void AsyncReader::reset(void* buffer,
                        size_t bufferSize,
                        Looper::FdWatch* watch) {
    mBuffer = reinterpret_cast<uint8_t*>(buffer);
    mBufferSize = bufferSize;
    mPos = 0U;
    mFdWatch = watch;
    if (bufferSize > 0) {
        watch->wantRead();
    }
}

AsyncStatus AsyncReader::run() {
    if (mPos >= mBufferSize) {
        return kAsyncCompleted;
    }

    do {
        ssize_t ret = socketRecv(mFdWatch->fd(),
                                 mBuffer + mPos,
                                 mBufferSize - mPos);
        if (ret == 0) {
            // Disconnection!
            errno = ECONNRESET;
            return kAsyncError;
        }

        if (ret < 0) {
            if (errno == EWOULDBLOCK || errno == EAGAIN) {
                mFdWatch->wantRead();
                return kAsyncAgain;
            }
            return kAsyncError;
        }
        mPos += static_cast<size_t>(ret);
    } while (mPos < mBufferSize);

    mFdWatch->dontWantRead();
    return kAsyncCompleted;
}

}  // namespace base
}  // namespace android