コード例 #1
0
ファイル: sshserver.go プロジェクト: d-dorazio/gork
func (server *SshServer) handleChannel(user string, newChannel ssh.NewChannel) {
	if t := newChannel.ChannelType(); t != "session" {
		newChannel.Reject(ssh.UnknownChannelType, fmt.Sprintf("unknown channel type: %s", t))
		return
	}

	connection, requests, err := newChannel.Accept()
	if err != nil {
		fmt.Printf("Could not accept channel (%s)", err)
		return
	}
	defer connection.Close()

	logfile, err := os.Create(fmt.Sprintf("%s_%s", user, storyLogFilename(server.story)))
	if err != nil {
		panic(err)
	}
	defer logfile.Close()

	logger := log.New(logfile, "", log.LstdFlags)

	terminal := terminal.NewTerminal(connection, "")
	zsshterm := &gork.ZSshTerminal{Term: terminal}

	zm, err := gork.NewZMachine(server.mem, server.header, zsshterm, logger)
	if err != nil {
		fmt.Println(err)
		return
	}

	go func() {
		for req := range requests {
			switch req.Type {
			case "shell":
				if len(req.Payload) == 0 {
					req.Reply(true, nil)
				}
			case "pty-req":
				termLen := req.Payload[3]
				w, h := parseDims(req.Payload[termLen+4:])
				terminal.SetSize(w, h)
			case "window-change":
				w, h := parseDims(req.Payload)
				terminal.SetSize(w, h)
			}
		}
	}()

	defer func() {
		recover()
	}()

	zm.InterpretAll()

}
コード例 #2
0
ファイル: main.go プロジェクト: d-dorazio/gork
func terminalUI(story string, mem *gork.ZMemory, header *gork.ZHeader) {
	logfile, err := os.Create(storyLogFilename(story))
	if err != nil {
		panic(err)
	}
	defer logfile.Close()

	logger := log.New(logfile, "", log.LstdFlags)

	zm, err := gork.NewZMachine(mem, header, gork.ZTerminal{}, logger)
	if err != nil {
		panic(err)
	}

	if err := zm.InterpretAll(); err != nil {
		panic(err)
	}
}
コード例 #3
0
ファイル: wsserver.go プロジェクト: d-dorazio/gork
func (server *WSServer) run(addr string) {
	var upgrader = websocket.Upgrader{
		CheckOrigin: func(r *http.Request) bool { return true },
	}

	wsHandler := func(w http.ResponseWriter, r *http.Request) {
		conn, err := upgrader.Upgrade(w, r, nil)
		if err != nil {
			fmt.Printf("Failed to upgrade %s\n", err)
			return
		}

		remoteAddr := conn.RemoteAddr().String()
		logFilename := storyLogFilename(server.story)
		logfile, err := os.Create(fmt.Sprintf("wsserver_%s_%s", remoteAddr, logFilename))
		if err != nil {
			panic(err)
		}
		defer logfile.Close()
		logger := log.New(logfile, "", log.LstdFlags)

		wsdev := &gork.ZWSDev{Conn: conn}

		zm, err := gork.NewZMachine(server.mem, server.header, wsdev, logger)
		if err != nil {
			panic(err)
		}
		zm.InterpretAll()
	}

	http.HandleFunc("/play", wsHandler)
	err := http.ListenAndServe(addr, nil)
	if err != nil {
		panic(err)
	}
}