blob: bb1fb3558568230aac92f194f0c9205a881aea9d (
plain)
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
|
//
// Copyright 2014-2015 Ettus Research LLC
// Copyright 2018 Ettus Research, a National Instruments Company
//
// SPDX-License-Identifier: GPL-3.0-or-later
//
#include <uhd/exception.hpp>
#include <uhd/image_loader.hpp>
#include <uhd/utils/log.hpp>
#include <uhd/utils/static.hpp>
#include <boost/filesystem.hpp>
#include <boost/format.hpp>
#include <iostream>
#include <map>
#include <utility>
namespace fs = boost::filesystem;
typedef std::map<std::string, uhd::image_loader::loader_fcn_t> loader_fcn_map_t;
typedef std::map<std::string, std::string> string_map_t;
// Nice typedefs for iterating over std::map
typedef std::pair<std::string, uhd::image_loader::loader_fcn_t> loader_fcn_pair_t;
typedef std::pair<std::string, std::string> string_pair_t;
UHD_SINGLETON_FCN(loader_fcn_map_t, get_image_loaders);
UHD_SINGLETON_FCN(string_map_t, get_recovery_strings);
/*
* Registration
*/
void uhd::image_loader::register_image_loader(const std::string& device_type,
const loader_fcn_t& loader_fcn,
const std::string& recovery_instructions)
{
// UHD_LOGGER_TRACE("UHD") << "Registering image loader and recovery instructions for
// "
// << device_type;
get_image_loaders().insert(loader_fcn_pair_t(device_type, loader_fcn));
get_recovery_strings().insert(string_pair_t(device_type, recovery_instructions));
}
/*
* Actual loading
*/
bool uhd::image_loader::load(
const uhd::image_loader::image_loader_args_t& image_loader_args)
{
// If "type=foo" given in args, see if we have an image loader for that
if (image_loader_args.args.has_key("type")) {
std::string type = image_loader_args.args.get("type");
if (get_image_loaders().find(type) == get_image_loaders().end()) {
throw uhd::runtime_error(
str(boost::format(
"There is no image loader registered for given type \"%s\".")
% type));
} else
return get_image_loaders().at(type)(image_loader_args);
} else {
for (const auto& loader_fcn_pair : get_image_loaders()) {
if (loader_fcn_pair.second(image_loader_args))
return true;
}
return false;
}
}
/*
* Get recovery instructions for particular device
*/
std::string uhd::image_loader::get_recovery_instructions(const std::string& device_type)
{
if (get_recovery_strings().count(device_type) == 0) {
return "A firmware or FPGA loading process was interrupted by the user. This can "
"leave your device in a non-working state.";
} else
return get_recovery_strings().at(device_type);
}
|
