ps3stor_region.c

   1
   2 /*
   3  * This program is free software; you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License as published by
   5  * the Free Software Foundation; version 2 of the License.
   6  *
   7  * This program is distributed in the hope that it will be useful,
   8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10  * GNU General Public License for more details.
  11  *
  12  * You should have received a copy of the GNU General Public License
  13  * along with this program; if not, write to the Free Software
  14  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  15  */
  16
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <string.h>
  20 #include <getopt.h>
  21 #include <errno.h>
  22
  23 #include "ps3stor_mgr.h"
  24
  25 #define PS3STOR_REGION_VERSION          "0.0.1"
  26
  27 struct opts
  28 {
  29         char *device_name;
  30         char *cmd;
  31         int do_help;
  32         int do_verbose;
  33         int do_version;
  34 };
  35
  36 static struct option long_opts[] = {
  37         { "help",       no_argument, NULL, 'h' },
  38         { "verbose",    no_argument, NULL, 'v' },
  39         { "version",    no_argument, NULL, 'V' },
  40         { NULL, 0, NULL, 0 }
  41 };
  42
  43 /*
  44  * usage
  45  */
  46 static void usage(void) {
  47         fprintf(stderr,
  48                 "Usage: ps3stor_region [OPTIONS] DEVICE COMMAND [ARGS]\n"
  49                 "\n"
  50                 "Options:\n"
  51                 "       -h, --help                              Show this message and exit\n"
  52                 "       -v, --verbose                           Increase verbosity\n"
  53                 "       -V, --version                           Show version information and exit\n"
  54                 "Commands:\n"
  55                 "       create DEVID START COUNT LAID           Creates storage region\n"
  56                 "       delete DEVID REGIONID                   Deletes storage region\n"
  57                 "       set_acl DEVID REGIONID LAID RIGHTS      Sets region access rights\n"
  58                 "       get_acl DEVID REGIONID INDEX            Returns region access rights\n"
  59                 "\n\n"
  60                 "Simple example: Create a HDD region:\n"
  61                 "       ps3stor_region /dev/ps3stormgr create 3 0x8 0x1000 0x1070000002000001\n");
  62 }
  63
  64 /*
  65  * version
  66  */
  67 static void version(void)
  68 {
  69         fprintf(stderr,
  70                 "ps3stor_region " PS3STOR_REGION_VERSION "\n"
  71                 "Copyright (C) 2011 graf_chokolo <grafchokolo@googlemail.com>\n"
  72                 "This is free software.  You may redistribute copies of it "
  73                 "under the terms of\n"
  74                 "the GNU General Public License 2 "
  75                 "<http://www.gnu.org/licenses/gpl2.html>.\n"
  76                 "There is NO WARRANTY, to the extent permitted by law.\n");
  77 }
  78
  79 /*
  80  * process_opts
  81  */
  82 static int process_opts(int argc, char **argv, struct opts *opts)
  83 {
  84         int c;
  85
  86         while ((c = getopt_long(argc, argv, "hvV", long_opts, NULL)) != -1) {
  87                 switch (c) {
  88                 case 'h':
  89                 case '?':
  90                         opts->do_help = 1;
  91                         return 0;
  92
  93                 case 'v':
  94                         opts->do_verbose++;
  95                         break;
  96
  97                 case 'V':
  98                         opts->do_version = 1;
  99                         return 0;
 100
 101                 default:
 102                         fprintf(stderr, "Invalid command option: %c\n", c);
 103                         return -1;
 104                 }
 105         }
 106
 107         if (optind >= argc) {
 108                 fprintf(stderr, "No device specified\n");
 109                 return -1;
 110         }
 111
 112         opts->device_name = argv[optind];
 113         optind++;
 114
 115         if (optind >= argc) {
 116                 fprintf(stderr, "No command specified\n");
 117                 return -1;
 118         }
 119
 120         opts->cmd = argv[optind];
 121         optind++;
 122
 123         return 0;
 124 }
 125
 126 /*
 127  * cmd_create
 128  */
 129 static int cmd_create(int fd, struct opts *opts, int argc, char **argv)
 130 {
 131         uint64_t dev_id, start_sector, sector_count, laid, region_id;
 132         char *endptr;
 133         int error;
 134
 135         if (optind >= argc) {
 136                 fprintf(stderr, "No device identifier specified\n");
 137                 return -1;
 138         }
 139
 140         dev_id = strtoull(argv[optind], &endptr, 0);
 141         if (*endptr != '\0') {
 142                 fprintf(stderr, "Invalid device identifier specified\n");
 143                 return -1;
 144         }
 145
 146         optind++;
 147
 148         if (optind >= argc) {
 149                 fprintf(stderr, "No start sector specified\n");
 150                 return -1;
 151         }
 152
 153         start_sector = strtoull(argv[optind], &endptr, 0);
 154         if (*endptr != '\0') {
 155                 fprintf(stderr, "Invalid start sector specified\n");
 156                 return -1;
 157         }
 158
 159         optind++;
 160
 161         if (optind >= argc) {
 162                 fprintf(stderr, "No sector count specified\n");
 163                 return -1;
 164         }
 165
 166         sector_count = strtoull(argv[optind], &endptr, 0);
 167         if (*endptr != '\0') {
 168                 fprintf(stderr, "Invalid sector count specified\n");
 169                 return -1;
 170         }
 171
 172         optind++;
 173
 174         if (optind >= argc) {
 175                 fprintf(stderr, "No LPAR authentication identifier specified\n");
 176                 return -1;
 177         }
 178
 179         laid = strtoull(argv[optind], &endptr, 0);
 180         if (*endptr != '\0') {
 181                 fprintf(stderr, "Invalid LPAR authentication identifier specified\n");
 182                 return -1;
 183         }
 184
 185         optind++;
 186
 187         error = ps3stor_mgr_create_region(fd, dev_id, start_sector, sector_count, laid,
 188                 &region_id);
 189
 190         if (error)
 191                 fprintf(stderr, "%s: %s\n", opts->device_name, strerror(errno));
 192         else
 193                 fprintf(stdout, "0x%016lx\n", region_id);
 194
 195         return error;
 196 }
 197
 198 /*
 199  * cmd_delete
 200  */
 201 static int cmd_delete(int fd, struct opts *opts, int argc, char **argv)
 202 {
 203         uint64_t dev_id, region_id;
 204         char *endptr;
 205         int error;
 206
 207         if (optind >= argc) {
 208                 fprintf(stderr, "No device identifier specified\n");
 209                 return -1;
 210         }
 211
 212         dev_id = strtoull(argv[optind], &endptr, 0);
 213         if (*endptr != '\0') {
 214                 fprintf(stderr, "Invalid device identifier specified\n");
 215                 return -1;
 216         }
 217
 218         optind++;
 219
 220         if (optind >= argc) {
 221                 fprintf(stderr, "No region identifier specified\n");
 222                 return -1;
 223         }
 224
 225         region_id = strtoull(argv[optind], &endptr, 0);
 226         if (*endptr != '\0') {
 227                 fprintf(stderr, "Invalid region identifier specified\n");
 228                 return -1;
 229         }
 230
 231         optind++;
 232
 233         error = ps3stor_mgr_delete_region(fd, dev_id, region_id);
 234
 235         if (error)
 236                 fprintf(stderr, "%s: %s\n", opts->device_name, strerror(errno));
 237
 238         return error;
 239 }
 240
 241 /*
 242  * cmd_set_acl
 243  */
 244 static int cmd_set_acl(int fd, struct opts *opts, int argc, char **argv)
 245 {
 246         uint64_t dev_id, region_id, laid, access_rights;
 247         char *endptr;
 248         int error;
 249
 250         if (optind >= argc) {
 251                 fprintf(stderr, "No device identifier specified\n");
 252                 return -1;
 253         }
 254
 255         dev_id = strtoull(argv[optind], &endptr, 0);
 256         if (*endptr != '\0') {
 257                 fprintf(stderr, "Invalid device identifier specified\n");
 258                 return -1;
 259         }
 260
 261         optind++;
 262
 263         if (optind >= argc) {
 264                 fprintf(stderr, "No region identifier specified\n");
 265                 return -1;
 266         }
 267
 268         region_id = strtoull(argv[optind], &endptr, 0);
 269         if (*endptr != '\0') {
 270                 fprintf(stderr, "Invalid region identifier specified\n");
 271                 return -1;
 272         }
 273
 274         optind++;
 275
 276         if (optind >= argc) {
 277                 fprintf(stderr, "No LPAR authentication identifier specified\n");
 278                 return -1;
 279         }
 280
 281         laid = strtoull(argv[optind], &endptr, 0);
 282         if (*endptr != '\0') {
 283                 fprintf(stderr, "Invalid LPAR authentication identifier specified\n");
 284                 return -1;
 285         }
 286
 287         optind++;
 288
 289         if (optind >= argc) {
 290                 fprintf(stderr, "No access rights specified\n");
 291                 return -1;
 292         }
 293
 294         access_rights = strtoull(argv[optind], &endptr, 0);
 295         if (*endptr != '\0') {
 296                 fprintf(stderr, "Invalid access rights specified\n");
 297                 return -1;
 298         }
 299
 300         optind++;
 301
 302         error = ps3stor_mgr_set_region_acl(fd, dev_id, region_id, laid, access_rights);
 303
 304         if (error)
 305                 fprintf(stderr, "%s: %s\n", opts->device_name, strerror(errno));
 306
 307         return error;
 308 }
 309
 310 /*
 311  * cmd_get_acl
 312  */
 313 static int cmd_get_acl(int fd, struct opts *opts, int argc, char **argv)
 314 {
 315         uint64_t dev_id, region_id, entry_index, laid, access_rights;
 316         char *endptr;
 317         int error;
 318
 319         if (optind >= argc) {
 320                 fprintf(stderr, "No device identifier specified\n");
 321                 return -1;
 322         }
 323
 324         dev_id = strtoull(argv[optind], &endptr, 0);
 325         if (*endptr != '\0') {
 326                 fprintf(stderr, "Invalid device identifier specified\n");
 327                 return -1;
 328         }
 329
 330         optind++;
 331
 332         if (optind >= argc) {
 333                 fprintf(stderr, "No region identifier specified\n");
 334                 return -1;
 335         }
 336
 337         region_id = strtoull(argv[optind], &endptr, 0);
 338         if (*endptr != '\0') {
 339                 fprintf(stderr, "Invalid region identifier specified\n");
 340                 return -1;
 341         }
 342
 343         optind++;
 344
 345         if (optind >= argc) {
 346                 fprintf(stderr, "No entry index specified\n");
 347                 return -1;
 348         }
 349
 350         entry_index = strtoull(argv[optind], &endptr, 0);
 351         if (*endptr != '\0') {
 352                 fprintf(stderr, "Invalid entry index specified\n");
 353                 return -1;
 354         }
 355
 356         optind++;
 357
 358         error = ps3stor_mgr_get_region_acl(fd, dev_id, region_id, entry_index, &laid, &access_rights);
 359
 360         if (error)
 361                 fprintf(stderr, "%s: %s\n", opts->device_name, strerror(errno));
 362         else
 363                 fprintf(stdout, "0x%016lx 0x%016lx\n", laid, access_rights);
 364
 365         return error;
 366 }
 367
 368 /*
 369  * main
 370  */
 371 int main(int argc, char **argv)
 372 {
 373         struct opts opts;
 374         int fd = 0, error = 0;
 375
 376         memset(&opts, 0, sizeof(opts));
 377
 378         if (process_opts(argc, argv, &opts)) {
 379                 usage();
 380                 error = 1;
 381                 goto done;
 382         }
 383
 384         if (opts.do_help) {
 385                 usage();
 386                 goto done;
 387         } else if (opts.do_version) {
 388                 version();
 389                 goto done;
 390         }
 391
 392         fd = ps3stor_mgr_open(opts.device_name);
 393         if (fd < 0) {
 394                 fprintf(stderr, "%s: %s\n", opts.device_name, strerror(errno));
 395                 error = 2;
 396                 goto done;
 397         }
 398
 399         if (!strcmp(opts.cmd, "create")) {
 400                 error = cmd_create(fd, &opts, argc, argv);
 401         } else if (!strcmp(opts.cmd, "delete")) {
 402                 error = cmd_delete(fd, &opts, argc, argv);
 403         } else if (!strcmp(opts.cmd, "set_acl")) {
 404                 error = cmd_set_acl(fd, &opts, argc, argv);
 405         } else if (!strcmp(opts.cmd, "get_acl")) {
 406                 error = cmd_get_acl(fd, &opts, argc, argv);
 407         } else {
 408                 usage();
 409                 error = 1;
 410                 goto done;
 411         }
 412
 413         if (error)
 414                 error = 3;
 415
 416 done:
 417
 418         if (fd >= 0)
 419                 ps3stor_mgr_close(fd);
 420
 421         exit(error);
 422 }