nativeMain.kotlinx.io.streams.PosixOutput.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlinx-io-linuxx64 Show documentation

IO support libraries for Kotlin

There is a newer version: 0.1.16

package kotlinx.io.streams

import kotlinx.cinterop.*
import kotlinx.io.core.*
import kotlinx.io.errors.*
import kotlinx.io.internal.utils.*
import platform.posix.*

private const val ZERO: size_t = 0u

/**
 * Create a blocking [Output] writing to the specified [fileDescriptor] using [write].
 */
@ExperimentalIoApi
fun Output(fileDescriptor: Int): Output = PosixFileDescriptorOutput(fileDescriptor)

/**
 * Create a blocking [Output] writing to the specified [file] instance using [fwrite].
 */
@ExperimentalIoApi
fun Output(file: CPointer): Output = PosixFileInstanceOutput(file)

private class PosixFileDescriptorOutput(val fileDescriptor: Int) : AbstractOutput() {
    private var closed = false

    init {
        check(fileDescriptor >= 0) { "Illegal fileDescriptor: $fileDescriptor" }
        check(kx_internal_is_non_blocking(fileDescriptor) == 0) { "File descriptor is in O_NONBLOCK mode." }
    }

    override fun flush(buffer: IoBuffer) {
        while (buffer.canRead()) {
            if (write(fileDescriptor, buffer) <= 0) {
                throw PosixException.forErrno(posixFunctionName = "write()").wrapIO()
            }
        }
    }

    override fun closeDestination() {
        if (closed) return
        closed = true

        if (close(fileDescriptor) != 0) {
            val error = errno
            if (error != EBADF) { // EBADF is already closed or not opened
                throw PosixException.forErrno(error, posixFunctionName = "close()").wrapIO()
            }
        }
    }
}

private class PosixFileInstanceOutput(val file: CPointer) : AbstractOutput() {
    private var closed = false

    override fun flush(buffer: IoBuffer) {
        while (buffer.canRead()) {
            if (fwrite(buffer, file) == ZERO) {
                throw PosixException.forErrno(posixFunctionName = "fwrite()").wrapIO()
            }
        }
    }

    override fun closeDestination() {
        if (closed) return
        closed = true

        if (fclose(file) != 0) {
            throw PosixException.forErrno(posixFunctionName = "fclose").wrapIO()
        }
    }
}