First round of cleanup:
* Renamed CopyEngine to WorkerThread, since it's used for other things than just copying. * Removed the code borrowed from Tracker. * Removed the InstallerCopyLoopControl, since the CopyEngine handles notifications and cancelling by itself now. * Renamed a few message constants. * Since we only support BFS partitions as install targets, I removed the FS type from the target menu item label. However, I am going to solve this differently. I guess all partitions should be seen in the target list, just disabled if they are not a valid target, and with a reason given too. git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@30600 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Stephan Aßmus
@@ -1,591 +0,0 @@
|
||||
/*
|
||||
* Copyright 2005-2008, Jérôme DUVAL. All rights reserved.
|
||||
* Distributed under the terms of the MIT License.
|
||||
*/
|
||||
|
||||
#include "CopyEngine.h"
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#include <Alert.h>
|
||||
#include <DiskDeviceVisitor.h>
|
||||
#include <DiskDeviceTypes.h>
|
||||
#include <FindDirectory.h>
|
||||
#include <Message.h>
|
||||
#include <Messenger.h>
|
||||
#include <Path.h>
|
||||
#include <String.h>
|
||||
#include <VolumeRoster.h>
|
||||
|
||||
#include "CopyEngine2.h"
|
||||
#include "InstallerWindow.h"
|
||||
#include "FSUndoRedo.h"
|
||||
#include "FSUtils.h"
|
||||
#include "PackageViews.h"
|
||||
#include "PartitionMenuItem.h"
|
||||
|
||||
|
||||
//#define COPY_TRACE
|
||||
#ifdef COPY_TRACE
|
||||
#define CALLED() printf("CALLED %s\n",__PRETTY_FUNCTION__)
|
||||
#define ERR2(x, y...) fprintf(stderr, "CopyEngine: "x" %s\n", y, strerror(err))
|
||||
#define ERR(x) fprintf(stderr, "CopyEngine: "x" %s\n", strerror(err))
|
||||
#else
|
||||
#define CALLED()
|
||||
#define ERR(x)
|
||||
#define ERR2(x, y...)
|
||||
#endif
|
||||
|
||||
const char BOOT_PATH[] = "/boot";
|
||||
|
||||
extern void SizeAsString(off_t size, char *string);
|
||||
|
||||
class SourceVisitor : public BDiskDeviceVisitor
|
||||
{
|
||||
public:
|
||||
SourceVisitor(BMenu *menu);
|
||||
virtual bool Visit(BDiskDevice *device);
|
||||
virtual bool Visit(BPartition *partition, int32 level);
|
||||
private:
|
||||
BMenu *fMenu;
|
||||
};
|
||||
|
||||
|
||||
class TargetVisitor : public BDiskDeviceVisitor
|
||||
{
|
||||
public:
|
||||
TargetVisitor(BMenu *menu);
|
||||
virtual bool Visit(BDiskDevice *device);
|
||||
virtual bool Visit(BPartition *partition, int32 level);
|
||||
private:
|
||||
void _MakeLabel(BPartition *partition, char *label, char *menuLabel);
|
||||
BMenu *fMenu;
|
||||
};
|
||||
|
||||
|
||||
CopyEngine::CopyEngine(InstallerWindow *window)
|
||||
: BLooper("copy_engine"),
|
||||
fWindow(window),
|
||||
fPackages(NULL),
|
||||
fSpaceRequired(0)
|
||||
{
|
||||
fControl = new InstallerCopyLoopControl(window);
|
||||
Run();
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::MessageReceived(BMessage* message)
|
||||
{
|
||||
CALLED();
|
||||
|
||||
switch (message->what) {
|
||||
case ENGINE_START:
|
||||
Start(fWindow->GetSourceMenu(), fWindow->GetTargetMenu());
|
||||
break;
|
||||
|
||||
case kWriteBootSector:
|
||||
{
|
||||
int32 id;
|
||||
if (message->FindInt32("id", &id) != B_OK) {
|
||||
SetStatusMessage("Boot sector not written because of an "
|
||||
" internal error.");
|
||||
break;
|
||||
}
|
||||
|
||||
// TODO: Refactor with Start()
|
||||
BPath targetDirectory;
|
||||
BDiskDevice device;
|
||||
BPartition* partition;
|
||||
|
||||
if (fDDRoster.GetPartitionWithID(id, &device, &partition) == B_OK) {
|
||||
if (!partition->IsMounted()) {
|
||||
if (partition->Mount() < B_OK) {
|
||||
SetStatusMessage("The partition can't be mounted. "
|
||||
"Please choose a different partition.");
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (partition->GetMountPoint(&targetDirectory) != B_OK) {
|
||||
SetStatusMessage("The mount point could not be retrieve.");
|
||||
break;
|
||||
}
|
||||
} else if (fDDRoster.GetDeviceWithID(id, &device) == B_OK) {
|
||||
if (!device.IsMounted()) {
|
||||
if (device.Mount() < B_OK) {
|
||||
SetStatusMessage("The disk can't be mounted. Please "
|
||||
"choose a different disk.");
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (device.GetMountPoint(&targetDirectory) != B_OK) {
|
||||
SetStatusMessage("The mount point could not be retrieve.");
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
LaunchFinishScript(targetDirectory);
|
||||
// TODO: Get error from executing script!
|
||||
SetStatusMessage("Boot sector successfully written.");
|
||||
}
|
||||
default:
|
||||
BLooper::MessageReceived(message);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::SetStatusMessage(const char *status)
|
||||
{
|
||||
BMessage msg(STATUS_MESSAGE);
|
||||
msg.AddString("status", status);
|
||||
BMessenger(fWindow).SendMessage(&msg);
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::LaunchInitScript(BPath &path)
|
||||
{
|
||||
BPath bootPath;
|
||||
find_directory(B_BEOS_BOOT_DIRECTORY, &bootPath);
|
||||
BString command("/bin/sh ");
|
||||
command += bootPath.Path();
|
||||
command += "/InstallerInitScript ";
|
||||
command += path.Path();
|
||||
SetStatusMessage("Starting Installation.");
|
||||
system(command.String());
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::LaunchFinishScript(BPath &path)
|
||||
{
|
||||
BPath bootPath;
|
||||
find_directory(B_BEOS_BOOT_DIRECTORY, &bootPath);
|
||||
BString command("/bin/sh ");
|
||||
command += bootPath.Path();
|
||||
command += "/InstallerFinishScript ";
|
||||
command += path.Path();
|
||||
SetStatusMessage("Finishing Installation.");
|
||||
system(command.String());
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::Start(BMenu *srcMenu, BMenu *targetMenu)
|
||||
{
|
||||
CALLED();
|
||||
|
||||
BPath targetDirectory, srcDirectory;
|
||||
BDirectory targetDir, srcDir;
|
||||
BDiskDevice device;
|
||||
BPartition *partition;
|
||||
BVolume targetVolume;
|
||||
status_t err = B_OK;
|
||||
int32 entries = 0;
|
||||
entry_ref testRef;
|
||||
bigtime_t now;
|
||||
|
||||
fControl->Reset();
|
||||
|
||||
PartitionMenuItem *targetItem = (PartitionMenuItem *)targetMenu->FindMarked();
|
||||
PartitionMenuItem *srcItem = (PartitionMenuItem *)srcMenu->FindMarked();
|
||||
if (!srcItem || !targetItem) {
|
||||
ERR("bad menu items\n");
|
||||
goto error;
|
||||
}
|
||||
|
||||
// check if target is initialized
|
||||
// ask if init or mount as is
|
||||
|
||||
if (fDDRoster.GetPartitionWithID(targetItem->ID(), &device, &partition) == B_OK) {
|
||||
if (!partition->IsMounted()) {
|
||||
if ((err = partition->Mount()) < B_OK) {
|
||||
SetStatusMessage("The disk can't be mounted. Please choose a "
|
||||
"different disk.");
|
||||
ERR("BPartition::Mount");
|
||||
goto error;
|
||||
}
|
||||
}
|
||||
if ((err = partition->GetVolume(&targetVolume)) != B_OK) {
|
||||
ERR("BPartition::GetVolume");
|
||||
goto error;
|
||||
}
|
||||
if ((err = partition->GetMountPoint(&targetDirectory)) != B_OK) {
|
||||
ERR("BPartition::GetMountPoint");
|
||||
goto error;
|
||||
}
|
||||
} else if (fDDRoster.GetDeviceWithID(targetItem->ID(), &device) == B_OK) {
|
||||
if (!device.IsMounted()) {
|
||||
if ((err = device.Mount()) < B_OK) {
|
||||
SetStatusMessage("The disk can't be mounted. Please choose a "
|
||||
"different disk.");
|
||||
ERR("BDiskDevice::Mount");
|
||||
goto error;
|
||||
}
|
||||
}
|
||||
if ((err = device.GetVolume(&targetVolume)) != B_OK) {
|
||||
ERR("BDiskDevice::GetVolume");
|
||||
goto error;
|
||||
}
|
||||
if ((err = device.GetMountPoint(&targetDirectory)) != B_OK) {
|
||||
ERR("BDiskDevice::GetMountPoint");
|
||||
goto error;
|
||||
}
|
||||
} else
|
||||
goto error; // shouldn't happen
|
||||
|
||||
// check if target has enough space
|
||||
if ((fSpaceRequired > 0 && targetVolume.FreeBytes() < fSpaceRequired)
|
||||
&& ((new BAlert("", "The destination disk may not have enough space. "
|
||||
"Try choosing a different disk or choose to not install optional "
|
||||
"items.", "Try installing anyway", "Cancel", 0,
|
||||
B_WIDTH_AS_USUAL, B_STOP_ALERT))->Go() != 0)) {
|
||||
goto error;
|
||||
}
|
||||
|
||||
if (fDDRoster.GetPartitionWithID(srcItem->ID(), &device, &partition) == B_OK) {
|
||||
if ((err = partition->GetMountPoint(&srcDirectory)) != B_OK) {
|
||||
ERR("BPartition::GetMountPoint");
|
||||
goto error;
|
||||
}
|
||||
} else if (fDDRoster.GetDeviceWithID(srcItem->ID(), &device) == B_OK) {
|
||||
if ((err = device.GetMountPoint(&srcDirectory)) != B_OK) {
|
||||
ERR("BDiskDevice::GetMountPoint");
|
||||
goto error;
|
||||
}
|
||||
} else
|
||||
goto error; // shouldn't happen
|
||||
|
||||
// check not installing on itself
|
||||
if (strcmp(srcDirectory.Path(), targetDirectory.Path()) == 0) {
|
||||
SetStatusMessage("You can't install the contents of a disk onto "
|
||||
"itself. Please choose a different disk.");
|
||||
goto error;
|
||||
}
|
||||
|
||||
// check not installing on boot volume
|
||||
if ((strncmp(BOOT_PATH, targetDirectory.Path(), strlen(BOOT_PATH)) == 0)
|
||||
&& ((new BAlert("", "Are you sure you want to install onto the "
|
||||
"current boot disk? The Installer will have to reboot your "
|
||||
"machine if you proceed.", "OK", "Cancel", 0,
|
||||
B_WIDTH_AS_USUAL, B_STOP_ALERT))->Go() != 0)) {
|
||||
SetStatusMessage("Installation stopped.");
|
||||
goto error;
|
||||
}
|
||||
|
||||
targetDir.SetTo(targetDirectory.Path());
|
||||
|
||||
// check target volume not empty
|
||||
// NOTE: It's ok if exactly this path exists: home/Desktop/Trash
|
||||
// and nothing else.
|
||||
while (targetDir.GetNextRef(&testRef) == B_OK) {
|
||||
if (strcmp(testRef.name, "home") == 0) {
|
||||
BDirectory homeDir(&testRef);
|
||||
while (homeDir.GetNextRef(&testRef) == B_OK) {
|
||||
if (strcmp(testRef.name, "Desktop") == 0) {
|
||||
BDirectory desktopDir(&testRef);
|
||||
while (desktopDir.GetNextRef(&testRef) == B_OK) {
|
||||
if (strcmp(testRef.name, "Trash") == 0) {
|
||||
BDirectory trashDir(&testRef);
|
||||
while (trashDir.GetNextRef(&testRef) == B_OK) {
|
||||
// Something in the Trash
|
||||
entries++;
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
// Something besides Trash
|
||||
entries++;
|
||||
}
|
||||
|
||||
if (entries > 0)
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
// Something besides Desktop
|
||||
entries++;
|
||||
}
|
||||
|
||||
if (entries > 0)
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
// Something besides home
|
||||
entries++;
|
||||
}
|
||||
|
||||
if (entries > 0)
|
||||
break;
|
||||
}
|
||||
if (entries != 0
|
||||
&& ((new BAlert("", "The target volume is not empty. Are you sure you "
|
||||
"want to install anyways?", "Install Anyways", "Cancel", 0,
|
||||
B_WIDTH_AS_USUAL, B_STOP_ALERT))->Go() != 0)) {
|
||||
err = B_CANCELED;
|
||||
goto error;
|
||||
}
|
||||
|
||||
|
||||
LaunchInitScript(targetDirectory);
|
||||
|
||||
// copy source volume
|
||||
targetDir.Rewind();
|
||||
srcDir.SetTo(srcDirectory.Path());
|
||||
now = system_time();
|
||||
#if 0
|
||||
err = CopyFolder(srcDir, targetDir);
|
||||
#else
|
||||
{
|
||||
BMessenger messenger(fWindow);
|
||||
CopyEngine2 engine(messenger, new BMessage(STATUS_MESSAGE));
|
||||
err = engine.CopyFolder(srcDirectory.Path(), targetDirectory.Path(),
|
||||
fCancelLock);
|
||||
if (err != B_OK)
|
||||
printf("error: %s\n", strerror(err));
|
||||
}
|
||||
#endif
|
||||
printf("copy time: %.3fs\n", (system_time() - now) / 1000000.0);
|
||||
|
||||
if (err != B_OK || fControl->CheckUserCanceled())
|
||||
goto error;
|
||||
|
||||
// copy selected packages
|
||||
if (fPackages) {
|
||||
srcDirectory.Append(PACKAGES_DIRECTORY);
|
||||
srcDir.SetTo(srcDirectory.Path());
|
||||
BDirectory packageDir;
|
||||
int32 count = fPackages->CountItems();
|
||||
for (int32 i = 0; i < count; i++) {
|
||||
Package *p = static_cast<Package*>(fPackages->ItemAt(i));
|
||||
packageDir.SetTo(&srcDir, p->Folder());
|
||||
err = CopyFolder(packageDir, targetDir);
|
||||
if (err != B_OK || fControl->CheckUserCanceled())
|
||||
goto error;
|
||||
}
|
||||
}
|
||||
|
||||
LaunchFinishScript(targetDirectory);
|
||||
|
||||
BMessenger(fWindow).SendMessage(INSTALL_FINISHED);
|
||||
|
||||
return;
|
||||
error:
|
||||
if (err == B_CANCELED || fControl->CheckUserCanceled())
|
||||
SetStatusMessage("Installation canceled.");
|
||||
ERR("Start failed");
|
||||
BMessenger(fWindow).SendMessage(RESET_INSTALL);
|
||||
}
|
||||
|
||||
|
||||
status_t
|
||||
CopyEngine::CopyFolder(BDirectory &srcDir, BDirectory &targetDir)
|
||||
{
|
||||
BEntry entry;
|
||||
status_t err;
|
||||
while (srcDir.GetNextEntry(&entry) == B_OK
|
||||
&& !fControl->CheckUserCanceled()) {
|
||||
StatStruct statbuf;
|
||||
entry.GetStat(&statbuf);
|
||||
|
||||
Undo undo;
|
||||
if (S_ISDIR(statbuf.st_mode)) {
|
||||
char name[B_FILE_NAME_LENGTH];
|
||||
if (entry.GetName(name) == B_OK
|
||||
&& (strcmp(name, PACKAGES_DIRECTORY) == 0
|
||||
|| strcmp(name, VAR_DIRECTORY) == 0)) {
|
||||
continue;
|
||||
}
|
||||
err = FSCopyFolder(&entry, &targetDir, fControl, NULL, false, undo);
|
||||
} else {
|
||||
err = FSCopyFile(&entry, &statbuf, &targetDir, fControl, NULL,
|
||||
false, undo);
|
||||
}
|
||||
if (err != B_OK) {
|
||||
BPath path;
|
||||
entry.GetPath(&path);
|
||||
ERR2("error while copying %s", path.Path());
|
||||
return err;
|
||||
}
|
||||
}
|
||||
|
||||
return B_OK;
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::ScanDisksPartitions(BMenu *srcMenu, BMenu *targetMenu)
|
||||
{
|
||||
// NOTE: This is actually executed in the window thread.
|
||||
BDiskDevice device;
|
||||
BPartition *partition = NULL;
|
||||
|
||||
printf("\nScanDisksPartitions source partitions begin\n");
|
||||
SourceVisitor srcVisitor(srcMenu);
|
||||
fDDRoster.VisitEachMountedPartition(&srcVisitor, &device, &partition);
|
||||
|
||||
printf("\nScanDisksPartitions target partitions begin\n");
|
||||
TargetVisitor targetVisitor(targetMenu);
|
||||
fDDRoster.VisitEachPartition(&targetVisitor, &device, &partition);
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::SetPackagesList(BList *list)
|
||||
{
|
||||
delete fPackages;
|
||||
fPackages = list;
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
CopyEngine::Cancel()
|
||||
{
|
||||
return fControl->Cancel();
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
CopyEngine::WriteBootSector(BMenu* targetMenu)
|
||||
{
|
||||
// Executed in window thread.
|
||||
CALLED();
|
||||
|
||||
PartitionMenuItem* item = (PartitionMenuItem*)targetMenu->FindMarked();
|
||||
if (item == NULL) {
|
||||
ERR("bad menu items\n");
|
||||
return;
|
||||
}
|
||||
|
||||
BMessage message(kWriteBootSector);
|
||||
message.AddInt32("id", item->ID());
|
||||
PostMessage(&message, this);
|
||||
}
|
||||
|
||||
|
||||
// #pragma mark -
|
||||
|
||||
|
||||
SourceVisitor::SourceVisitor(BMenu *menu)
|
||||
: fMenu(menu)
|
||||
{
|
||||
}
|
||||
|
||||
bool
|
||||
SourceVisitor::Visit(BDiskDevice *device)
|
||||
{
|
||||
return Visit(device, 0);
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
SourceVisitor::Visit(BPartition *partition, int32 level)
|
||||
{
|
||||
BPath path;
|
||||
if (partition->GetPath(&path) == B_OK)
|
||||
printf("SourceVisitor::Visit(BPartition *) : %s\n", path.Path());
|
||||
printf("SourceVisitor::Visit(BPartition *) : %s\n", partition->ContentName());
|
||||
|
||||
if (partition->ContentType() == NULL)
|
||||
return false;
|
||||
|
||||
bool isBootPartition = false;
|
||||
if (partition->IsMounted()) {
|
||||
BPath mountPoint;
|
||||
partition->GetMountPoint(&mountPoint);
|
||||
isBootPartition = strcmp(BOOT_PATH, mountPoint.Path()) == 0;
|
||||
}
|
||||
|
||||
if (!isBootPartition
|
||||
&& strcmp(partition->ContentType(), kPartitionTypeBFS) != 0) {
|
||||
// Except only BFS partitions, except this is the boot partition
|
||||
// (ISO9660 with write overlay for example).
|
||||
return false;
|
||||
}
|
||||
|
||||
// TODO: We could probably check if this volume contains
|
||||
// the Haiku kernel or something. Does it make sense to "install"
|
||||
// from your BFS volume containing the music collection?
|
||||
// TODO: Then the check for BFS could also be removed above.
|
||||
|
||||
PartitionMenuItem *item = new PartitionMenuItem(NULL,
|
||||
partition->ContentName(), NULL, new BMessage(SRC_PARTITION),
|
||||
partition->ID());
|
||||
item->SetMarked(isBootPartition);
|
||||
fMenu->AddItem(item);
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
// #pragma mark -
|
||||
|
||||
|
||||
TargetVisitor::TargetVisitor(BMenu *menu)
|
||||
: fMenu(menu)
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
TargetVisitor::Visit(BDiskDevice *device)
|
||||
{
|
||||
if (device->IsReadOnlyMedia())
|
||||
return false;
|
||||
return Visit(device, 0);
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
TargetVisitor::Visit(BPartition *partition, int32 level)
|
||||
{
|
||||
BPath path;
|
||||
if (partition->GetPath(&path) == B_OK)
|
||||
printf("TargetVisitor::Visit(BPartition *) : %s\n", path.Path());
|
||||
printf("TargetVisitor::Visit(BPartition *) : %s\n", partition->ContentName());
|
||||
|
||||
if (partition->ContentType() == NULL
|
||||
|| strcmp(partition->ContentType(), kPartitionTypeBFS) != 0) {
|
||||
// Except only valid BFS partitions
|
||||
printf(" not BFS\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
if (partition->ContentSize() < 20 * 1024 * 1024) {
|
||||
// reject partitions which are too small anyways
|
||||
// TODO: Could depend on the source size
|
||||
printf(" too small\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
if (partition->CountChildren() > 0) {
|
||||
// Looks like an extended partition, or the device itself.
|
||||
// Do not accept this as target...
|
||||
printf(" no leaf partition\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
// TODO: After running DriveSetup and doing another scan, it would
|
||||
// be great to pick the partition which just appeared!
|
||||
|
||||
char label[255], menuLabel[255];
|
||||
_MakeLabel(partition, label, menuLabel);
|
||||
fMenu->AddItem(new PartitionMenuItem(partition->ContentName(), label,
|
||||
menuLabel, new BMessage(TARGET_PARTITION), partition->ID()));
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
TargetVisitor::_MakeLabel(BPartition *partition, char *label, char *menuLabel)
|
||||
{
|
||||
char size[15];
|
||||
SizeAsString(partition->ContentSize(), size);
|
||||
BPath path;
|
||||
partition->GetPath(&path);
|
||||
|
||||
sprintf(label, "%s - %s [%s] [%s]", partition->ContentName(),
|
||||
size, partition->ContentType(), path.Path());
|
||||
sprintf(menuLabel, "%s - %s [%s]", partition->ContentName(), size,
|
||||
partition->ContentType());
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user