NotifyWatcher

Author: utgheith

.gitignore

@@ -6,8 +6,10 @@ target/
 .project
 .cache
 .sbtserver
+.bsp/
 project/.sbtserver
 tags
 nohup.out
 out
 lowered.hnir
+random_data/

build.sc

@@ -50,7 +50,8 @@ object os extends Module {
       def platformSegment = "jvm"
       def moduleDeps = super.moduleDeps :+ os.jvm()
       def ivyDeps = Agg(
-        ivy"net.java.dev.jna:jna:5.0.0"
+        ivy"net.java.dev.jna:jna:5.0.0",
+        ivy"com.lihaoyi::sourcecode::0.2.5",
       )
       object test extends Tests with OsLibTestModule {
         def platformSegment = "jvm"
@@ -107,7 +108,11 @@ trait OsLibModule extends CrossScalaModule with PublishModule{
 trait OsLibTestModule extends ScalaModule with TestModule{
   def ivyDeps = Agg(
     ivy"com.lihaoyi::utest::0.7.8",
-    ivy"com.lihaoyi::sourcecode::0.2.5"
+    ivy"com.lihaoyi::sourcecode::0.2.5",
+    if (scalaVersion().startsWith("2.11"))
+       ivy"org.scalacheck::scalacheck::1.15.2"
+    else
+       ivy"org.scalacheck::scalacheck::1.15.3"
   )
 
   def platformSegment: String
@@ -116,7 +121,7 @@ trait OsLibTestModule extends ScalaModule with TestModule{
     millSourcePath / s"src-$platformSegment"
   )
 
-  def testFrameworks = Seq("utest.runner.Framework")
+  def testFrameworks = Seq("utest.runner.Framework", "org.scalacheck.ScalaCheckFramework")
   // we check the textual output of system commands and expect it in english
   override def forkEnv: Target[Map[String, String]] = super.forkEnv() ++ Map("LC_ALL" -> "C")
 }

os/src/Internals.scala

@@ -22,4 +22,8 @@ object Internals{
     src,
     dest.write(_, 0, _)
   )
+
+  def linux() = {
+    System.getProperty("os.name") == "Linux"
+  }
 }

os/watch/src/inotify/Errno.scala

@@ -0,0 +1,8 @@
+package os.watch.inotify
+
+object Errno {
+  
+  val EAGAIN: Int = 11
+  val ENOTDIR: Int = 20
+
+}

os/watch/src/inotify/Event.scala

@@ -0,0 +1,31 @@
+package os.watch.inotify
+
+import java.nio.ByteBuffer
+
+case class Event(val wd: Int, val mask: Mask, val cookie: Int, val name: String) {
+
+
+
+}
+
+object Event {
+  def apply(buf: ByteBuffer): Event = {
+    val wd = buf.getInt
+    val mask = buf.getInt
+    val cookie = buf.getInt
+    val len = buf.getInt
+    val sb = new StringBuilder()
+    
+    var i = 0
+    
+    while (i < len) {
+      val b = buf.get()
+      if (b != 0) {
+        sb.append(b.toChar)
+      }
+      i += 1
+    }
+    
+    Event(wd, Mask(mask), cookie, sb.toString)
+  }
+}

os/watch/src/inotify/Mask.scala

@@ -0,0 +1,60 @@
+package os.watch.inotify
+
+import scala.collection.mutable
+
+case class Mask(mask: Int) {
+  def |(rhs: Mask): Mask = Mask(mask|rhs.mask)
+  def contains(rhs: Mask) : Boolean = (mask & rhs.mask) == rhs.mask
+
+  override def toString: String = {
+
+    val things = Mask.named_masks.toList.sortBy(_._1).flatMap { case(name,m) =>
+      if (this.contains(m)) Seq(name) else Seq()
+    }.mkString("+")
+
+    f"Mask($mask%08x = $things)"
+  }
+}
+
+object Mask {
+  val named_masks : mutable.Map[String,Mask] = mutable.Map()
+
+  private def named(bit: Int)(implicit name: sourcecode.Name): Mask = {
+    val a = Mask(1 << bit)
+    named_masks.put(name.value,a)
+    a
+  }
+
+  val access: Mask = named(0)
+  val modify: Mask = named(1)
+  val attrib: Mask = named(2)
+  val close_write: Mask = named(3)
+  val close_nowrite: Mask = named(4)
+  val open: Mask = named(5)
+  val move_from: Mask = named(6)
+  val move_to: Mask = named(7)
+  val create: Mask = named(8)
+  val delete: Mask = named(9)
+  val delete_self: Mask = named(10)
+
+  val unmount: Mask = named(13)
+  val overflow: Mask = named(14)
+  val ignored: Mask = named(15)
+
+  val only_dir: Mask = named(24)
+  val do_not_follow: Mask = named(25)
+  val exclude_unlink: Mask = named(26)
+
+  val mask_create: Mask = named(28)
+  val mask_add: Mask = named(29)
+  val is_dir: Mask = named(30)
+  val one_shot: Mask = named(31)
+
+
+  val close : Mask = close_write | close_nowrite
+  val move : Mask = move_from | move_to
+
+  val all_events = access | modify | attrib |
+    close | open | move | create |
+    delete | delete_self | unmount
+}

os/watch/src/inotify/Notify.scala

@@ -0,0 +1,101 @@
+package os.watch.inotify
+
+import com.sun.jna.{LastErrorException, Library, Native}
+import geny.Generator
+
+import java.nio.{ByteBuffer, ByteOrder}
+import java.util.concurrent.atomic.AtomicReference
+import scala.annotation.tailrec
+import scala.collection.mutable
+import scala.util.Try
+
+trait Notify extends Library {
+  //@throws[LastErrorException]
+  def inotify_init() : Int;
+  
+  //@throws[LastErrorException]
+  def inotify_init1(flags: Int) : Int
+
+  //@throws[LastErrorException]
+  def inotify_add_watch(fd: Int, path: String, mask: Int): Int
+
+  //@throws[LastErrorException]
+  def inotify_rm_watch(fd: Int, wd: Int): Int
+
+  //@throws[LastErrorException]
+  def read(fd: Int, buf: Array[Byte], count: Long): Long
+
+  //@throws[LastErrorException]
+  def close(fd: Int): Int
+}
+
+object Notify {
+  val it : Notify = Native.load("c",classOf[Notify])
+  
+  val O_NONBLOCK: Int = 1 << 11
+  
+  
+  import Generator._
+
+  // convenience
+  def add_watch(fd: Int, path : os.Path, actions: Mask) : Int = {
+    it.inotify_add_watch(fd,path.toString,actions.mask)
+  }
+
+  def events(buf: ByteBuffer): Generator[Event] = new Generator[Event]() {
+    override def generate(handleItem: Event => Action): Action = {
+      while (buf.hasRemaining) {
+        if (handleItem(Event(buf)) == End) return End
+      }
+      Continue
+    }
+  }
+
+  def buffers(fd: => AtomicReference[Option[Int]]): Generator[ByteBuffer] = {
+    new Generator[ByteBuffer] {
+      override def generate(handleItem: ByteBuffer => Action): Action = {
+
+        @tailrec
+        def loop(arr: Array[Byte]): Action = fd.get match {
+          case Some(fd) =>
+            it.read(fd, arr, arr.length) match {
+              case 0 =>
+                Continue
+              case n if n < 0 =>
+                val errno = Native.getLastError()
+                if (errno == Errno.EAGAIN) {
+                  Thread.sleep(10)
+                  loop(arr)
+                } else {
+                  throw new NotifyException(s"read error ${Native.getLastError()}, fd = $fd")
+                }
+              //throw new Exception (s"n = $n")
+              case n =>
+                val buf = ByteBuffer.wrap(arr, 0, n.toInt).order(ByteOrder.nativeOrder())
+                if (handleItem(buf) == End) {
+                  End
+                } else {
+                  loop(arr)
+                }
+            }
+          case None =>
+            End
+        }
+
+
+        loop(Array.fill[Byte](1000)(0))
+      }
+    }
+  }
+
+  def events(fd: AtomicReference[Option[Int]]): Generator[Event] = for {
+    b <- buffers(fd)
+    e <- events(b)
+  } yield e
+
+
+
+
+
+}
+

os/watch/src/inotify/NotifyException.scala

@@ -0,0 +1,5 @@
+package os.watch.inotify
+
+class NotifyException(msg: String) extends Exception(msg) {
+
+}