1*cab87e9cSJagpal Singh Gill #include "software_utils.hpp"
2*cab87e9cSJagpal Singh Gill
3*cab87e9cSJagpal Singh Gill #include <phosphor-logging/lg2.hpp>
4*cab87e9cSJagpal Singh Gill
5*cab87e9cSJagpal Singh Gill PHOSPHOR_LOG2_USING;
6*cab87e9cSJagpal Singh Gill
7*cab87e9cSJagpal Singh Gill namespace phosphor::software::utils
8*cab87e9cSJagpal Singh Gill {
9*cab87e9cSJagpal Singh Gill
writeToFile(int imageFd,FILE * outStream)10*cab87e9cSJagpal Singh Gill static bool writeToFile(int imageFd, FILE* outStream)
11*cab87e9cSJagpal Singh Gill {
12*cab87e9cSJagpal Singh Gill const int bSize = 100;
13*cab87e9cSJagpal Singh Gill ssize_t nRead = 0;
14*cab87e9cSJagpal Singh Gill unsigned char buf[bSize];
15*cab87e9cSJagpal Singh Gill
16*cab87e9cSJagpal Singh Gill while ((nRead = read(imageFd, buf, bSize)) > 0)
17*cab87e9cSJagpal Singh Gill {
18*cab87e9cSJagpal Singh Gill if (fwrite(buf, 1, nRead, outStream) != (size_t)nRead)
19*cab87e9cSJagpal Singh Gill {
20*cab87e9cSJagpal Singh Gill error("Failed to write to file");
21*cab87e9cSJagpal Singh Gill return false;
22*cab87e9cSJagpal Singh Gill }
23*cab87e9cSJagpal Singh Gill }
24*cab87e9cSJagpal Singh Gill if (nRead < 0)
25*cab87e9cSJagpal Singh Gill {
26*cab87e9cSJagpal Singh Gill error("Failed to read from input file");
27*cab87e9cSJagpal Singh Gill return false;
28*cab87e9cSJagpal Singh Gill }
29*cab87e9cSJagpal Singh Gill return true;
30*cab87e9cSJagpal Singh Gill }
31*cab87e9cSJagpal Singh Gill
unTar(int imageFd,const std::string & extractDirPath)32*cab87e9cSJagpal Singh Gill bool unTar(int imageFd, const std::string& extractDirPath)
33*cab87e9cSJagpal Singh Gill {
34*cab87e9cSJagpal Singh Gill std::string tarCmd = "tar -xf - -C " + extractDirPath + " --no-same-owner";
35*cab87e9cSJagpal Singh Gill info("Executing command: {CMD}", "CMD", tarCmd);
36*cab87e9cSJagpal Singh Gill FILE* outStream = popen(tarCmd.c_str(), "w");
37*cab87e9cSJagpal Singh Gill if (outStream == nullptr)
38*cab87e9cSJagpal Singh Gill {
39*cab87e9cSJagpal Singh Gill error("Failed to open pipe to execute command: {CMD}", "CMD", tarCmd);
40*cab87e9cSJagpal Singh Gill return false;
41*cab87e9cSJagpal Singh Gill }
42*cab87e9cSJagpal Singh Gill
43*cab87e9cSJagpal Singh Gill if (!writeToFile(imageFd, outStream))
44*cab87e9cSJagpal Singh Gill {
45*cab87e9cSJagpal Singh Gill error("Failed to write to file");
46*cab87e9cSJagpal Singh Gill pclose(outStream);
47*cab87e9cSJagpal Singh Gill return false;
48*cab87e9cSJagpal Singh Gill }
49*cab87e9cSJagpal Singh Gill
50*cab87e9cSJagpal Singh Gill if (pclose(outStream) != 0)
51*cab87e9cSJagpal Singh Gill {
52*cab87e9cSJagpal Singh Gill error("Failed to close pipe");
53*cab87e9cSJagpal Singh Gill return false;
54*cab87e9cSJagpal Singh Gill }
55*cab87e9cSJagpal Singh Gill return true;
56*cab87e9cSJagpal Singh Gill }
57*cab87e9cSJagpal Singh Gill
58*cab87e9cSJagpal Singh Gill } // namespace phosphor::software::utils
59*cab87e9cSJagpal Singh Gill
getIOContext()60*cab87e9cSJagpal Singh Gill boost::asio::io_context& getIOContext()
61*cab87e9cSJagpal Singh Gill {
62*cab87e9cSJagpal Singh Gill static boost::asio::io_context io;
63*cab87e9cSJagpal Singh Gill return io;
64*cab87e9cSJagpal Singh Gill }
65