luke
/
sbt
mirror of https://github.com/sbt/sbt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix canonical input for network client 

It is valid for the thin client input stream to return -1 as an EOF when
the user inputs ctrl+d in canonical mode.
This commit is contained in:
Ethan Atkins 2020-10-21 13:58:38 -07:00
parent 50f649c3c9
commit a0db985c36
2 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main-command/src/main/scala/sbt/internal/client/NetworkClient.scala
View File

@ -934,12 +934,9 @@ class NetworkClient(
val stopped = new AtomicBoolean(false)
override final def run(): Unit = {
def read(): Unit = {
inputStream.read match {
case -1 =>
case b =>
inLock.synchronized(stdinBytes.offer(b))
if (attached.get()) drain()
}
val b = inputStream.read
inLock.synchronized(stdinBytes.offer(b))
if (attached.get()) drain()
}
try read()
catch { case _: InterruptedException | NonFatal(_) => stopped.set(true) } finally {

12
main/src/main/scala/sbt/internal/server/NetworkChannel.scala
View File

@ -11,7 +11,6 @@ package server
import java.io.{ IOException, InputStream, OutputStream }
import java.net.{ Socket, SocketTimeoutException }
import java.nio.channels.ClosedChannelException
import java.util.concurrent.{
ConcurrentHashMap,
Executors,
@ -85,7 +84,7 @@ final class NetworkChannel(
private var initialized = false
private val pendingRequests: mutable.Map[String, JsonRpcRequestMessage] = mutable.Map()
private[this] val inputBuffer = new LinkedBlockingQueue[Byte]()
private[this] val inputBuffer = new LinkedBlockingQueue[Int]()
private[this] val pendingWrites = new LinkedBlockingQueue[(Array[Byte], Boolean)]()
private[this] val attached = new AtomicBoolean(false)
private[this] val alive = new AtomicBoolean(true)
@ -107,7 +106,7 @@ final class NetworkChannel(
interactive.set(true)
jsonRpcNotify(promptChannel, "")
}
private[sbt] def write(byte: Byte) = inputBuffer.add(byte)
private[sbt] def write(byte: Byte) = inputBuffer.add(byte.toInt)
private[this] val terminalHolder = new AtomicReference[Terminal](Terminal.NullTerminal)
override private[sbt] def terminal: Terminal = terminalHolder.get
@ -634,15 +633,12 @@ final class NetworkChannel(
)
}
private[this] lazy val inputStream: InputStream = new InputStream {
private[this] lazy val inputStream: InputStream = new Terminal.SimpleInputStream {
override def read(): Int = {
import sjsonnew.BasicJsonProtocol._
try {
jsonRpcNotify(readSystemIn, "")
inputBuffer.take & 0xFF match {
case -1 => throw new ClosedChannelException()
case b => b
}
inputBuffer.take
} catch {
case e: IOException =>
try jsonRpcNotify(cancelReadSystemIn, "")