2016-09-19 11:50:08 +08:00
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
#include <Windows.h>
|
|
|
|
|
|
|
|
#include <fstream>
|
|
|
|
#include <memory>
|
|
|
|
#include <cassert>
|
|
|
|
#include "vcpkg_Commands.h"
|
|
|
|
#include "metrics.h"
|
|
|
|
#include <Shlobj.h>
|
|
|
|
#include "vcpkg_Files.h"
|
|
|
|
#include "vcpkg_System.h"
|
2016-10-01 07:54:07 +08:00
|
|
|
#include "vcpkg_Input.h"
|
2016-11-08 08:06:36 +08:00
|
|
|
#include "Paragraphs.h"
|
2017-01-25 11:08:51 +08:00
|
|
|
#include "vcpkg_Strings.h"
|
2017-03-04 22:25:05 +08:00
|
|
|
#include "vcpkg_Chrono.h"
|
2017-03-14 07:39:40 +08:00
|
|
|
#include "vcpkglib.h"
|
2016-09-19 11:50:08 +08:00
|
|
|
|
|
|
|
using namespace vcpkg;
|
|
|
|
|
|
|
|
void invalid_command(const std::string& cmd)
|
|
|
|
{
|
|
|
|
System::println(System::color::error, "invalid command: %s", cmd);
|
2017-01-13 14:03:57 +08:00
|
|
|
Commands::Help::print_usage();
|
2017-03-23 08:46:05 +08:00
|
|
|
Checks::exit_fail(VCPKG_LINE_INFO);
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static void inner(const vcpkg_cmd_arguments& args)
|
|
|
|
{
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::track_property("command", args.command);
|
2016-09-19 11:50:08 +08:00
|
|
|
if (args.command.empty())
|
|
|
|
{
|
2017-01-13 14:03:57 +08:00
|
|
|
Commands::Help::print_usage();
|
2017-03-23 08:46:05 +08:00
|
|
|
Checks::exit_fail(VCPKG_LINE_INFO);
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
|
2017-01-13 09:35:33 +08:00
|
|
|
if (auto command_function = Commands::find(args.command, Commands::get_available_commands_type_c()))
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
|
|
|
return command_function(args);
|
|
|
|
}
|
|
|
|
|
|
|
|
fs::path vcpkg_root_dir;
|
|
|
|
if (args.vcpkg_root_dir != nullptr)
|
|
|
|
{
|
|
|
|
vcpkg_root_dir = fs::absolute(Strings::utf8_to_utf16(*args.vcpkg_root_dir));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2017-02-15 07:35:34 +08:00
|
|
|
const optional<std::wstring> vcpkg_root_dir_env = System::get_environmental_variable(L"VCPKG_ROOT");
|
2017-03-29 06:05:55 +08:00
|
|
|
if (auto v = vcpkg_root_dir_env.get())
|
2016-09-20 10:12:46 +08:00
|
|
|
{
|
2017-03-29 06:05:55 +08:00
|
|
|
vcpkg_root_dir = fs::absolute(*v);
|
2016-09-20 10:12:46 +08:00
|
|
|
}
|
2016-09-19 11:50:08 +08:00
|
|
|
else
|
2016-09-20 10:12:46 +08:00
|
|
|
{
|
2016-09-22 12:43:59 +08:00
|
|
|
vcpkg_root_dir = Files::find_file_recursively_up(fs::absolute(System::get_exe_path_of_current_process()), ".vcpkg-root");
|
2016-09-20 10:12:46 +08:00
|
|
|
}
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
|
2017-03-14 08:38:04 +08:00
|
|
|
Checks::check_exit(VCPKG_LINE_INFO, !vcpkg_root_dir.empty(), "Error: Could not detect vcpkg-root.");
|
2016-09-20 10:12:46 +08:00
|
|
|
|
2016-09-19 11:50:08 +08:00
|
|
|
const expected<vcpkg_paths> expected_paths = vcpkg_paths::create(vcpkg_root_dir);
|
2017-03-14 08:38:04 +08:00
|
|
|
Checks::check_exit(VCPKG_LINE_INFO, !expected_paths.error_code(), "Error: Invalid vcpkg root directory %s: %s", vcpkg_root_dir.string(), expected_paths.error_code().message());
|
2017-04-01 07:23:48 +08:00
|
|
|
const vcpkg_paths paths = expected_paths.value_or_exit(VCPKG_LINE_INFO);
|
2016-09-19 11:50:08 +08:00
|
|
|
int exit_code = _wchdir(paths.root.c_str());
|
2017-03-14 08:38:04 +08:00
|
|
|
Checks::check_exit(VCPKG_LINE_INFO, exit_code == 0, "Changing the working dir failed");
|
2016-09-19 11:50:08 +08:00
|
|
|
|
2017-01-13 09:35:33 +08:00
|
|
|
if (auto command_function = Commands::find(args.command, Commands::get_available_commands_type_b()))
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
|
|
|
return command_function(args, paths);
|
|
|
|
}
|
|
|
|
|
2016-09-22 18:00:27 +08:00
|
|
|
triplet default_target_triplet;
|
2016-09-29 03:39:05 +08:00
|
|
|
if (args.target_triplet != nullptr)
|
2016-09-22 18:00:27 +08:00
|
|
|
{
|
2016-10-04 08:20:52 +08:00
|
|
|
default_target_triplet = triplet::from_canonical_name(*args.target_triplet);
|
2016-09-22 18:00:27 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2017-02-15 07:35:34 +08:00
|
|
|
const optional<std::wstring> vcpkg_default_triplet_env = System::get_environmental_variable(L"VCPKG_DEFAULT_TRIPLET");
|
2017-03-29 06:05:55 +08:00
|
|
|
if (auto v = vcpkg_default_triplet_env.get())
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
2017-03-29 06:05:55 +08:00
|
|
|
default_target_triplet = triplet::from_canonical_name(Strings::utf16_to_utf8(*v));
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
2016-09-23 15:44:30 +08:00
|
|
|
else
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
2016-09-23 15:44:30 +08:00
|
|
|
default_target_triplet = triplet::X86_WINDOWS;
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-10-01 07:54:07 +08:00
|
|
|
Input::check_triplet(default_target_triplet, paths);
|
2016-09-23 15:44:30 +08:00
|
|
|
|
2017-01-13 09:35:33 +08:00
|
|
|
if (auto command_function = Commands::find(args.command, Commands::get_available_commands_type_a()))
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
|
|
|
return command_function(args, paths, default_target_triplet);
|
|
|
|
}
|
|
|
|
|
|
|
|
return invalid_command(args.command);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void loadConfig()
|
|
|
|
{
|
|
|
|
fs::path localappdata;
|
|
|
|
{
|
|
|
|
// Config path in AppDataLocal
|
|
|
|
wchar_t* localappdatapath = nullptr;
|
|
|
|
if (S_OK != SHGetKnownFolderPath(FOLDERID_LocalAppData, 0, nullptr, &localappdatapath))
|
|
|
|
__fastfail(1);
|
|
|
|
localappdata = localappdatapath;
|
|
|
|
CoTaskMemFree(localappdatapath);
|
|
|
|
}
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
2017-03-30 00:49:09 +08:00
|
|
|
auto maybe_pghs = Paragraphs::get_paragraphs(localappdata / "vcpkg" / "config");
|
|
|
|
if (auto p_pghs = maybe_pghs.get())
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
2017-03-30 00:49:09 +08:00
|
|
|
const auto& pghs = *p_pghs;
|
|
|
|
|
|
|
|
std::unordered_map<std::string, std::string> keys;
|
|
|
|
if (pghs.size() > 0)
|
|
|
|
keys = pghs[0];
|
2016-09-19 11:50:08 +08:00
|
|
|
|
2017-03-30 00:49:09 +08:00
|
|
|
for (size_t x = 1; x < pghs.size(); ++x)
|
|
|
|
{
|
|
|
|
for (auto&& p : pghs[x])
|
|
|
|
keys.insert(p);
|
|
|
|
}
|
2016-09-19 11:50:08 +08:00
|
|
|
|
2017-03-30 00:49:09 +08:00
|
|
|
auto user_id = keys["User-Id"];
|
|
|
|
auto user_time = keys["User-Since"];
|
2017-04-01 08:00:24 +08:00
|
|
|
if (!user_id.empty() && !user_time.empty())
|
|
|
|
{
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::set_user_information(user_id, user_time);
|
2017-04-01 08:00:24 +08:00
|
|
|
return;
|
|
|
|
}
|
2017-03-30 00:49:09 +08:00
|
|
|
}
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
2017-04-01 08:00:24 +08:00
|
|
|
catch (...) { }
|
2016-09-19 11:50:08 +08:00
|
|
|
|
|
|
|
// config file not found, could not be read, or invalid
|
|
|
|
std::string user_id, user_time;
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::init_user_information(user_id, user_time);
|
|
|
|
Metrics::set_user_information(user_id, user_time);
|
2016-09-19 11:50:08 +08:00
|
|
|
try
|
|
|
|
{
|
|
|
|
std::error_code ec;
|
|
|
|
fs::create_directory(localappdata / "vcpkg", ec);
|
|
|
|
std::ofstream(localappdata / "vcpkg" / "config", std::ios_base::out | std::ios_base::trunc)
|
|
|
|
<< "User-Id: " << user_id << "\n"
|
|
|
|
<< "User-Since: " << user_time << "\n";
|
|
|
|
}
|
2017-04-01 08:00:24 +08:00
|
|
|
catch (...) { }
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static std::string trim_path_from_command_line(const std::string& full_command_line)
|
|
|
|
{
|
2017-03-14 08:38:04 +08:00
|
|
|
Checks::check_exit(VCPKG_LINE_INFO, full_command_line.size() > 0, "Internal failure - cannot have empty command line");
|
2016-09-19 11:50:08 +08:00
|
|
|
|
|
|
|
if (full_command_line[0] == '"')
|
|
|
|
{
|
|
|
|
auto it = std::find(full_command_line.cbegin() + 1, full_command_line.cend(), '"');
|
|
|
|
if (it != full_command_line.cend()) // Skip over the quote
|
|
|
|
++it;
|
|
|
|
while (it != full_command_line.cend() && *it == ' ') // Skip over a space
|
|
|
|
++it;
|
|
|
|
return std::string(it, full_command_line.cend());
|
|
|
|
}
|
|
|
|
|
|
|
|
auto it = std::find(full_command_line.cbegin(), full_command_line.cend(), ' ');
|
|
|
|
while (it != full_command_line.cend() && *it == ' ')
|
|
|
|
++it;
|
|
|
|
return std::string(it, full_command_line.cend());
|
|
|
|
}
|
|
|
|
|
2017-03-04 22:25:05 +08:00
|
|
|
static ElapsedTime g_timer;
|
|
|
|
|
2016-09-19 11:50:08 +08:00
|
|
|
int wmain(const int argc, const wchar_t* const* const argv)
|
|
|
|
{
|
|
|
|
if (argc == 0)
|
|
|
|
std::abort();
|
|
|
|
|
2017-03-04 22:25:05 +08:00
|
|
|
g_timer = ElapsedTime::createStarted();
|
2016-09-19 11:50:08 +08:00
|
|
|
atexit([]()
|
|
|
|
{
|
2017-03-04 22:25:05 +08:00
|
|
|
auto elapsed_us = g_timer.microseconds();
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::track_metric("elapsed_us", elapsed_us);
|
|
|
|
Metrics::flush();
|
2016-09-19 11:50:08 +08:00
|
|
|
});
|
|
|
|
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::track_property("version", Commands::Version::version());
|
2016-09-19 11:50:08 +08:00
|
|
|
|
|
|
|
const std::string trimmed_command_line = trim_path_from_command_line(Strings::utf16_to_utf8(GetCommandLineW()));
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::track_property("cmdline", trimmed_command_line);
|
2016-09-19 11:50:08 +08:00
|
|
|
loadConfig();
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::track_property("sqmuser", Metrics::get_SQM_user());
|
2016-09-19 11:50:08 +08:00
|
|
|
|
|
|
|
const vcpkg_cmd_arguments args = vcpkg_cmd_arguments::create_from_command_line(argc, argv);
|
|
|
|
|
2017-04-04 05:43:44 +08:00
|
|
|
if (args.printmetrics != OptBoolT::UNSPECIFIED)
|
|
|
|
Metrics::set_print_metrics(args.printmetrics == OptBoolT::ENABLED);
|
|
|
|
if (args.sendmetrics != OptBoolT::UNSPECIFIED)
|
|
|
|
Metrics::set_send_metrics(args.sendmetrics == OptBoolT::ENABLED);
|
2016-09-19 11:50:08 +08:00
|
|
|
|
2017-04-04 05:43:44 +08:00
|
|
|
if (args.debug != OptBoolT::UNSPECIFIED)
|
2016-09-19 11:50:08 +08:00
|
|
|
{
|
2017-04-04 05:43:44 +08:00
|
|
|
g_debugging = (args.debug == OptBoolT::ENABLED);
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if (g_debugging)
|
|
|
|
{
|
|
|
|
inner(args);
|
2017-03-23 08:46:05 +08:00
|
|
|
Checks::exit_fail(VCPKG_LINE_INFO);
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
std::string exc_msg;
|
|
|
|
try
|
|
|
|
{
|
|
|
|
inner(args);
|
2017-03-23 08:46:05 +08:00
|
|
|
Checks::exit_fail(VCPKG_LINE_INFO);
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|
|
|
|
catch (std::exception& e)
|
|
|
|
{
|
|
|
|
exc_msg = e.what();
|
|
|
|
}
|
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
exc_msg = "unknown error(...)";
|
|
|
|
}
|
2017-04-04 05:41:36 +08:00
|
|
|
Metrics::track_property("error", exc_msg);
|
2017-03-04 22:10:59 +08:00
|
|
|
|
|
|
|
fflush(stdout);
|
|
|
|
System::print(
|
2017-04-01 08:00:24 +08:00
|
|
|
"vcpkg.exe has crashed.\n"
|
|
|
|
"Please send an email to:\n"
|
|
|
|
" %s\n"
|
|
|
|
"containing a brief summary of what you were trying to do and the following data blob:\n"
|
|
|
|
"\n"
|
|
|
|
"Version=%s\n"
|
|
|
|
"EXCEPTION='%s'\n"
|
|
|
|
"CMD=\n",
|
|
|
|
Commands::Contact::email(),
|
|
|
|
Commands::Version::version(),
|
|
|
|
exc_msg);
|
2017-03-04 22:10:59 +08:00
|
|
|
fflush(stdout);
|
2016-09-19 11:50:08 +08:00
|
|
|
for (int x = 0; x < argc; ++x)
|
2017-03-04 22:10:59 +08:00
|
|
|
System::println("%s|", Strings::utf16_to_utf8(argv[x]));
|
|
|
|
fflush(stdout);
|
2016-09-19 11:50:08 +08:00
|
|
|
}
|