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
|
//
// Copyright 2015 Ettus Research LLC
//
// This program 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.
//
// This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
//
#include <csignal>
#include <cstdlib>
#include <iostream>
#include <boost/assign.hpp>
#include <boost/filesystem.hpp>
#include <boost/format.hpp>
#include <uhd/config.hpp>
#include <uhd/image_loader.hpp>
#include <uhd/types/device_addr.hpp>
#include <uhd/utils/safe_main.hpp>
#include <boost/program_options.hpp>
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[]){
std::string fw_path = "";
std::string fpga_path = "";
po::options_description desc("Allowed options");
desc.add_options()
("help", "help message")
("args", po::value<std::string>()->default_value(""), "Device args, optional loader args")
("fw-path", po::value<std::string>(&fw_path)->default_value(""), "Firmware path (uses default if none specified)")
("fpga-path", po::value<std::string>(&fpga_path)->default_value(""), "FPGA path (uses default if none specified)")
("no-fw", "Don't burn firmware")
("no-fpga", "Don't burn FPGA")
;
po::variables_map vm;
po::store(po::parse_command_line(argc, argv, desc), 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.firmware_path = vm["fw-path"].as<std::string>();
image_loader_args.fpga_path = vm["fpga-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 != ""){
#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 != ""){
#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();
}
// 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;
}
|
