1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
//
// Copyright 2015-2017 Ettus Research
// Copyright 2018 Ettus Research, a National Instruments Company
//
// SPDX-License-Identifier: GPL-3.0-or-later
//
#include <uhd/config.hpp>
#include <uhd/image_loader.hpp>
#include <uhd/types/device_addr.hpp>
#include <uhd/utils/safe_main.hpp>
#include <boost/assign.hpp>
#include <boost/filesystem.hpp>
#include <boost/format.hpp>
#include <boost/program_options.hpp>
#include <csignal>
#include <cstdlib>
#include <iostream>
namespace fs = boost::filesystem;
namespace po = boost::program_options;
static std::string device_type = "";
static int num_ctrl_c = 0;
/*
* If the user presses Ctrl+C, warn them that they may corrupt their device.
* If they press it again, provide instructions on restoring the device
* (if applicable) and exit.
*/
void sigint_handler(int)
{
num_ctrl_c++;
if (num_ctrl_c == 1) {
std::cout << std::endl
<< "Are you sure you want to abort? If you do, your device will likely"
<< std::endl
<< "be in an unstable or unusable state." << std::endl
<< "Press Ctrl+C again to abort." << std::endl
<< std::endl;
} else {
std::cout << std::endl
<< uhd::image_loader::get_recovery_instructions(device_type)
<< std::endl;
exit(EXIT_FAILURE);
}
}
int UHD_SAFE_MAIN(int argc, char* argv[])
{
po::options_description desc("Allowed options");
// clang-format off
desc.add_options()
("help", "help message")
("args", po::value<std::string>()->default_value(""), "Device args, optional loader args")
("fw-path", po::value<std::string>()->default_value(""), "Firmware path (uses default if none specified)")
("fpga-path", po::value<std::string>()->default_value(""), "FPGA path (uses default if none specified)")
("out-path", po::value<std::string>()->default_value(""), "Output path/filename of the downloaded FPGA .bit file")
("no-fw", "Don't burn firmware")
("no-fpga", "Don't Burn FPGA")
("download", "Download an image to a bit/bin file")
;
// clang-format on
po::variables_map vm;
po::positional_options_description pos_desc;
po::store(
po::command_line_parser(argc, argv).options(desc).positional(pos_desc).run(), vm);
po::notify(vm);
// Help message
if (vm.count("help")) {
std::cout << "UHD Image Loader" << std::endl
<< std::endl
<< "Load firmware and/or FPGA images onto an Ettus Research device."
<< std::endl
<< std::endl
<< desc << std::endl;
return EXIT_FAILURE;
}
// Convert user options
uhd::image_loader::image_loader_args_t image_loader_args;
image_loader_args.args = vm["args"].as<std::string>();
image_loader_args.load_firmware = (vm.count("no-fw") == 0);
image_loader_args.load_fpga = (vm.count("no-fpga") == 0);
image_loader_args.download = (vm.count("download") != 0);
image_loader_args.firmware_path = vm["fw-path"].as<std::string>();
image_loader_args.fpga_path = vm["fpga-path"].as<std::string>();
image_loader_args.out_path = vm["out-path"].as<std::string>();
// Force user to specify a device
if (not image_loader_args.args.has_key("type")) {
throw uhd::runtime_error("You must specify a device type.");
}
// Clean up paths, if given
if (!image_loader_args.firmware_path.empty()) {
#ifndef UHD_PLATFORM_WIN32
if (image_loader_args.firmware_path.find("~") == 0) {
image_loader_args.firmware_path.replace(0, 1, getenv("HOME"));
}
#endif /* UHD_PLATFORM_WIN32 */
image_loader_args.firmware_path =
fs::absolute(image_loader_args.firmware_path).string();
}
if (!image_loader_args.fpga_path.empty()) {
#ifndef UHD_PLATFORM_WIN32
if (image_loader_args.fpga_path.find("~") == 0) {
image_loader_args.fpga_path.replace(0, 1, getenv("HOME"));
}
#endif /* UHD_PLATFORM_WIN32 */
image_loader_args.fpga_path = fs::absolute(image_loader_args.fpga_path).string();
}
if (!image_loader_args.out_path.empty()) {
#ifndef UHD_PLATFORM_WIN32
if (image_loader_args.out_path.find("~") == 0) {
image_loader_args.out_path.replace(0, 1, getenv("HOME"));
}
#endif /* UHD_PLATFORM_WIN32 */
image_loader_args.out_path = fs::absolute(image_loader_args.out_path).string();
}
// Detect which type of device we're working with
device_type = image_loader_args.args.get("type", "");
std::signal(SIGINT, &sigint_handler);
if (not uhd::image_loader::load(image_loader_args)) {
std::cerr << "No applicable UHD devices found" << std::endl;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|
