mirror of
https://github.com/ethereum/solidity
synced 2023-10-03 13:03:40 +00:00
137 lines
4.0 KiB
C++
137 lines
4.0 KiB
C++
/*
|
|
This file is part of solidity.
|
|
|
|
solidity is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
solidity is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with solidity. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
// SPDX-License-Identifier: GPL-3.0
|
|
#include <libsolidity/interface/SMTSolverCommand.h>
|
|
|
|
#include <liblangutil/Exceptions.h>
|
|
|
|
#include <libsolutil/CommonIO.h>
|
|
#include <libsolutil/Exceptions.h>
|
|
#include <libsolutil/Keccak256.h>
|
|
#include <libsolutil/TemporaryDirectory.h>
|
|
|
|
#include <boost/algorithm/string/join.hpp>
|
|
#include <boost/algorithm/string/predicate.hpp>
|
|
#include <boost/filesystem.hpp>
|
|
#include <boost/filesystem/fstream.hpp>
|
|
#include <boost/process.hpp>
|
|
|
|
using solidity::langutil::InternalCompilerError;
|
|
using solidity::util::errinfo_comment;
|
|
|
|
|
|
namespace solidity::frontend
|
|
{
|
|
|
|
namespace
|
|
{
|
|
ReadCallback::Result solveWithTemporaryFile(
|
|
boost::filesystem::path const& _pathToBinary,
|
|
std::vector<std::string> _commandArgs,
|
|
std::string const& _query
|
|
)
|
|
{
|
|
auto tempDir = solidity::util::TemporaryDirectory("smt");
|
|
util::h256 queryHash = util::keccak256(_query);
|
|
auto queryFileName = tempDir.path() / ("query_" + queryHash.hex() + ".smt2");
|
|
auto queryFile = boost::filesystem::ofstream(queryFileName);
|
|
queryFile << _query << std::flush;
|
|
|
|
_commandArgs.push_back(queryFileName.string());
|
|
|
|
boost::process::ipstream out; // output from subprocess read by main process
|
|
boost::process::child solver(
|
|
_pathToBinary,
|
|
boost::process::args(_commandArgs),
|
|
boost::process::std_out > out
|
|
);
|
|
|
|
std::vector<std::string> data;
|
|
std::string line;
|
|
while (std::getline(out, line))
|
|
if (!line.empty())
|
|
data.push_back(line);
|
|
|
|
solver.wait();
|
|
|
|
return ReadCallback::Result{true, boost::join(data, "\n")};
|
|
}
|
|
|
|
ReadCallback::Result solveWithPipe(
|
|
boost::filesystem::path const& _pathToBinary,
|
|
std::vector<std::string> _commandArgs,
|
|
std::string const& _query
|
|
)
|
|
{
|
|
boost::process::opstream in; // input to subprocess written to by main process
|
|
boost::process::ipstream out; // output from subprocess read by main process
|
|
boost::process::child solver(
|
|
_pathToBinary,
|
|
boost::process::args(_commandArgs),
|
|
boost::process::std_out > out,
|
|
boost::process::std_in < in
|
|
);
|
|
|
|
in << _query << std::flush;
|
|
in.pipe().close();
|
|
in.close();
|
|
|
|
std::vector<std::string> data;
|
|
std::string line;
|
|
while (std::getline(out, line))
|
|
if (!line.empty())
|
|
data.push_back(line);
|
|
|
|
solver.wait();
|
|
|
|
return ReadCallback::Result{true, boost::join(data, "\n")};
|
|
}
|
|
} // namespace
|
|
|
|
ReadCallback::Result SMTSolverCommand::solve(std::string const& _kind, std::string const& _query)
|
|
{
|
|
try
|
|
{
|
|
auto pos = _kind.find(":");
|
|
solAssert(pos != std::string::npos);
|
|
auto kind = _kind.substr(0, pos);
|
|
auto solverCommand = _kind.substr(pos + 1);
|
|
solAssert(kind == ReadCallback::kindString(ReadCallback::Kind::SMTQuery), "SMTQuery callback used as callback kind " + kind);
|
|
|
|
std::vector<std::string> commandArgs;
|
|
boost::split(commandArgs, solverCommand, boost::is_any_of(" "));
|
|
solAssert(commandArgs.size() > 0, "SMT command was empty");
|
|
auto solverBinary = commandArgs[0];
|
|
auto pathToBinary = boost::process::search_path(solverBinary);
|
|
|
|
if (pathToBinary.empty())
|
|
return ReadCallback::Result{false, solverBinary + " binary not found."};
|
|
|
|
commandArgs.erase(commandArgs.begin());
|
|
|
|
if (solverBinary == "eld")
|
|
return solveWithTemporaryFile(pathToBinary, std::move(commandArgs), _query);
|
|
return solveWithPipe(pathToBinary, std::move(commandArgs), _query);
|
|
}
|
|
catch (...)
|
|
{
|
|
return ReadCallback::Result{false, "Unknown exception in SMTQuery callback: " + boost::current_exception_diagnostic_information()};
|
|
}
|
|
}
|
|
|
|
}
|