handleDevice.c

/*
 * Copyright (c) 2001-2007 Apple Inc. All Rights Reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 * 
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 * 
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 * 
 * @APPLE_LICENSE_HEADER_END@
 */
/*
 * handleDevice.c
 * bless
 *
 * Created by Shantonu Sen <ssen@apple.com> on Thu Dec 6 2001.
 * Copyright (c) 2001-2007 Apple Inc. All Rights Reserved.
 *
 * $Id: handleDevice.c,v 1.54 2006/07/19 00:15:36 ssen Exp $
 *
 *
 */

#include<stdlib.h>
#include<stdio.h>
#include<unistd.h>
#include<paths.h>
#include<sys/stat.h>
#include<sys/mount.h>
#include<sys/paths.h>
#include<string.h>

#include<APFS/APFS.h>

#include"enums.h"
#include"structs.h"

#include"bless.h"
#include"bless_private.h"
#include"protos.h"

intmodeDevice(BLContextPtrcontext, structclargactargs[klast]) {
intret=0;
CFDataReflabeldata=NULL;
CFDataReflabeldata2=NULL;
CFDataRefbootXdata=NULL;
io_object_tdevMediaObj;

BLPreBootEnvTypepreboot;

if(!(geteuid() ==0)) {
blesscontextprintf(context, kBLLogLevelError, "Not run as root\n" );
return1;
 }

ret=BLGetPreBootEnvironmentType(context, &preboot);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Could not determine preboot environment\n");
return1;
 }



/* try to grovel the HFS+ catalog and update a label if present */
if(actargs[klabelfile].present) {
ret=BLLoadFile(context, actargs[klabelfile].argument, 0, &labeldata);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Can't load label '%s'\n",
actargs[klabelfile].argument);
return2;
 }
 } elseif(actargs[klabel].present) {
ret=BLGenerateLabelData(context, actargs[klabel].argument, kBitmapScale_1x, &labeldata);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Can't render scale 1 label '%s'\n",
actargs[klabel].argument);
return3;
 }
ret=BLGenerateLabelData(context, actargs[klabel].argument, kBitmapScale_2x, &labeldata2);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Can't render scale 2 label '%s'\n",
actargs[klabel].argument);
return3;
 }
 }

if(actargs[kbootinfo].present) {
if(!actargs[kbootinfo].hasArg) {
blesscontextprintf(context, kBLLogLevelError,
"BootX file must be specified in Device Mode\n");
return4;
 }

ret=BLLoadFile(context, actargs[kbootinfo].argument, 0, &bootXdata);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Could not load BootX data from %s\n",
actargs[kbootinfo].argument);
 }
 }


devMediaObj=IOServiceGetMatchingService(kIOMasterPortDefault, IOBSDNameMatching(kIOMasterPortDefault, 0,
actargs[kdevice].argument+strlen(_PATH_DEV)));
if (!devMediaObj) {
blesscontextprintf(context, kBLLogLevelError, "Couldn't find I/O Registry information for device %s\n", actargs[kdevice].argument);
return4;
 }
if (IOObjectConformsTo(devMediaObj, APFS_VOLUME_OBJECT)) {
// This is an APFS volume. We need to mess with the preboot volume.
ret=BlessPrebootVolume(context, actargs[kdevice].argument+strlen(_PATH_DEV), NULL, NULL, NULL,
actargs);
 }
IOObjectRelease(devMediaObj);
if (ret) {
blesscontextprintf(context, kBLLogLevelError, "Couldn't set bless data in preboot volume for device %s\n", actargs[kdevice].argument);
return5;
 }

/* Set Open Firmware to boot off the specified volume*/
if(actargs[ksetboot].present) {
if(preboot==kBLPreBootEnvType_EFI) {

ret=setefidevice(context, actargs[kdevice].argument+strlen("/dev/"),
actargs[knextonly].present,
actargs[klegacy].present,
actargs[klegacydrivehint].present ? actargs[klegacydrivehint].argument : NULL,
actargs[koptions].present ? actargs[koptions].argument : NULL,
actargs[kshortform].present ? true : false);
 } else { 
ret=setboot(context, actargs[kdevice].argument, bootXdata, labeldata);
 }

if(ret) {
return3;
 }
 } elseif (labeldata) {
ret=BLSetDiskLabelForDevice(context, actargs[kdevice].argument, labeldata, 1);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Error while setting scale 1 label for %s\n", actargs[kdevice].argument );
return3;
 }
if (labeldata2) {
ret=BLSetDiskLabelForDevice(context, actargs[kdevice].argument, labeldata2, 2);
if(ret) {
blesscontextprintf(context, kBLLogLevelError, "Error while setting scale 2 label for %s\n", actargs[kdevice].argument );
return3;
 }
 }
 }

if (labeldata) CFRelease(labeldata);
if (labeldata2) CFRelease(labeldata2);
if (bootXdata) CFRelease(bootXdata);

return0;
}