Example #1
0
func (s *GardenServer) streamInput(decoder *json.Decoder, in *io.PipeWriter, process garden.Process, connCloseCh chan struct{}) {
	for {
		var payload transport.ProcessPayload
		err := decoder.Decode(&payload)
		if err != nil {
			close(connCloseCh)
			in.CloseWithError(errors.New("Connection closed"))
			return
		}

		switch {
		case payload.TTY != nil:
			process.SetTTY(*payload.TTY)

		case payload.Source != nil:
			if payload.Data == nil {
				in.Close()
				return
			} else {
				_, err := in.Write([]byte(*payload.Data))
				if err != nil {
					return
				}
			}

		case payload.Signal != nil:
			s.logger.Info("stream-input-process-signal", lager.Data{"payload": payload})

			switch *payload.Signal {
			case garden.SignalKill:
				err = process.Signal(garden.SignalKill)
				if err != nil {
					s.logger.Error("stream-input-process-signal-kill-failed", err, lager.Data{"payload": payload})
				}
			case garden.SignalTerminate:
				err = process.Signal(garden.SignalTerminate)
				if err != nil {
					s.logger.Error("stream-input-process-signal-terminate-failed", err, lager.Data{"payload": payload})
				}
			default:
				s.logger.Error("stream-input-unknown-process-payload-signal", nil, lager.Data{"payload": payload})
				in.Close()
				return
			}

		default:
			s.logger.Error("stream-input-unknown-process-payload", nil, lager.Data{"payload": payload})
			in.Close()
			return
		}
	}
}
						Stderr: GinkgoWriter,
					}, nil, signaller)
				Expect(err).NotTo(HaveOccurred())

				Eventually(stdout).Should(gbytes.Say("pid"))
			})

			AfterEach(func() {
				if cmd.ProcessState != nil && !cmd.ProcessState.Exited() {
					cmd.Process.Signal(os.Kill)
				}
			})

			Context("when the signaller is an LinkSignaller", func() {
				It("sends a kill message on the extra file descriptor", func(done Done) {
					Expect(process.Signal(garden.SignalKill)).To(Succeed())
					Eventually(stdout).Should(gbytes.Say("Received: killed"))
					close(done)
				}, 2.0)

				It("kills the process with a terminate signal", func(done Done) {
					Expect(process.Signal(garden.SignalTerminate)).To(Succeed())
					Eventually(stdout).Should(gbytes.Say("Received: terminated"))
					close(done)
				}, 2.0)

				Context("when an unsupported signal is sent", func() {
					AfterEach(func() {
						Expect(process.Signal(garden.SignalKill)).To(Succeed())
					})
						Stdout: io.MultiWriter(stdout, GinkgoWriter),
						Stderr: GinkgoWriter,
					}, nil)
				Expect(err).NotTo(HaveOccurred())

				Eventually(stdout).Should(gbytes.Say("trapping"))
			})

			AfterEach(func() {
				if cmd.ProcessState != nil && !cmd.ProcessState.Exited() {
					cmd.Process.Signal(os.Kill)
				}
			})

			It("kill the process", func(done Done) {
				Expect(process.Signal(garden.SignalKill)).To(Succeed())

				Expect(cmd.Wait()).NotTo(Succeed())

				close(done)
			}, 2.0)

			It("kills the process with a terminate signal", func(done Done) {
				Expect(process.Signal(garden.SignalTerminate)).To(Succeed())

				Eventually(stdout).Should(gbytes.Say("terminated"))
				Expect(cmd.Wait()).NotTo(Succeed())

				close(done)
			}, 2.0)
		})
					})
				})

				Describe("Signal", func() {
					BeforeEach(func() {
						errs := make(chan error, 1)
						errs <- io.EOF
						close(errs)

						fakeProcess.SignalStub = func(garden.Signal) error {
							return <-errs
						}
					})

					It("reattaches on use of closed connection", func() {
						Ω(process.Signal(garden.SignalTerminate)).Should(Succeed())
						Ω(fakeProcess.SignalArgsForCall(0)).Should(Equal(garden.SignalTerminate))

						Ω(innerConnection.AttachCallCount()).Should(Equal(2))
						handle, processID, calledProcessIO := innerConnection.AttachArgsForCall(1)
						Ω(handle).Should(Equal("la-contineur"))
						Ω(processID).Should(Equal(uint32(6)))
						Ω(calledProcessIO).Should(Equal(processIO))
					})
				})

				Describe("SetTTY", func() {
					BeforeEach(func() {
						errs := make(chan error, 1)
						errs <- io.EOF
						close(errs)