diff options
Diffstat (limited to 'libs/samba/src/main/java/org')
3 files changed, 145 insertions, 2 deletions
diff --git a/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/NativeSamba.kt b/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/NativeSamba.kt index 1863e22..c951d37 100644 --- a/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/NativeSamba.kt +++ b/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/NativeSamba.kt @@ -27,6 +27,7 @@ internal object NativeSamba { fun removeDir(path: String): Boolean fun unlink(path: String): Boolean fun readLink(path: String): String? + fun openFile(path: String, mode: OpenMode): File? } interface Url : Object { @@ -53,6 +54,25 @@ internal object NativeSamba { fun list(): Array<DirEntry> } + enum class OpenMode(val value: Int) { + READ(0), + WRITE_CREATE_TRUNCATE(1), + } + + enum class SeekWhence(val value: Int) { + SET(0), + CURRENT(1), + } + + interface File : Object { + val path: String + + fun read(bytes: ByteArray, offset: Int, length: Int): Int + fun seek(offset: Long, whence: SeekWhence): Long + + fun write(bytes: ByteArray, offset: Int, length: Int): Int + } + private class NativeContext(private var ptr: Long): Context { override fun destroy() { if (ptr == 0L) return @@ -97,6 +117,11 @@ internal object NativeSamba { override fun readLink(path: String): String? { return nativeContextReadLink(ptr, path) } + + override fun openFile(path: String, mode: NativeSamba.OpenMode): File? { + val file = nativeContextOpenFile(ptr, path, mode.value) + return if (file != 0L) NativeFile(path, file) else null + } } private class NativeUrl(private var ptr: Long): Url { @@ -123,6 +148,26 @@ internal object NativeSamba { } } + private class NativeFile(override val path: String, private var ptr: Long): File { + override fun read(bytes: ByteArray, offset: Int, length: Int): Int { + return nativeFileRead(ptr, bytes, offset, length) + } + + override fun seek(offset: Long, whence: SeekWhence): Long { + return nativeFileSeek(ptr, offset, whence.value) + } + + override fun write(bytes: ByteArray, offset: Int, length: Int): Int { + return nativeFileWrite(ptr, bytes, offset, length) + } + + override fun destroy() { + if (ptr == 0L) return + nativeFileDestroy(ptr) + ptr = 0L + } + } + init { System.loadLibrary("samba") } @@ -154,10 +199,16 @@ internal object NativeSamba { private external fun nativeContextRemoveDir(ptr: Long, path: String): Boolean private external fun nativeContextUnlink(ptr: Long, path: String): Boolean private external fun nativeContextReadLink(otr: Long, path: String): String? + private external fun nativeContextOpenFile(ptr: Long, path: String, mode: Int): Long private external fun nativeUrlDestroy(ptr: Long) private external fun nativeUrlPath(ptr: Long): String private external fun nativeDirDestroy(ptr: Long) private external fun nativeDirList(ptr: Long): Array<DirEntry> + + private external fun nativeFileDestroy(ptr: Long) + private external fun nativeFileRead(ptr: Long, bytes: ByteArray, offset: Int, length: Int): Int + private external fun nativeFileSeek(ptr: Long, offset: Long, whence: Int): Long + private external fun nativeFileWrite(ptr: Long, bytes: ByteArray, offset: Int, length: Int): Int } diff --git a/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaConnection.kt b/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaConnection.kt index d7302cd..6eda092 100644 --- a/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaConnection.kt +++ b/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaConnection.kt @@ -40,6 +40,9 @@ internal class SambaConnection(uri: String, credentials: SambaCredentials) { fun readLink(path: String): String? = if (connected) context.readLink(join(url!!.path(), path)) else null + fun openFile(path: String, read: NativeSamba.OpenMode): NativeSamba.File? = + if (connected) context.openFile(join(url!!.path(), path), read) else null + companion object { fun join(a: String, b: String): String { if (a.isEmpty() || b.startsWith("/")) return b diff --git a/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaFile.kt b/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaFile.kt index 817c1bf..9913035 100644 --- a/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaFile.kt +++ b/libs/samba/src/main/java/org/the_jk/cleversync/io/samba/SambaFile.kt @@ -15,11 +15,96 @@ internal class SambaFile( private var cacheEndOfLife: Instant = Instant.now().plusSeconds(60), ) : ModifiableFile { override fun write(): OutputStream { - TODO("Not yet implemented") + val file = conn.openFile(path, NativeSamba.OpenMode.WRITE_CREATE_TRUNCATE) + ?: throw IOException(conn.error) + return object : OutputStream() { + override fun write(b: Int) { + val buffer = ByteArray(1) + buffer[0] = b.toByte() + if (file.write(buffer, 0, 1) != 1) throw IOException(conn.error) + } + + override fun write(b: ByteArray?) = write(b, 0, b?.size ?: 0) + + override fun write(b: ByteArray?, off: Int, len: Int) { + if (b == null) throw NullPointerException("b == null") + if (off < 0) throw IndexOutOfBoundsException("off < 0") + if (len < 0) throw java.lang.IndexOutOfBoundsException("len < 0") + if (off + len > b.size) throw IndexOutOfBoundsException("off + len > b.size") + if (file.write(b, off, len) != len) throw IOException(conn.error) + } + + override fun flush() { + clearCache() + } + + override fun close() { + file.destroy() + clearCache() + } + } } override fun read(): InputStream { - TODO("Not yet implemented") + val file = conn.openFile(path, NativeSamba.OpenMode.READ) ?: throw IOException(conn.error) + return object : InputStream() { + private var markedPosition = 0L + + override fun read(): Int { + val buffer = ByteArray(1) + val got = file.read(buffer, 0, 1) + if (got == 0) return -1 + if (got < 0) throw IOException(conn.error) + return buffer[0].toInt() + } + + override fun read(b: ByteArray?) = read(b, 0, b?.size ?: 0) + + override fun read(b: ByteArray?, off: Int, len: Int): Int { + if (b == null) throw NullPointerException("b == null") + if (off < 0) throw IndexOutOfBoundsException("off < 0") + if (len < 0) throw java.lang.IndexOutOfBoundsException("len < 0") + if (off + len > b.size) throw IndexOutOfBoundsException("off + len > b.size") + if (len == 0) return 0 + val got = file.read(b, off, len) + if (got == 0) return -1 + if (got < 0) throw IOException(conn.error) + return got + } + + override fun skip(n: Long): Long { + if (n <= 0) return 0 + val offset = file.seek(n, NativeSamba.SeekWhence.CURRENT) + if (offset < 0) throw IOException(conn.error) + return offset - n + } + + override fun available(): Int { + val current = file.seek(0, NativeSamba.SeekWhence.CURRENT) + if (current < 0) throw IOException(conn.error) + val total = size + if (current.toULong() >= total) return 0 + val left = total - current.toULong() + if (left >= Int.MAX_VALUE.toULong()) return Int.MAX_VALUE + return left.toInt() + } + + override fun close() { + file.destroy() + } + + override fun mark(readlimit: Int) { + markedPosition = file.seek(0, NativeSamba.SeekWhence.CURRENT) + } + + override fun markSupported() = true + + override fun reset() { + if (file.seek(markedPosition, NativeSamba.SeekWhence.SET) != markedPosition) { + throw IOException(conn.error) + } + } + } } override val size: ULong get() { @@ -37,4 +122,8 @@ internal class SambaFile( private fun useCached(): Boolean { return Instant.now().isBefore(cacheEndOfLife) } + + private fun clearCache() { + cacheEndOfLife = Instant.EPOCH + } } |