The-Powder-Toy/src/update.c

206 lines
3.5 KiB
C
Raw Normal View History

2010-09-27 05:53:05 -05:00
/**
* Powder Toy - Update Helper
*
* Copyright (c) 2008 - 2010 Stanislaw Skowronek.
* Copyright (c) 2010 Simon Robertshaw
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111-1301 USA
*/
#include <stdio.h>
2011-04-08 09:44:13 -05:00
#include <stdlib.h>
2010-09-27 05:53:05 -05:00
#ifndef MACOSX
#include <malloc.h>
#endif
#include <string.h>
#ifdef WIN32
#include <windows.h>
#else
#include <unistd.h>
#include <sys/stat.h>
#endif
#ifdef MACOSX
#include <mach-o/dyld.h>
#include <errno.h>
#include <sys/param.h>
#endif
#include <update.h>
char *exe_name(void)
2010-09-27 05:53:05 -05:00
{
#if defined WIN32
char *name= (char *)malloc(64);
DWORD max=64, res;
while ((res = GetModuleFileName(NULL, name, max)) >= max)
2011-01-06 16:26:31 -06:00
{
2010-09-27 05:53:05 -05:00
#elif defined MACOSX
char *fn=malloc(64),*name=malloc(PATH_MAX);
uint32_t max=64, res;
2011-01-06 16:26:31 -06:00
if (_NSGetExecutablePath(fn, &max) != 0)
{
fn = realloc(fn, max);
_NSGetExecutablePath(fn, &max);
}
if (realpath(fn, name) == NULL)
{
free(fn);
free(name);
return NULL;
}
res = 1;
2010-09-27 05:53:05 -05:00
#else
char fn[64], *name=malloc(64);
size_t max=64, res;
2011-01-06 16:26:31 -06:00
sprintf(fn, "/proc/self/exe");
memset(name, 0, max);
while ((res = readlink(fn, name, max)) >= max-1)
{
2010-09-27 05:53:05 -05:00
#endif
#ifndef MACOSX
2011-01-06 16:26:31 -06:00
max *= 2;
name = realloc(name, max);
memset(name, 0, max);
}
2010-09-27 05:53:05 -05:00
#endif
2011-01-06 16:26:31 -06:00
if (res <= 0)
{
free(name);
return NULL;
}
return name;
2010-09-27 05:53:05 -05:00
}
int update_start(char *data, int len)
{
2011-01-06 16:26:31 -06:00
char *self=exe_name(), *temp;
2010-09-27 05:53:05 -05:00
#ifdef WIN32
2011-01-06 16:26:31 -06:00
char *p;
2010-09-27 05:53:05 -05:00
#endif
2011-01-06 16:26:31 -06:00
FILE *f;
int res = 1;
2010-09-27 05:53:05 -05:00
2011-01-06 16:26:31 -06:00
if (!self)
return 1;
2010-09-27 05:53:05 -05:00
#ifdef WIN32
2011-01-06 16:26:31 -06:00
temp = malloc(strlen(self)+12);
strcpy(temp, self);
p = temp + strlen(temp) - 4;
if (_stricmp(p, ".exe"))
p += 4;
strcpy(p, "_update.exe");
if (!MoveFile(self, temp))
goto fail;
f = fopen(self, "wb");
if (!f)
goto fail;
if (fwrite(data, 1, len, f) != len)
{
fclose(f);
DeleteFile(self);
goto fail;
}
fclose(f);
if ((int)ShellExecute(NULL, "open", self, NULL, NULL, SW_SHOWNORMAL) <= 32)
{
DeleteFile(self);
goto fail;
}
return 0;
2010-09-27 05:53:05 -05:00
#else
2011-01-06 16:26:31 -06:00
temp = malloc(strlen(self)+8);
strcpy(temp, self);
strcat(temp, "-update");
f = fopen(temp, "w");
if (!f)
goto fail;
if (fwrite(data, 1, len, f) != len)
{
fclose(f);
unlink(temp);
goto fail;
}
fclose(f);
if (chmod(temp, 0755))
{
unlink(temp);
goto fail;
}
if (rename(temp, self))
{
unlink(temp);
goto fail;
}
execl(self, "powder-update", NULL);
2010-09-27 05:53:05 -05:00
#endif
fail:
2011-01-06 16:26:31 -06:00
free(temp);
free(self);
return res;
2010-09-27 05:53:05 -05:00
}
int update_finish(void)
{
#ifdef WIN32
2011-01-06 16:26:31 -06:00
char *temp, *self=exe_name(), *p;
int timeout = 60, err;
temp = malloc(strlen(self)+12);
strcpy(temp, self);
p = temp + strlen(temp) - 4;
if (_stricmp(p, ".exe"))
p += 4;
strcpy(p, "_update.exe");
while (!DeleteFile(temp))
{
err = GetLastError();
if (err == ERROR_FILE_NOT_FOUND)
{
// just as well, then
free(temp);
return 0;
}
Sleep(500);
timeout--;
if (timeout <= 0)
{
free(temp);
return 1;
}
}
free(temp);
2010-09-27 05:53:05 -05:00
#endif
2011-01-06 16:26:31 -06:00
return 0;
2010-09-27 05:53:05 -05:00
}
void update_cleanup(void)
{
#ifdef WIN32
2011-01-06 16:26:31 -06:00
update_finish();
2010-09-27 05:53:05 -05:00
#endif
}