ladybird/Userland/Games/Chess/Engine.cpp
Ben Wiederhake bdeccf8844 Chess+ChessEngine: Fix stockfish by setting correct blocking flag
Stockfish apparently cannot handle non-blocking I/O, and it does not
make sense to assume that all chess engines can do so.

Fixes #18946.
2023-05-27 18:44:21 +02:00

98 lines
2.5 KiB
C++

/*
* Copyright (c) 2020, the SerenityOS developers.
* Copyright (c) 2023, Tim Ledbetter <timledbetter@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "Engine.h"
#include <LibCore/File.h>
#include <LibCore/System.h>
#include <fcntl.h>
#include <spawn.h>
#include <stdio.h>
#include <unistd.h>
Engine::~Engine()
{
quit();
}
Engine::Engine(StringView command)
: m_command(command)
{
connect_to_engine_service();
}
void Engine::connect_to_engine_service()
{
int wpipefds[2];
int rpipefds[2];
if (pipe2(wpipefds, O_CLOEXEC) < 0) {
perror("pipe2");
VERIFY_NOT_REACHED();
}
if (pipe2(rpipefds, O_CLOEXEC) < 0) {
perror("pipe2");
VERIFY_NOT_REACHED();
}
posix_spawn_file_actions_t file_actions;
posix_spawn_file_actions_init(&file_actions);
posix_spawn_file_actions_adddup2(&file_actions, wpipefds[0], STDIN_FILENO);
posix_spawn_file_actions_adddup2(&file_actions, rpipefds[1], STDOUT_FILENO);
char const* argv[] = { m_command.characters(), nullptr };
pid_t pid = -1;
if (posix_spawnp(&pid, m_command.characters(), &file_actions, nullptr, const_cast<char**>(argv), environ) < 0) {
perror("posix_spawnp");
VERIFY_NOT_REACHED();
}
posix_spawn_file_actions_destroy(&file_actions);
close(wpipefds[0]);
close(rpipefds[1]);
auto infile = Core::File::adopt_fd(rpipefds[0], Core::File::OpenMode::Read).release_value_but_fixme_should_propagate_errors();
infile->set_blocking(false).release_value_but_fixme_should_propagate_errors();
set_in(move(infile)).release_value_but_fixme_should_propagate_errors();
auto outfile = Core::File::adopt_fd(wpipefds[1], Core::File::OpenMode::Write).release_value_but_fixme_should_propagate_errors();
outfile->set_blocking(false).release_value_but_fixme_should_propagate_errors();
set_out(move(outfile));
send_command(Chess::UCI::UCICommand());
m_connected = true;
}
void Engine::handle_bestmove(Chess::UCI::BestMoveCommand const& command)
{
if (m_bestmove_callback)
m_bestmove_callback(command.move());
m_bestmove_callback = nullptr;
}
void Engine::quit()
{
if (!m_connected)
return;
send_command(Chess::UCI::QuitCommand());
m_connected = false;
}
void Engine::handle_unexpected_eof()
{
m_connected = false;
if (m_bestmove_callback)
m_bestmove_callback(Error::from_errno(EPIPE));
m_bestmove_callback = nullptr;
if (on_connection_lost)
on_connection_lost();
}