freebsd-src/sbin/sysinstall/utils.c
Poul-Henning Kamp 9170aeb809 Made the changes needed here to install from a CDrom.
Cleaned some cruft in the process.
This program becomes curiouser curiouser...
1995-02-13 06:52:16 +00:00

350 lines
6.6 KiB
C

/*
* ----------------------------------------------------------------------------
* "THE BEER-WARE LICENSE" (Revision 42):
* <phk@login.dkuug.dk> wrote this file. As long as you retain this notice you
* can do whatever you want with this stuff. If we meet some day, and you think
* this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp
* ----------------------------------------------------------------------------
*
* $Id: utils.c,v 1.35 1995/01/14 10:31:29 jkh Exp $
*
*/
#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <unistd.h>
#include <string.h>
#include <dialog.h>
#include <errno.h>
#include <sys/file.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/param.h>
#include <sys/mount.h>
#include <sys/reboot.h>
#include <sys/dkbad.h>
#include <sys/disklabel.h>
#include "sysinstall.h"
void
strip_trailing_newlines(char *p)
{
int len = strlen(p);
while (len > 0 && p[len-1] == '\n')
p[--len] = '\0';
}
void
Debug(char *fmt, ...)
{
char *p;
va_list ap;
p = Malloc(2048);
va_start(ap,fmt);
vsnprintf(p, 2048, fmt, ap);
va_end(ap);
write(debug_fd,"Debug <",7);
write(debug_fd,p,strlen(p));
write(debug_fd,">\n\r",3);
free(p);
}
void
TellEm(char *fmt, ...)
{
char *p;
va_list ap;
p = Malloc(2048);
va_start(ap,fmt);
vsnprintf(p, 2048, fmt, ap);
va_end(ap);
strip_trailing_newlines(p);
write(debug_fd,"Progress <",10);
write(debug_fd,p,strlen(p));
write(debug_fd,">\n\r",3);
if(!on_serial) {
dialog_clear_norefresh();
dialog_msgbox("Progress", p, -1, -1, 0);
}
free(p);
}
void
Fatal(char *fmt, ...)
{
char *p;
va_list ap;
p = Malloc(2048);
va_start(ap,fmt);
vsnprintf(p, 2048, fmt, ap);
va_end(ap);
strip_trailing_newlines(p);
if (dialog_active && !on_serial)
dialog_msgbox("Fatal", p, -1, -1, 1);
else
fprintf(stderr, "Fatal -- %s\n", p);
free(p);
ExitSysinstall();
}
void
AskAbort(char *fmt, ...)
{
char *p;
va_list ap;
p = Malloc(2048);
va_start(ap,fmt);
vsnprintf(p, 2048, fmt, ap);
va_end(ap);
strcat(p, "\n\nDo you wish to abort the installation?");
if (!dialog_yesno("Abort", p, -1, -1)) {
dialog_clear_norefresh();
Abort();
}
dialog_clear();
free(p);
}
void
Abort()
{
if (dialog_yesno("Exit sysinstall","\n\nAre you sure you want to quit?",
-1, -1)) {
dialog_clear();
return;
}
ExitSysinstall();
}
void
ExitSysinstall()
{
if (dialog_active) {
clear();
dialog_update();
}
if (getpid() == 1) {
if (reboot(RB_AUTOBOOT) == -1)
if (dialog_active) {
clear();
dialog_msgbox(TITLE, "\n\nCan't reboot machine -- hit reset button",
-1,-1,0);
} else
fprintf(stderr, "Can't reboot the machine -- hit the reset button");
while(1);
} else {
if (dialog_active) {
end_dialog();
dialog_active = 0;
}
exit(0);
}
}
void *
Malloc(size_t size)
{
void *p = malloc(size);
if (!p) {
exit(7); /* XXX longjmp bad */
}
return p;
}
char *
StrAlloc(char *str)
{
char *p;
p = (char *)Malloc(strlen(str) + 1);
strcpy(p,str);
return p;
}
void
MountUfs(char *device, char *mountpoint, int do_mkdir, int flags)
{
struct ufs_args ufsargs;
char dbuf[90];
memset(&ufsargs,0,sizeof ufsargs);
if(do_mkdir && access(mountpoint,R_OK)) {
Mkdir(mountpoint, TRUE);
}
strcpy(dbuf,"/dev/");
strcat(dbuf,device);
TellEm("mount %s %s",dbuf,mountpoint);
ufsargs.fspec = dbuf;
if (mount(MOUNT_UFS, mountpoint, flags, (caddr_t) &ufsargs) == -1) {
Fatal("Error mounting %s on %s : %s\n",
dbuf, mountpoint, strerror(errno));
}
}
void
Mkdir(char *ipath, int die)
{
struct stat sb;
int final=0;
char *p,*path=StrAlloc(ipath);
Debug("mkdir(%s)",path);
p = path;
if (p[0] == '/') /* Skip leading '/'. */
++p;
for (;!final; ++p) {
if (p[0] == '\0' || (p[0] == '/' && p[1] == '\0'))
final++;
else if (p[0] != '/')
continue;
*p = '\0';
if (stat(path, &sb)) {
if (errno != ENOENT && die)
Fatal("Couldn't stat directory %s: %s\n",
path,strerror(errno));
Debug("mkdir(%s..)",path);
if (mkdir(path, S_IRWXU | S_IRWXG | S_IRWXO) < 0 &&
die)
Fatal("Couldn't create directory %s: %s\n",
path,strerror(errno));
}
*p = '/';
}
free(path);
return;
}
void
Link(char *from, char *to)
{
TellEm("ln %s %s", from, to);
if(fixit)
unlink(to);
if (link(from, to) == -1)
Fatal("Couldn't create link: %s -> %s\n", from, to);
}
void
CopyFile(char *p1, char *p2)
{
char buf[BUFSIZ];
int fd1,fd2;
int i;
struct stat st;
TellEm("Copy %s to %s",p1,p2);
fd1 = open(p1,O_RDONLY);
if (fd1 < 0) Fatal("Couldn't open %s: %s\n",p1,strerror(errno));
fd2 = open(p2,O_TRUNC|O_CREAT|O_WRONLY,0200);
if (fd2 < 0) Fatal("Couldn't open %s: %s\n",p2,strerror(errno));
for(;;) {
i = read(fd1,buf,sizeof buf);
if (i > 0)
if (i != write(fd2,buf,i)) {
Fatal("Write errror on %s: %s\n",
p2,strerror(errno));
}
if (i != sizeof buf)
break;
}
fstat(fd1,&st);
fchmod(fd2,st.st_mode & 07777);
fchown(fd2,st.st_uid,st.st_gid);
close(fd1);
close(fd2);
}
u_long
PartMb(struct disklabel *lbl,int part)
{
u_long l;
l = 1024*1024/lbl->d_secsize;
return (lbl->d_partitions[part].p_size + l/2)/l;
}
void
CleanMount(int disk, int part)
{
int i = MP[disk][part];
Faction[i] = 0;
if (Fmount[i]) {
free(Fmount[i]);
Fmount[i] = 0;
}
if (Fname[i]) {
free(Fname[i]);
Fname[i] = 0;
}
if (Ftype[i]) {
free(Ftype[i]);
Ftype[i] = 0;
}
MP[disk][part] = 0;
}
char *
SetMount(int disk, int part, char *path)
{
int k;
char buf[80];
CleanMount(disk,part);
for (k = 1; k < MAX_NO_FS; k++)
if (!Fmount[k])
break;
if (k >= MAX_NO_FS)
return "Maximum number of filesystems exceeded";
Fmount[k] = StrAlloc(path);
sprintf(buf, "%s%c", Dname[disk], part + 'a');
Fname[k] = StrAlloc(buf);
switch (Dlbl[disk]->d_partitions[part].p_fstype) {
case FS_BSDFFS:
Ftype[k] = StrAlloc("ufs");
if(!fixit)
Faction[k] = 1;
break;
case FS_MSDOS:
Ftype[k] = StrAlloc("msdos");
Faction[k] = 0;
break;
case FS_SWAP:
Ftype[k] = StrAlloc("swap");
Faction[k] = 1;
break;
default:
CleanMount(disk,part);
return "Unknown filesystem-type";
}
Fsize[k] = (Dlbl[disk]->d_partitions[part].p_size+1024)/2048;
MP[disk][part] = k;
return NULL;
}
void
enable_label(int fd)
{
int flag = 1;
if (ioctl(fd, DIOCWLABEL, &flag) < 0)
Fatal("ioctl(DIOCWLABEL,1) failed: %s",strerror(errno));
}
void
disable_label(int fd)
{
int flag = 0;
if (ioctl(fd, DIOCWLABEL, &flag) < 0)
Fatal("ioctl(DIOCWLABEL,0) failed: %s",strerror(errno));
}