/*
 * Copyright (C) 2010 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.
 */

#ifndef NETFD_H_
#define NETFD_H_

#include <conscrypt/jniutil.h>

/**
 * Wraps access to the int inside a java.io.FileDescriptor, taking care of throwing exceptions.
 */
class NetFd {
 public:
    NetFd(JNIEnv* env, jobject fileDescriptor)
        : mEnv(env), mFileDescriptor(fileDescriptor), mFd(-1) {}

    bool isClosed() {
        mFd = conscrypt::jniutil::jniGetFDFromFileDescriptor(mEnv, mFileDescriptor);
        bool closed = (mFd == -1);
        if (closed) {
            conscrypt::jniutil::throwException(mEnv, "java/net/SocketException",
                                                  "Socket closed");
        }
        return closed;
    }

    int get() const {
        return mFd;
    }

 private:
    JNIEnv* mEnv;
    jobject mFileDescriptor;
    int mFd;

    // Disallow copy and assignment.
    NetFd(const NetFd&);
    void operator=(const NetFd&);
};

/**
 * Used to retry syscalls that can return EINTR. This differs from TEMP_FAILURE_RETRY in that
 * it also considers the case where the reason for failure is that another thread called
 * Socket.close.
 */
#define NET_FAILURE_RETRY(fd, exp)                     \
    ({                                                 \
        typeof(exp) _rc;                               \
        do {                                           \
            _rc = (exp);                               \
            if (_rc == -1) {                           \
                if (fd.isClosed() || errno != EINTR) { \
                    break;                             \
                }                                      \
            }                                          \
        } while (_rc == -1);                           \
        _rc;                                           \
    })

#endif  // NETFD_H_