1 /*
2 * Copyright (C) 2014 Andrew Duggan
3 * Copyright (C) 2014 Synaptics Inc
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17
18 #include <stdio.h>
19 #include <string.h>
20 #include <errno.h>
21 #include <getopt.h>
22 #include <sys/types.h>
23 #include <sys/stat.h>
24 #include <fcntl.h>
25 #include <dirent.h>
26 #include <unistd.h>
27 #include <time.h>
28 #include <string>
29 #include <sstream>
30
31 #include "hiddevice.h"
32 #include "rmi4update.h"
33
34 #define VERSION_MAJOR 1
35 #define VERSION_MINOR 2
36 #define VERSION_SUBMINOR 0
37
38 #define RMI4UPDATE_GETOPTS "hfd:plv"
39
printHelp(const char * prog_name)40 void printHelp(const char *prog_name)
41 {
42 fprintf(stdout, "Usage: %s [OPTIONS] FIRMWAREFILE\n", prog_name);
43 fprintf(stdout, "\t-h, --help\tPrint this message\n");
44 fprintf(stdout, "\t-f, --force\tForce updating firmware even it the image provided is older\n\t\t\tthen the current firmware on the device.\n");
45 fprintf(stdout, "\t-d, --device\thidraw device file associated with the device being updated.\n");
46 fprintf(stdout, "\t-p, --fw-props\tPrint the firmware properties.\n");
47 fprintf(stdout, "\t-l, --lockdown\tPerform lockdown.\n");
48 fprintf(stdout, "\t-v, --version\tPrint version number.\n");
49 }
50
printVersion()51 void printVersion()
52 {
53 fprintf(stdout, "rmi4update version %d.%d.%d\n",
54 VERSION_MAJOR, VERSION_MINOR, VERSION_SUBMINOR);
55 }
56
UpdateDevice(FirmwareImage & image,bool force,bool performLockdown,const char * deviceFile)57 int UpdateDevice(FirmwareImage & image, bool force, bool performLockdown, const char * deviceFile)
58 {
59 HIDDevice rmidevice;
60 int rc;
61
62 rc = rmidevice.Open(deviceFile);
63 if (rc)
64 return rc;
65
66 RMI4Update update(rmidevice, image);
67 rc = update.UpdateFirmware(force, performLockdown);
68 if (rc != UPDATE_SUCCESS)
69 return rc;
70
71 return rc;
72 }
73
GetFirmwareProps(const char * deviceFile,std::string & props)74 int GetFirmwareProps(const char * deviceFile, std::string &props)
75 {
76 HIDDevice rmidevice;
77 int rc = UPDATE_SUCCESS;
78 std::stringstream ss;
79
80 rc = rmidevice.Open(deviceFile);
81 if (rc)
82 return rc;
83
84 rmidevice.ScanPDT(0x1);
85 rmidevice.QueryBasicProperties();
86
87 ss << rmidevice.GetFirmwareVersionMajor() << "."
88 << rmidevice.GetFirmwareVersionMinor() << "."
89 << std::hex << rmidevice.GetFirmwareID();
90
91 if (rmidevice.InBootloader())
92 ss << " bootloader";
93
94 props = ss.str();
95
96 return rc;
97 }
98
main(int argc,char ** argv)99 int main(int argc, char **argv)
100 {
101 int rc;
102 FirmwareImage image;
103 int opt;
104 int index;
105 char *deviceName = NULL;
106 const char *firmwareName = NULL;
107 bool force = false;
108 static struct option long_options[] = {
109 {"help", 0, NULL, 'h'},
110 {"force", 0, NULL, 'f'},
111 {"device", 1, NULL, 'd'},
112 {"fw-props", 0, NULL, 'p'},
113 {"lockdown", 0, NULL, 'l'},
114 {"version", 0, NULL, 'v'},
115 {0, 0, 0, 0},
116 };
117 struct dirent * devDirEntry;
118 DIR * devDir;
119 bool printFirmwareProps = false;
120 bool performLockdown = false;
121
122 while ((opt = getopt_long(argc, argv, RMI4UPDATE_GETOPTS, long_options, &index)) != -1) {
123 switch (opt) {
124 case 'h':
125 printHelp(argv[0]);
126 return 0;
127 case 'f':
128 force = true;
129 break;
130 case 'd':
131 deviceName = optarg;
132 break;
133 case 'p':
134 printFirmwareProps = true;
135 break;
136 case 'l':
137 performLockdown = true;
138 break;
139 case 'v':
140 printVersion();
141 return 0;
142 default:
143 break;
144
145 }
146 }
147
148 if (printFirmwareProps) {
149 std::string props;
150
151 if (!deviceName) {
152 fprintf(stderr, "Specifiy which device to query\n");
153 return 1;
154 }
155 rc = GetFirmwareProps(deviceName, props);
156 if (rc) {
157 fprintf(stderr, "Failed to read properties from device: %s\n", update_err_to_string(rc));
158 return 1;
159 }
160 fprintf(stdout, "%s\n", props.c_str());
161 return 0;
162 }
163
164 if (optind < argc) {
165 firmwareName = argv[optind];
166 } else {
167 printHelp(argv[0]);
168 return -1;
169 }
170
171 rc = image.Initialize(firmwareName);
172 if (rc != UPDATE_SUCCESS) {
173 fprintf(stderr, "Failed to initialize the firmware image: %s\n", update_err_to_string(rc));
174 return 1;
175 }
176
177 if (deviceName) {
178 rc = UpdateDevice(image, force, performLockdown, deviceName);
179
180 return rc;
181 } else {
182 char deviceFile[PATH_MAX];
183 bool found = false;
184
185 devDir = opendir("/dev");
186 if (!devDir)
187 return -1;
188
189 while ((devDirEntry = readdir(devDir)) != NULL) {
190 if (strstr(devDirEntry->d_name, "hidraw")) {
191 char rawDevice[PATH_MAX];
192 strncpy(rawDevice, devDirEntry->d_name, PATH_MAX);
193 snprintf(deviceFile, PATH_MAX, "/dev/%s", devDirEntry->d_name);
194 rc = UpdateDevice(image, force, performLockdown, deviceFile);
195 if (rc != 0) {
196 continue;
197 } else {
198 found = true;
199 break;
200 }
201 }
202 }
203 closedir(devDir);
204
205 if (!found)
206 return rc;
207 }
208
209 return 0;
210 }
211