blob: 1771cad24844352617c949a4f860d6aaf1a258f9 [file] [log] [blame]
/*
* Copyright (c) 2004-2015 Douglas Gilbert.
* All rights reserved.
* Use of this source code is governed by a BSD-style
* license that can be found in the BSD_LICENSE file.
*/
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <getopt.h>
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "sg_lib.h"
#include "sg_cmds_basic.h"
#include "sg_pr2serr.h"
/* A utility program originally written for the Linux OS SCSI subsystem.
*
* This program issues the SCSI PREVENT ALLOW MEDIUM REMOVAL command to the
* given SCSI device.
*/
static const char * version_str = "1.08 20151219";
#define ME "sg_prevent: "
static struct option long_options[] = {
{"allow", 0, 0, 'a'},
{"help", 0, 0, 'h'},
{"prevent", 1, 0, 'p'},
{"verbose", 0, 0, 'v'},
{"version", 0, 0, 'V'},
{0, 0, 0, 0},
};
static void usage()
{
pr2serr("Usage: "
"sg_prevent [--allow] [--help] [--prevent=PC] [--verbose] "
"[--version]\n"
" DEVICE\n"
" where:\n"
" --allow|-a allow media removal\n"
" --help|-h print usage message then exit\n"
" --prevent=PC|-p PC prevent code value (def: 1 -> "
"prevent)\n"
" 0 -> allow, 1 -> prevent\n"
" 2 -> persistent allow, 3 -> "
"persistent prevent\n"
" --verbose|-v increase verbosity\n"
" --version|-V print version string and exit\n\n"
"Performs a SCSI PREVENT ALLOW MEDIUM REMOVAL command\n");
}
int main(int argc, char * argv[])
{
int sg_fd, res, c;
int allow = 0;
int prevent = -1;
int verbose = 0;
const char * device_name = NULL;
int ret = 0;
while (1) {
int option_index = 0;
c = getopt_long(argc, argv, "ahp:vV", long_options,
&option_index);
if (c == -1)
break;
switch (c) {
case 'a':
allow = 1;
break;
case 'h':
case '?':
usage();
return 0;
case 'p':
prevent = sg_get_num(optarg);
if ((prevent < 0) || (prevent > 3)) {
pr2serr("bad argument to '--prevent'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'v':
++verbose;
break;
case 'V':
pr2serr(ME "version: %s\n", version_str);
return 0;
default:
pr2serr("unrecognised option code 0x%x ??\n", c);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (optind < argc) {
if (NULL == device_name) {
device_name = argv[optind];
++optind;
}
if (optind < argc) {
for (; optind < argc; ++optind)
pr2serr("Unexpected extra argument: %s\n", argv[optind]);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (NULL == device_name) {
pr2serr("missing device name!\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
if (allow && (prevent >= 0)) {
pr2serr("can't give both '--allow' and '--prevent='\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
if (allow)
prevent = 0;
else if (prevent < 0)
prevent = 1; /* default is to prevent, as utility name suggests */
sg_fd = sg_cmds_open_device(device_name, 0 /* rw */, verbose);
if (sg_fd < 0) {
pr2serr(ME "open error: %s: %s\n", device_name,
safe_strerror(-sg_fd));
return SG_LIB_FILE_ERROR;
}
res = sg_ll_prevent_allow(sg_fd, prevent, 1, verbose);
ret = res;
if (res) {
char b[80];
sg_get_category_sense_str(res, sizeof(b), b, verbose);
pr2serr("Prevent allow medium removal: %s\n", b);
}
res = sg_cmds_close_device(sg_fd);
if (res < 0) {
pr2serr("close error: %s\n", safe_strerror(-res));
if (0 == ret)
return SG_LIB_FILE_ERROR;
}
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}