Beispiel #1
0
func main() {

	flag.Usage = func() {
		fmt.Fprintf(os.Stderr, "\nUsage: %s [-v] [config]\n", os.Args[0])
	}
	verbose := flag.Bool("v", false, "a bool")
	flag.Parse()

	if !*verbose {
		e := logger.Initialize("hooktftp")
		if e != nil {
			log.Fatal("Failed to initialize logger")
		}
	}

	if len(flag.Args()) > 0 {
		CONFIG_PATH = flag.Args()[0]
	}

	logger.Info("Reading hooks from %s", CONFIG_PATH)

	configData, err := ioutil.ReadFile(CONFIG_PATH)

	if err != nil {
		logger.Crit("Failed to read config: %s", err)
		return
	}

	conf, err := config.ParseYaml(configData)
	if err != nil {
		logger.Crit("Failed to parse config: %s", err)
		return
	}

	for _, hookDef := range conf.HookDefs {
		logger.Notice("Compiling hook %s", hookDef)

		// Create new hookDef variable for the hookDef pointer for each loop
		// iteration. Go reuses the hookDef variable and if we pass pointer to
		// that terrible things happen.
		newPointer := hookDef
		hook, err := hooks.CompileHook(&newPointer)
		if err != nil {
			logger.Crit("Failed to compile hook %s: %s", hookDef, err)
			return
		}
		HOOKS = append(HOOKS, hook)
	}

	if conf.Port == "" {
		conf.Port = "69"
	}

	addr, err := net.ResolveUDPAddr("udp", ":"+conf.Port)
	if err != nil {
		logger.Crit("Failed to resolve address: %s", err)
		return
	}

	server, err := tftp.NewTFTPServer(addr)
	if err != nil {
		logger.Crit("Failed to listen: %s", err)
		return
	}

	logger.Notice("Listening on %d", conf.Port)

	if conf.User != "" {
		err := DropPrivileges(conf.User)
		if err != nil {
			logger.Crit("Failed to drop privileges to '%s' error: %v", conf.User, err)
			return
		}
		currentUser, _ := user.Current()
		logger.Notice("Dropped privileges to %s", currentUser)
	}

	if conf.User == "" && syscall.Getuid() == 0 {
		logger.Warning("Running as root and 'user' is not set in %s", CONFIG_PATH)
	}

	for {
		res, err := server.Accept()
		if err != nil {
			logger.Err("Bad tftp request: %s", err)
			continue
		}

		go handleRRQ(res)
	}

	logger.Close()

}
Beispiel #2
0
func main() {
	// 独自のLoggerを使用
	InitLogger()
	logger := GetLogger()
	defer logger.Close()

	// master ban
	var master *TBan
	var tesuu int = 0
	//player := NewPlayer("Slide")
	//player := NewPlayer("Random")
	player := NewPlayer("Main")

	// 将棋所とのやりとり
	// TODO:いつでも返答すべきコマンドは常時listenするイメージで。GoRoutineとChannelを使えばよさげ
	scanner := bufio.NewScanner(os.Stdin)
	for scanner.Scan() {
		text := scanner.Text()
		logger.Req(text)
		switch text {
		// エンジン登録時は、usiとquitのみ入力される。
		case "usi":
			respUSI(logger)
		case "quit":
			// TODO 終了前処理
			os.Exit(0)
		case "setoption name USI_Ponder value true":
			// TODO 設定を保存する
		case "setoption name USI_Hash value 256":
			// TODO 設定を保存する
		case "isready":
			master = CreateInitialState()
			tesuu = 0
			Resp("readyok", logger)
		case "usinewgame":
			// TODO: モードを切り替えるべきか。
		case "gameover":
			// TODO: 対局待ち状態に戻る。
		default:
			if s.HasPrefix(text, "position") {
				logger.Trace(text)
				split_text := s.Split(text, " ")
				// 通常の対局
				// position startpos moves 7g7f 8b7b 2g2f
				is_sfen := false
				if split_text[1] == "sfen" {
					is_sfen = true
					// 局面編集からの検討だとこのように
					// position sfen lnsgkgsnl/1r5b1/1pppppppp/p8/9/9/PPPPPPPPP/1B5R1/LNSGKGSNL b - 1 moves 2g2f
					sfen_index := s.Index(text, "sfen")
					moves_index := s.Index(text, "moves")
					var sfen_str string
					if moves_index == -1 {
						sfen_str = text[sfen_index:]
					} else {
						sfen_str = text[sfen_index : moves_index-1]
					}
					master = FromSFEN(sfen_str)
				}
				if is_sfen {
					// こちらのルートはどうすればいいのか不明。デッドコピーとしておく。
					for index, value := range split_text {
						if index < 7 {
							continue
						}
						// 何度も一手ずつ反映する必要はないので、スキップしている。
						if index-7 < tesuu {
							continue
						}
						logger.Trace("to apply: " + value)
						master.ApplyMove(value, true, true, true)
						logger.Trace(master.Display())
						tesuu++
					}
					// resp("info string "+text, logger)
				} else {
					for index, value := range split_text {
						if index < 3 {
							continue
						}
						// 何度も一手ずつ反映する必要はないので、スキップしている。
						if index-3 < tesuu {
							continue
						}
						logger.Trace("to apply: " + value)
						master.ApplyMove(value, true, true, true)
						logger.Trace(master.Display())
						logger.Trace(master.ToSFEN(true))
						tesuu++
					}
				}
			} else if s.HasPrefix(text, "go") {
				split_text := s.Split(text, " ")
				btime := split_text[2]
				wtime := split_text[4]
				teban := *(master.Teban)
				var ms int = 0
				if teban {
					ms, _ = strconv.Atoi(btime)
				} else {
					ms, _ = strconv.Atoi(wtime)
				}
				bestmove, score := player.Search(master, ms)
				if len(bestmove) < 6 {
					master.ApplyMove(bestmove, true, true, true)
					logger.Trace(master.Display())
					logger.Trace(master.ToSFEN(true))
					tesuu++
				}
				Resp(("info time 0 depth 1 nodes 1 score cp " + ToDisplayScore(score, teban) + " pv " + bestmove), logger)
				bestmove_str := "bestmove " + bestmove
				Resp(bestmove_str, logger)
			}
		}
	}
}