The installer works except for the not overwriting.
This commit is contained in:
@@ -9,31 +9,40 @@
|
|||||||
|
|
||||||
using namespace std;
|
using namespace std;
|
||||||
|
|
||||||
#include "StdString.h"
|
|
||||||
#include "InstallerFile.h"
|
|
||||||
#include "Processor.h"
|
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <sys/param.h>
|
#include <sys/param.h>
|
||||||
|
#include <sys/wait.h>
|
||||||
#include <cstdio>
|
#include <cstdio>
|
||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
#include <stack>
|
#include <stack>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
#include "StdString.h"
|
||||||
|
#include "InstallerFile.h"
|
||||||
|
#include "Processor.h"
|
||||||
|
#include "Util.h"
|
||||||
|
|
||||||
void HandleFile(const CString& file, const CString& dir, const CString& archivePath, bool overwrite)
|
void HandleFile(const CString& file, const CString& dir, const CString& archivePath, bool overwrite)
|
||||||
{
|
{
|
||||||
static bool archiveMade = false;
|
static bool archiveMade = false;
|
||||||
CString command;
|
char path[] = "/usr/bin/tar";
|
||||||
|
char arg1[] = "-rf";
|
||||||
|
char arg2[archivePath.length() + 1];
|
||||||
|
char arg3[] = "-C";
|
||||||
|
char arg4[dir.length() + 1];
|
||||||
|
char arg5[file.length() + 1];
|
||||||
|
|
||||||
if (archiveMade)
|
if (!archiveMade)
|
||||||
command = "tar rfPC '";
|
|
||||||
else
|
|
||||||
{
|
{
|
||||||
|
arg1[1] = 'c';
|
||||||
archiveMade = true;
|
archiveMade = true;
|
||||||
command = "tar cfPC '";
|
|
||||||
}
|
}
|
||||||
command += archivePath + "' '" + dir +"' '" + file + "'";
|
|
||||||
|
|
||||||
system(command);
|
strcpy(arg2, archivePath);
|
||||||
|
strcpy(arg4, dir);
|
||||||
|
strcpy(arg5, file);
|
||||||
|
|
||||||
|
if (CallTool(path, arg1, arg2, arg3, arg4, arg5, NULL))
|
||||||
|
exit(-10);
|
||||||
}
|
}
|
||||||
|
|
||||||
const CString GetPath(const CString& ID)
|
const CString GetPath(const CString& ID)
|
||||||
@@ -108,12 +117,9 @@ int main(int argc, char *argv[])
|
|||||||
return 3;
|
return 3;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Create the directory--the lazy man's way */
|
if (mkdir_p(outDir))
|
||||||
CString command = "mkdir -p '" + outDir + "'";
|
|
||||||
|
|
||||||
if (system(command))
|
|
||||||
{
|
{
|
||||||
printf("The system(\"%s\") call failed.\n", command.c_str());
|
printf("The mkdir_p call failed.\n");
|
||||||
return 4;
|
return 4;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -126,14 +132,17 @@ int main(int argc, char *argv[])
|
|||||||
|
|
||||||
printf("Compressing the archive.\n");
|
printf("Compressing the archive.\n");
|
||||||
|
|
||||||
/* Compress the archive after it is created */
|
char toolPath[] = "/usr/bin/gzip";
|
||||||
command = "gzip '" + archivePath + "'";
|
if (CallTool(toolPath, archivePath.c_str(), NULL))
|
||||||
system(command);
|
{
|
||||||
|
printf("Gzip failed.\n");
|
||||||
|
return 6;
|
||||||
|
}
|
||||||
|
|
||||||
if (!config.WriteFile(outDir + "/config"))
|
if (!config.WriteFile(outDir + "/config"))
|
||||||
{
|
{
|
||||||
printf("%s\n", strerror(errno));
|
printf("%s\n", strerror(errno));
|
||||||
return 6;
|
return 7;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
|
|||||||
@@ -6,23 +6,70 @@
|
|||||||
// Copyright (c) 2003 Steve Checkoway. All rights reserved.
|
// Copyright (c) 2003 Steve Checkoway. All rights reserved.
|
||||||
//
|
//
|
||||||
|
|
||||||
|
#import <Security/Authorization.h>
|
||||||
|
#import <Security/AuthorizationTags.h>
|
||||||
|
#import <sys/types.h>
|
||||||
|
#import <sys/wait.h>
|
||||||
#import <string.h>
|
#import <string.h>
|
||||||
|
#import <unistd.h>
|
||||||
|
#import <fcntl.h>
|
||||||
#import "Helper.h"
|
#import "Helper.h"
|
||||||
#import "InstallerWindowController.h"
|
#import "InstallerWindowController.h"
|
||||||
#include "InstallerFile.h"
|
#import "InstallerFile.h"
|
||||||
#include "Processor.h"
|
#import "Processor.h"
|
||||||
|
#import "Util.h"
|
||||||
|
|
||||||
void Error(const char *fmt, ...);
|
void Error(const char *fmt, ...);
|
||||||
|
|
||||||
static id c;
|
static id c;
|
||||||
|
static AuthorizationRef auth;
|
||||||
|
static BOOL privilegedInstall;
|
||||||
|
|
||||||
void HandleFile(const CString& file, const CString& dir, const CString& archivePath, bool overwrite)
|
void HandleFile(const CString& file, const CString& dir, const CString& archivePath, bool overwrite)
|
||||||
{
|
{
|
||||||
[c postMessage:[NSString stringWithCString:file]];
|
NSString *filePath = [NSString stringWithCString:dir];
|
||||||
|
|
||||||
CString command = "tar zxfPC '" + archivePath + "' '" + dir + "' '" + file + "'";
|
filePath = [filePath stringByAppendingFormat:@"%s%s", (dir == "/" ? "" : "/"), file.c_str()];
|
||||||
if (system(command))
|
[c postMessage:filePath];
|
||||||
Error(strerror(errno));
|
/* This is rediculus */
|
||||||
|
char f[file.length() + 1];
|
||||||
|
char d[dir.length() + 1];
|
||||||
|
char p[archivePath.length() + 1];
|
||||||
|
char path[] = "/usr/bin/tar";
|
||||||
|
char arg1[] = "-xPf";
|
||||||
|
char arg3[] = "-C";
|
||||||
|
char *const arguments[] = {path, arg1, p, arg3, d, f, NULL};
|
||||||
|
|
||||||
|
strcpy(p, archivePath);
|
||||||
|
strcpy(d, dir);
|
||||||
|
strcpy(f, file);
|
||||||
|
|
||||||
|
if (privilegedInstall)
|
||||||
|
{
|
||||||
|
AuthorizationFlags flags = kAuthorizationFlagDefaults | kAuthorizationFlagInteractionAllowed |
|
||||||
|
kAuthorizationFlagExtendRights;
|
||||||
|
AuthorizationItem item = {kAuthorizationRightExecute, 0, NULL, 0};
|
||||||
|
AuthorizationRights rights = {1, &item};
|
||||||
|
OSStatus status = AuthorizationCopyRights(auth, &rights, NULL, flags, NULL);
|
||||||
|
|
||||||
|
if (status != errAuthorizationSuccess)
|
||||||
|
{
|
||||||
|
[c postMessage:@"Couldn't authorize"];
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
status = AuthorizationExecuteWithPrivileges(auth, path, kAuthorizationFlagDefaults, arguments+1, NULL);
|
||||||
|
if (status != errAuthorizationSuccess)
|
||||||
|
[c postMessage:@"failed"];
|
||||||
|
wait(&status);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
int fd_dev_null = open("/dev/null", O_RDWR, 0);
|
||||||
|
|
||||||
|
if (CallTool3(true, -1, fd_dev_null, fd_dev_null, path, arguments))
|
||||||
|
[c postMessage:@"failed"];
|
||||||
|
close(fd_dev_null);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const CString GetPath(const CString& ID)
|
const CString GetPath(const CString& ID)
|
||||||
@@ -44,7 +91,7 @@ const CString GetPath(const CString& ID)
|
|||||||
|
|
||||||
NSString *path = [panel filename]; //No need for NSOpenPanel's filenames method
|
NSString *path = [panel filename]; //No need for NSOpenPanel's filenames method
|
||||||
|
|
||||||
[c postMessage:[NSString stringWithFormat:@"Choose file \"%@\".", path]];
|
[c postMessage:[NSString stringWithFormat:@"Chose file \"%@\".", path]];
|
||||||
|
|
||||||
return [path cString];
|
return [path cString];
|
||||||
}
|
}
|
||||||
@@ -68,16 +115,58 @@ bool AskFunc(const CString& question)
|
|||||||
return answer;
|
return answer;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool Auth(void)
|
||||||
|
{
|
||||||
|
AuthorizationFlags flags = kAuthorizationFlagDefaults | kAuthorizationFlagInteractionAllowed |
|
||||||
|
kAuthorizationFlagPreAuthorize | kAuthorizationFlagExtendRights;
|
||||||
|
AuthorizationItem item = {kAuthorizationRightExecute, 0, NULL, 0};
|
||||||
|
AuthorizationRights rights = {1, &item};
|
||||||
|
OSStatus status = AuthorizationCopyRights(auth, &rights, NULL, flags, NULL);
|
||||||
|
return status == errAuthorizationSuccess;
|
||||||
|
}
|
||||||
|
|
||||||
|
void Echo(const CString& message, bool loud)
|
||||||
|
{
|
||||||
|
if (!loud)
|
||||||
|
{
|
||||||
|
[c postMessage:[NSString stringWithCString:message.c_str()]];
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
NSBeginCriticalAlertSheet(@"Warning", @"OK", nil, nil, [c window], c, nil, nil, NULL,
|
||||||
|
[NSString stringWithCString:message.c_str()]);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Priv(bool privileged)
|
||||||
|
{
|
||||||
|
privilegedInstall = privileged;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
@implementation Helper
|
@implementation Helper
|
||||||
- (id)initWithPath:(NSString *)path
|
- (id)initWithPath:(NSString *)path
|
||||||
{
|
{
|
||||||
[super init];
|
[super init];
|
||||||
mPath = [path retain];
|
mPath = [path retain];
|
||||||
|
|
||||||
|
OSStatus status = AuthorizationCreate(NULL, kAuthorizationEmptyEnvironment,
|
||||||
|
kAuthorizationFlagDefaults, &auth);
|
||||||
|
if (status != errAuthorizationSuccess)
|
||||||
|
{
|
||||||
|
auth = NULL;
|
||||||
|
[self autorelease];
|
||||||
|
return nil;
|
||||||
|
}
|
||||||
|
|
||||||
|
privilegedInstall = NO;
|
||||||
|
|
||||||
return self;
|
return self;
|
||||||
}
|
}
|
||||||
|
|
||||||
- (void)dealloc
|
- (void)dealloc
|
||||||
{
|
{
|
||||||
|
if (auth)
|
||||||
|
AuthorizationFree(auth, kAuthorizationFlagDestroyRights);
|
||||||
[mPath autorelease];
|
[mPath autorelease];
|
||||||
[super dealloc];
|
[super dealloc];
|
||||||
}
|
}
|
||||||
@@ -88,26 +177,39 @@ bool AskFunc(const CString& question)
|
|||||||
NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
|
NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
|
||||||
CString configPath = [mPath cString];
|
CString configPath = [mPath cString];
|
||||||
CString archivePath = configPath + "/archive.tar.gz";
|
CString archivePath = configPath + "/archive.tar.gz";
|
||||||
|
BOOL success = NO;
|
||||||
|
|
||||||
configPath += "/config";
|
configPath += "/config";
|
||||||
InstallerFile file(configPath);
|
InstallerFile file(configPath);
|
||||||
Processor p(archivePath, HandleFile, GetPath, AskFunc, YES);
|
try
|
||||||
|
|
||||||
p.SetErrorFunc(Error);
|
|
||||||
|
|
||||||
[c postMessage:@"Reading from the config file."];
|
|
||||||
|
|
||||||
if(!file.ReadFile())
|
|
||||||
{
|
{
|
||||||
[c postMessage:@"Couldn't read the config file"];
|
do {
|
||||||
[c finishedInstalling:NO];
|
Processor p(archivePath, HandleFile, GetPath, AskFunc, YES);
|
||||||
[pool release];
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
unsigned nextLine = 0;
|
p.SetErrorFunc(Error);
|
||||||
while (nextLine < file.GetNumLines())
|
p.SetAuthFunc(Auth);
|
||||||
p.ProcessLine(file.GetLine(nextLine), nextLine);
|
p.SetEchoFunc(Echo);
|
||||||
|
p.SetPrivFunc(Priv);
|
||||||
|
|
||||||
|
[c postMessage:@"Reading from the config file."];
|
||||||
|
|
||||||
|
if(!file.ReadFile())
|
||||||
|
{
|
||||||
|
[c postMessage:@"Couldn't read the config file"];
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned nextLine = 0;
|
||||||
|
while (nextLine < file.GetNumLines())
|
||||||
|
p.ProcessLine(file.GetLine(nextLine), nextLine);
|
||||||
|
success = YES;
|
||||||
|
}
|
||||||
|
while(0);
|
||||||
|
}
|
||||||
|
catch (CString& e)
|
||||||
|
{
|
||||||
|
[c postMessage:[NSString stringWithCString:e]];
|
||||||
|
}
|
||||||
|
|
||||||
[c postMessage:@"Installation complete"];
|
[c postMessage:@"Installation complete"];
|
||||||
[c finishedInstalling:YES];
|
[c finishedInstalling:YES];
|
||||||
|
|||||||
@@ -3,7 +3,9 @@
|
|||||||
@implementation InstallerWindowController
|
@implementation InstallerWindowController
|
||||||
- (void)awakeFromNib
|
- (void)awakeFromNib
|
||||||
{
|
{
|
||||||
helper = [[Helper alloc] initWithPath:@"files"];
|
NSString *path = [[[[[NSBundle mainBundle] bundlePath] stringByDeletingLastPathComponent]
|
||||||
|
stringByAppendingPathComponent:@"files"] stringByStandardizingPath];
|
||||||
|
helper = [[Helper alloc] initWithPath:path];
|
||||||
finished = NO;
|
finished = NO;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -9,27 +9,15 @@
|
|||||||
|
|
||||||
using namespace std;
|
using namespace std;
|
||||||
|
|
||||||
#include <vector>
|
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <sys/stat.h>
|
#include <sys/errno.h>
|
||||||
#include <sys/param.h>
|
#include <sys/param.h>
|
||||||
|
#include <sys/wait.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <fcntl.h>
|
||||||
#include "StdString.h"
|
#include "StdString.h"
|
||||||
#include "Processor.h"
|
#include "Processor.h"
|
||||||
|
#include "Util.h"
|
||||||
void split(const CString& Source, const CString& Deliminator, vector<CString>& AddIt)
|
|
||||||
{
|
|
||||||
unsigned startpos = 0;
|
|
||||||
|
|
||||||
do {
|
|
||||||
unsigned pos = Source.find(Deliminator, startpos);
|
|
||||||
if (pos == Source.npos) pos=Source.size();
|
|
||||||
|
|
||||||
CString AddCString = Source.substr(startpos, pos-startpos);
|
|
||||||
AddIt.push_back(AddCString);
|
|
||||||
|
|
||||||
startpos=pos+Deliminator.size();
|
|
||||||
} while (startpos <= Source.size());
|
|
||||||
}
|
|
||||||
|
|
||||||
bool IsAVar(const CString& var)
|
bool IsAVar(const CString& var)
|
||||||
{
|
{
|
||||||
@@ -44,7 +32,7 @@ CString StripVar(const CString& var)
|
|||||||
}
|
}
|
||||||
|
|
||||||
Processor::Processor(CString& path, handleFileFunc f, getPathFunc p, askFunc a, bool i)
|
Processor::Processor(CString& path, handleFileFunc f, getPathFunc p, askFunc a, bool i)
|
||||||
: mDoGoto(false), mPath(path), mHandleFile(f), mGetPath(p), mAsk(a),
|
: mDoGoto(false), mHandleFile(f), mGetPath(p), mAsk(a),
|
||||||
mError(Processor::DefaultError), mInstalling(i)
|
mError(Processor::DefaultError), mInstalling(i)
|
||||||
{
|
{
|
||||||
char cwd[MAXPATHLEN];
|
char cwd[MAXPATHLEN];
|
||||||
@@ -52,6 +40,33 @@ mError(Processor::DefaultError), mInstalling(i)
|
|||||||
getwd(cwd);
|
getwd(cwd);
|
||||||
mCWD = cwd;
|
mCWD = cwd;
|
||||||
mConditionals["INSTALLING"] = i;
|
mConditionals["INSTALLING"] = i;
|
||||||
|
|
||||||
|
if (mInstalling)
|
||||||
|
{
|
||||||
|
char archivePath[] = "/tmp/installerXXXXXX.tar";
|
||||||
|
int fd = mkstemps(archivePath, 4);
|
||||||
|
|
||||||
|
if (fd == -1)
|
||||||
|
throw CString("Couldn't create temporary file");
|
||||||
|
|
||||||
|
mPath = archivePath;
|
||||||
|
|
||||||
|
char toolPath[] = "/usr/bin/gunzip";
|
||||||
|
|
||||||
|
if(CallTool2(true, -1, fd, -1, toolPath, "-c", path.c_str(), NULL))
|
||||||
|
{
|
||||||
|
unlink(mPath);
|
||||||
|
throw CString("Archive file not handled correctly or not present.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
mPath = path;
|
||||||
|
}
|
||||||
|
|
||||||
|
Processor::~Processor()
|
||||||
|
{
|
||||||
|
if (mInstalling)
|
||||||
|
unlink(mPath);
|
||||||
}
|
}
|
||||||
|
|
||||||
void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
||||||
@@ -60,6 +75,7 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
|
|
||||||
split(line, ":", parts);
|
split(line, ":", parts);
|
||||||
|
|
||||||
|
#pragma mark LABEL
|
||||||
if (parts[0] == "LABEL")
|
if (parts[0] == "LABEL")
|
||||||
{
|
{
|
||||||
printf("%u: %s\n", nextLine, line.c_str());
|
printf("%u: %s\n", nextLine, line.c_str());
|
||||||
@@ -82,17 +98,50 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
|
|
||||||
printf("%u: %s\n", nextLine, line.c_str());
|
printf("%u: %s\n", nextLine, line.c_str());
|
||||||
|
|
||||||
|
#pragma mark FILE
|
||||||
if (parts[0] == "FILE")
|
if (parts[0] == "FILE")
|
||||||
{
|
{
|
||||||
if (parts.size() != 3)
|
if (parts.size() != 3)
|
||||||
goto error;
|
goto error;
|
||||||
bool overwrite = ResolveConditional(parts[2]);
|
CString file, dir;
|
||||||
(*mHandleFile)(parts[1], mCWD, mPath, overwrite);
|
if (parts[1][0] == '/')
|
||||||
|
{
|
||||||
|
file = parts[1].substr(1);
|
||||||
|
dir = "/";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
file = parts[1];
|
||||||
|
dir = mCWD;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (mInstalling)
|
||||||
|
(*mHandleFile)(file, dir, mPath, ResolveConditional(parts[2]));
|
||||||
|
else
|
||||||
|
{
|
||||||
|
CStringArray list, dirs;
|
||||||
|
if (IsADirectory((dir == "/" ? dir : dir + "/" ) + file))
|
||||||
|
{
|
||||||
|
int fd = open(".", O_RDONLY, 0);
|
||||||
|
|
||||||
|
chdir(dir);
|
||||||
|
FileListingForDirectoryWithIgnore(file, list, dirs, mIgnore);
|
||||||
|
fchdir(fd);
|
||||||
|
close(fd);
|
||||||
|
if (list.size() == 0) //empty directory or ignored directory
|
||||||
|
(*mHandleFile)(file, dir, mPath, true);
|
||||||
|
for (unsigned i=0; i<list.size(); ++i)
|
||||||
|
(*mHandleFile)(list[i], dir, mPath, true);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
(*mHandleFile)(file, dir, mPath, true);
|
||||||
|
}
|
||||||
|
|
||||||
++nextLine;
|
++nextLine;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark GETPATH
|
||||||
if (parts[0] == "GETPATH")
|
if (parts[0] == "GETPATH")
|
||||||
{
|
{
|
||||||
if (parts.size() != 3)
|
if (parts.size() != 3)
|
||||||
@@ -102,6 +151,7 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark CD
|
||||||
if (parts[0] == "CD")
|
if (parts[0] == "CD")
|
||||||
{
|
{
|
||||||
if (parts.size() != 3)
|
if (parts.size() != 3)
|
||||||
@@ -114,20 +164,26 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
mCWD = path;
|
mCWD = path;
|
||||||
else
|
else
|
||||||
mCWD += "/" + path;
|
mCWD += "/" + path;
|
||||||
struct stat st;
|
if (!IsADirectory(mCWD))
|
||||||
if (stat(mCWD, &st))
|
|
||||||
(*mError)("%s\n", strerror(errno));
|
|
||||||
if (!(st.st_mode & S_IFDIR))
|
|
||||||
(*mError)("%s is not a directory.\n", mCWD.c_str());
|
(*mError)("%s is not a directory.\n", mCWD.c_str());
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!mInstalling)
|
if (!mInstalling)
|
||||||
{
|
{
|
||||||
|
#pragma mark IGNORE
|
||||||
|
if (parts[0] == "IGNORE")
|
||||||
|
{
|
||||||
|
if (parts.size() != 2)
|
||||||
|
goto error;
|
||||||
|
mIgnore.insert(ResolveVar(parts[1]));
|
||||||
|
}
|
||||||
|
|
||||||
++nextLine;
|
++nextLine;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark GOTO
|
||||||
if (parts[0] == "GOTO")
|
if (parts[0] == "GOTO")
|
||||||
{
|
{
|
||||||
if (parts.size() != 2)
|
if (parts.size() != 2)
|
||||||
@@ -147,6 +203,7 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark IF
|
||||||
if (parts[0] == "IF")
|
if (parts[0] == "IF")
|
||||||
{
|
{
|
||||||
if (parts.size() != 4)
|
if (parts.size() != 4)
|
||||||
@@ -172,6 +229,7 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark RETURN
|
||||||
if (parts[0] == "RETURN")
|
if (parts[0] == "RETURN")
|
||||||
{
|
{
|
||||||
if (mReturnStack.empty())
|
if (mReturnStack.empty())
|
||||||
@@ -184,6 +242,7 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark SETVAR
|
||||||
if (parts[0] == "SETVAR")
|
if (parts[0] == "SETVAR")
|
||||||
{
|
{
|
||||||
if (parts.size() < 3)
|
if (parts.size() < 3)
|
||||||
@@ -196,38 +255,93 @@ void Processor::ProcessLine(const CString& line, unsigned& nextLine)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark MKDIR
|
||||||
if (parts[0] == "MKDIR")
|
if (parts[0] == "MKDIR")
|
||||||
{
|
|
||||||
if (parts.size() != 2)
|
|
||||||
goto error;
|
|
||||||
CString path = mCWD + "/" + ResolveVar(parts[1]);
|
|
||||||
if (mkdir(path, 0777))
|
|
||||||
{
|
|
||||||
if (errno != EEXIST)
|
|
||||||
(*mError)("%s\n", strerror(errno));
|
|
||||||
}
|
|
||||||
++nextLine;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (parts[0] == "RM")
|
|
||||||
{
|
{
|
||||||
if (parts.size() != 2)
|
if (parts.size() != 2)
|
||||||
goto error;
|
goto error;
|
||||||
CString path = ResolveVar(parts[1]);
|
CString path = ResolveVar(parts[1]);
|
||||||
if (path[0] != '/')
|
if (path[0] != '/')
|
||||||
path = mCWD + "/" + path;
|
path = mCWD + "/" + path;
|
||||||
CString command = "rm -rf '" + path + "'";
|
if (mkdir_p(path))
|
||||||
system(command); // lazy
|
(*mError)(strerror(errno));
|
||||||
++nextLine;
|
++nextLine;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#pragma mark RM
|
||||||
|
if (parts[0] == "RM")
|
||||||
|
{
|
||||||
|
if (parts.size() != 2)
|
||||||
|
goto error;
|
||||||
|
CString path = ResolveVar(parts[1]);
|
||||||
|
|
||||||
|
if (path[0] != '/')
|
||||||
|
path = mCWD + "/" + path;
|
||||||
|
|
||||||
|
if (IsADirectory(path))
|
||||||
|
{
|
||||||
|
CStringArray files, dirs;
|
||||||
|
FileListingForDirectoryWithIgnore(path, files, dirs, set<CString>(), false);
|
||||||
|
for (unsigned i=0; i<files.size(); ++i)
|
||||||
|
{
|
||||||
|
if (unlink(files[i]))
|
||||||
|
(*mError)("%s", strerror(errno));
|
||||||
|
}
|
||||||
|
|
||||||
|
while (!dirs.empty())
|
||||||
|
{
|
||||||
|
if (rmdir(dirs.back()))
|
||||||
|
(*mError)("%s", strerror(errno));
|
||||||
|
dirs.pop_back();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (unlink(path))
|
||||||
|
(*mError)("%s", strerror(errno));
|
||||||
|
}
|
||||||
|
|
||||||
|
++nextLine;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
#pragma mark ASK
|
||||||
if (parts[0] == "ASK")
|
if (parts[0] == "ASK")
|
||||||
{
|
{
|
||||||
if (parts.size() != 3)
|
if (parts.size() != 3)
|
||||||
goto error;
|
goto error;
|
||||||
mVars[parts[1]] = (*mAsk)(parts[2]);
|
mConditionals[parts[1]] = (*mAsk)(parts[2]);
|
||||||
|
++nextLine;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
#pragma mark AUTH
|
||||||
|
if (parts[0] == "AUTH")
|
||||||
|
{
|
||||||
|
if (parts.size() != 2)
|
||||||
|
goto error;
|
||||||
|
mConditionals[parts[1]] = (*mAuth)();
|
||||||
|
++nextLine;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
#pragma mark ECHO
|
||||||
|
if (parts[0] == "ECHO")
|
||||||
|
{
|
||||||
|
if (parts.size() != 3)
|
||||||
|
goto error;
|
||||||
|
(mEcho)(parts[1], ResolveConditional(parts[2]));
|
||||||
|
++nextLine;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
#pragma mark PRIVILEGED
|
||||||
|
if (parts[0] == "PRIVILEGED")
|
||||||
|
{
|
||||||
|
if (parts.size() != 2)
|
||||||
|
goto error;
|
||||||
|
(mPriv)(ResolveConditional(parts[1]));
|
||||||
++nextLine;
|
++nextLine;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -13,6 +13,7 @@
|
|||||||
using namespace std;
|
using namespace std;
|
||||||
#include <stack>
|
#include <stack>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
#include <set>
|
||||||
#include "StdString.h"
|
#include "StdString.h"
|
||||||
|
|
||||||
class Processor
|
class Processor
|
||||||
@@ -23,6 +24,9 @@ private:
|
|||||||
typedef const CString (*getPathFunc)(const CString& ID);
|
typedef const CString (*getPathFunc)(const CString& ID);
|
||||||
typedef void (*errorFunc)(const char *fmt, ...);
|
typedef void (*errorFunc)(const char *fmt, ...);
|
||||||
typedef bool (*askFunc)(const CString& question);
|
typedef bool (*askFunc)(const CString& question);
|
||||||
|
typedef bool (*authFunc)(void);
|
||||||
|
typedef void (*echoFunc)(const CString& message, bool loud);
|
||||||
|
typedef void (*privFunc)(bool privileged);
|
||||||
|
|
||||||
stack<unsigned> mReturnStack;
|
stack<unsigned> mReturnStack;
|
||||||
bool mDoGoto;
|
bool mDoGoto;
|
||||||
@@ -30,12 +34,16 @@ private:
|
|||||||
map<CString, bool> mConditionals;
|
map<CString, bool> mConditionals;
|
||||||
map<CString, CString> mVars;
|
map<CString, CString> mVars;
|
||||||
map<CString, unsigned> mLabels;
|
map<CString, unsigned> mLabels;
|
||||||
|
set<CString> mIgnore;
|
||||||
CString mPath;
|
CString mPath;
|
||||||
CString mCWD;
|
CString mCWD;
|
||||||
handleFileFunc mHandleFile;
|
handleFileFunc mHandleFile;
|
||||||
getPathFunc mGetPath;
|
getPathFunc mGetPath;
|
||||||
askFunc mAsk;
|
askFunc mAsk;
|
||||||
errorFunc mError;
|
errorFunc mError;
|
||||||
|
authFunc mAuth;
|
||||||
|
echoFunc mEcho;
|
||||||
|
privFunc mPriv;
|
||||||
bool mInstalling;
|
bool mInstalling;
|
||||||
|
|
||||||
const CString& ResolveVar(const CString& var);
|
const CString& ResolveVar(const CString& var);
|
||||||
@@ -43,8 +51,12 @@ private:
|
|||||||
static void DefaultError(const char *fmt, ...);
|
static void DefaultError(const char *fmt, ...);
|
||||||
public:
|
public:
|
||||||
Processor(CString& path, handleFileFunc f, getPathFunc p, askFunc a, bool i);
|
Processor(CString& path, handleFileFunc f, getPathFunc p, askFunc a, bool i);
|
||||||
|
~Processor();
|
||||||
void ProcessLine(const CString& line, unsigned& nextLine);
|
void ProcessLine(const CString& line, unsigned& nextLine);
|
||||||
void SetErrorFunc(errorFunc f) { mError = f; }
|
void SetErrorFunc(errorFunc f) { mError = f; }
|
||||||
|
void SetAuthFunc(authFunc f) { mAuth = f; }
|
||||||
|
void SetEchoFunc(echoFunc f) { mEcho = f; }
|
||||||
|
void SetPrivFunc(privFunc f) { mPriv = f; }
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -0,0 +1,149 @@
|
|||||||
|
/*
|
||||||
|
* Util.cpp
|
||||||
|
* stepmania
|
||||||
|
*
|
||||||
|
* Created by Steve Checkoway on Mon Sep 15 2003.
|
||||||
|
* Copyright (c) 2003 Steve Checkoway. All rights reserved.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <sys/errno.h>
|
||||||
|
#include <sys/wait.h>
|
||||||
|
#include <dirent.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include "Util.h"
|
||||||
|
|
||||||
|
void split(const CString& Source, const CString& Deliminator, vector<CString>& AddIt)
|
||||||
|
{
|
||||||
|
unsigned startpos = 0;
|
||||||
|
|
||||||
|
do {
|
||||||
|
unsigned pos = Source.find(Deliminator, startpos);
|
||||||
|
if (pos == Source.npos) pos=Source.size();
|
||||||
|
|
||||||
|
CString AddCString = Source.substr(startpos, pos-startpos);
|
||||||
|
AddIt.push_back(AddCString);
|
||||||
|
|
||||||
|
startpos=pos+Deliminator.size();
|
||||||
|
} while (startpos <= Source.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
inline bool IsADirectory(const CString& path)
|
||||||
|
{
|
||||||
|
struct stat s;
|
||||||
|
if (stat(path, &s))
|
||||||
|
return false;
|
||||||
|
return s.st_mode & S_IFDIR;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline CString LastPathComponent(const CString& path)
|
||||||
|
{
|
||||||
|
unsigned pos = path.rfind('/');
|
||||||
|
|
||||||
|
if (pos == path.npos)
|
||||||
|
pos = 0;
|
||||||
|
return path.substr(pos);
|
||||||
|
}
|
||||||
|
|
||||||
|
void FileListingForDirectoryWithIgnore(const CString& path, CStringArray& list, CStringArray& dirs,
|
||||||
|
const set<CString>& ignore, bool ignoreDot)
|
||||||
|
{
|
||||||
|
DIR *dir = opendir(path);
|
||||||
|
dirent *entry;
|
||||||
|
|
||||||
|
if (!dir)
|
||||||
|
return;
|
||||||
|
|
||||||
|
while ((entry = readdir(dir)))
|
||||||
|
{
|
||||||
|
if (ignoreDot && entry->d_name[0] == '.')
|
||||||
|
continue;
|
||||||
|
if (!strcmp(entry->d_name, ".") || !strcmp(entry->d_name, ".."))
|
||||||
|
continue;
|
||||||
|
if (ignore.find(LastPathComponent(path)) != ignore.end())
|
||||||
|
continue;
|
||||||
|
CString temp = path + "/";
|
||||||
|
|
||||||
|
temp += entry->d_name;
|
||||||
|
if (IsADirectory(temp))
|
||||||
|
{
|
||||||
|
dirs.push_back(temp);
|
||||||
|
FileListingForDirectoryWithIgnore(temp, list, dirs, ignore, ignoreDot);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
list.push_back(temp);
|
||||||
|
}
|
||||||
|
closedir(dir);
|
||||||
|
}
|
||||||
|
|
||||||
|
int mkdir_p(const CString& path)
|
||||||
|
{
|
||||||
|
CStringArray components;
|
||||||
|
int ret = 0;
|
||||||
|
int fd = open(".", O_RDONLY, 0);
|
||||||
|
|
||||||
|
split(path, "/", components);
|
||||||
|
|
||||||
|
if (path[0] == '/')
|
||||||
|
chdir("/");
|
||||||
|
for (unsigned i=0; i<components.size(); ++i)
|
||||||
|
{
|
||||||
|
if (components[i] == "")
|
||||||
|
continue;
|
||||||
|
if ((ret = mkdir(components[i], 0777)))
|
||||||
|
{
|
||||||
|
if (errno != EEXIST)
|
||||||
|
goto err;
|
||||||
|
else
|
||||||
|
ret = 0;
|
||||||
|
}
|
||||||
|
chdir(components[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
err:
|
||||||
|
if (ret)
|
||||||
|
fprintf(stderr, "%s\n", strerror(errno));
|
||||||
|
fchdir(fd);
|
||||||
|
close(fd);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int CallTool3(bool blocking, int fd_in, int fd_out, int fd_err, const char *path, char *const *arguments)
|
||||||
|
{
|
||||||
|
pid_t pid = fork();
|
||||||
|
|
||||||
|
switch (pid)
|
||||||
|
{
|
||||||
|
case -1: /* error */
|
||||||
|
_exit(-1);
|
||||||
|
case 0: /* child */
|
||||||
|
if (fd_in > -1)
|
||||||
|
dup2(fd_in, 0);
|
||||||
|
|
||||||
|
if (fd_out > -1)
|
||||||
|
dup2(fd_out, 1);
|
||||||
|
|
||||||
|
if (fd_err > -1)
|
||||||
|
dup2(fd_err, 2);
|
||||||
|
|
||||||
|
if (fd_err != fd_in && fd_err != fd_out)
|
||||||
|
close(fd_err);
|
||||||
|
|
||||||
|
if (fd_out != fd_in)
|
||||||
|
close(fd_out);
|
||||||
|
|
||||||
|
close(fd_in);
|
||||||
|
execv(path, arguments);
|
||||||
|
_exit(-1);
|
||||||
|
}
|
||||||
|
/* parent */
|
||||||
|
if (!blocking)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
int status;
|
||||||
|
|
||||||
|
waitpid(pid, &status, 0);
|
||||||
|
return WEXITSTATUS(status);
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
/*
|
||||||
|
* Util.h
|
||||||
|
* stepmania
|
||||||
|
*
|
||||||
|
* Created by Steve Checkoway on Mon Sep 15 2003.
|
||||||
|
* Copyright (c) 2003 Steve Checkoway. All rights reserved.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
#include "StdString.h"
|
||||||
|
#include <vector>
|
||||||
|
#include <set>
|
||||||
|
|
||||||
|
void split(const CString& Source, const CString& Deliminator, vector<CString>& AddIt);
|
||||||
|
bool IsADirectory(const CString& path);
|
||||||
|
CString LastPathComponent(const CString& path);
|
||||||
|
void FileListingForDirectoryWithIgnore(const CString& path, CStringArray& list, CStringArray& dirs,
|
||||||
|
const set<CString>& ignore, bool ignoreDot = true);
|
||||||
|
int mkdir_p(const CString& path);
|
||||||
|
int CallTool3(bool blocking, int fd_in, int fd_out, int fd_err, const char *path, char *const *arguments);
|
||||||
|
inline int CallTool(char *path, ...) { return CallTool3(true, -1, -1, -1, path, &path); }
|
||||||
|
inline int CallTool2(bool blocking, int fd_in, int fd_out, int fd_err, char *path, ...)
|
||||||
|
{ return CallTool3(blocking, fd_in, fd_out, fd_err, path, &path); }
|
||||||
Reference in New Issue
Block a user