The R Project SVN R

/*
 *  R : A Computer Language for Statistical Data Analysis
 *  Copyright (C) 2001-23  R Core Team
 *
 *  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,  a copy is available at
 *  https://www.R-project.org/Licenses/
 */

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>     /* for exit */
#include <Rversion.h>

#define PRODUCER "R-core"

#if (defined(__i386) || defined(__x86_64))
# ifdef _WIN64
#  define RK "R64"
# else
#  define RK "R32"
# endif
#else
# ifdef RK
#  undef RK
# endif
#endif

extern char *getRHOMElong(int m); /* in ../rhome.c */
extern void freeRHOMElong(char *s);

void R_Suicide(char *s) /* for call from ../rhome.o */
{
    fprintf(stderr, "FATAL ERROR:%s\n", s);
    exit(2);
}

int main (int argc, char **argv)
{
    HKEY hk = HKEY_LOCAL_MACHINE;
    DWORD subkeys = 0;
  
    char *RHome, version[40], keyname[60];
    LONG rc;
    HKEY hkey, hkey2;
    int delete = 0, personal = 0;

    for(int i = 1; i < argc; i++) if(!strcmp(argv[i], "/U")) delete = 1;
    for(int i = 1; i < argc; i++) if(!strcmp(argv[i], "/Personal")) personal = 1;
    
    if(personal) hk = HKEY_CURRENT_USER;

    /* Needs to match JRins */
    if(strncmp(R_STATUS, "Under ", 6) == 0)
    snprintf(version, 40, "%s.%s Pre-release", R_MAJOR, R_MINOR);
    else
    snprintf(version, 40, "%s.%s %s", R_MAJOR, R_MINOR, R_STATUS);

    if(delete) {
    printf("unregistering R %s ... ", version);

#if defined(RK) && defined(R_ARCH)
    if (strlen(R_ARCH) > 0) {
        snprintf(keyname, 60, "Software\\%s\\%s\\%s", PRODUCER, RK, version);
        if (RegOpenKeyEx(hk, keyname, 0, KEY_SET_VALUE, &hkey) == ERROR_SUCCESS) {
        RegDeleteValue(hkey, "InstallPath");
        RegDeleteKey(hkey, version);
        RegCloseKey(hkey);
        } else
        fprintf(stderr, "\nWarning: failed to open key '%s'\n", keyname);
        
        snprintf(keyname, 60, "Software\\%s\\%s", PRODUCER, RK);
        if (RegOpenKeyEx(hk, keyname, 0, KEY_SET_VALUE, &hkey) == ERROR_SUCCESS) {
        RegDeleteValue(hkey, "InstallPath");
        RegDeleteValue(hkey, "Current Version");
        RegDeleteKey(hkey, version);
        RegQueryInfoKey(hkey, NULL, NULL, NULL, &subkeys, NULL, NULL, NULL,
                        NULL, NULL, NULL, NULL);
        RegCloseKey(hkey);
        /* delete key if empty */
        if (!subkeys && 
            RegOpenKeyEx(hk, "Software\\R-core", 0, KEY_SET_VALUE, &hkey)
            == ERROR_SUCCESS) {
            RegDeleteKey(hkey, RK);
            RegCloseKey(hkey);
        }
        } else
        fprintf(stderr, "\nWarning: failed to open key '%s'\n", keyname);
    }
#endif

    snprintf(keyname, 60, "Software\\%s\\R\\%s", PRODUCER, version);
    if (RegOpenKeyEx(hk, keyname, 0, KEY_SET_VALUE, &hkey) == ERROR_SUCCESS) {
        RegDeleteValue(hkey, "InstallPath");
        RegDeleteKey(hkey, version);
        RegCloseKey(hkey);
    } else
        fprintf(stderr, "\nWarning: failed to open key '%s'\n", keyname);

    
    snprintf(keyname, 60, "Software\\%s\\R", PRODUCER);
    if (RegOpenKeyEx(hk, keyname, 0, KEY_SET_VALUE, &hkey) == ERROR_SUCCESS) {
        RegDeleteValue(hkey, "InstallPath");
        RegDeleteValue(hkey, "Current Version");
        RegDeleteKey(hkey, version);
        RegQueryInfoKey(hkey, NULL, NULL, NULL, &subkeys, NULL, NULL, NULL, NULL, NULL,
                        NULL, NULL);
        RegCloseKey(hkey);
        /* delete key if empty */
        if (!subkeys && 
        RegOpenKeyEx(hk, "Software\\R-core", 0, KEY_SET_VALUE, &hkey)
        == ERROR_SUCCESS) {
        RegDeleteKey(hkey, "R");
        RegCloseKey(hkey);
        }

        printf("succeeded\n");
    } else {
        printf("was not registered\n");
        exit(1);
    }
    } else {
    printf("registering R %s ... ", version);
    int dirstrip = 2;
#ifdef R_ARCH
    if (strlen(R_ARCH) > 0)
        dirstrip++;
#endif 
        RHome = getRHOMElong(dirstrip);

    snprintf(keyname, 60, "Software\\%s\\R", PRODUCER);
    if ((rc = RegOpenKeyEx(hk, keyname, 0, KEY_ALL_ACCESS, &hkey))
        != ERROR_SUCCESS) {
        /* failed to open key, so try to create it */
        rc = RegCreateKey(hk, keyname, &hkey);
    }
    if(rc != ERROR_SUCCESS) {
        fprintf(stderr, "\nError: failed to open key '%s'\n", keyname);
        freeRHOMElong(RHome);
        exit(1);
    }
    
    rc = RegSetValueEx(hkey, "InstallPath", 0, REG_SZ,
               (CONST BYTE *)RHome, lstrlen(RHome)+1);
    if (rc == ERROR_SUCCESS)
        rc = RegSetValueEx(hkey, "Current Version", 0, REG_SZ,
                   (CONST BYTE *)version, lstrlen(version)+1);
    
    if (rc == ERROR_SUCCESS)
        rc = RegCreateKey(hkey, version, &hkey2);
    
    if (rc == ERROR_SUCCESS) {
        rc = RegSetValueEx(hkey2, "InstallPath", 0, REG_SZ,
                   (CONST BYTE *)RHome, lstrlen(RHome)+1);
        RegCloseKey(hkey2);
    }
    RegCloseKey(hkey);  

#if defined(RK) && defined(R_ARCH)
    if (strlen(R_ARCH) > 0) {
        snprintf(keyname, 60, "Software\\%s\\%s", PRODUCER, RK);
        if ((rc = RegOpenKeyEx(hk, keyname, 0, 
                   KEY_ALL_ACCESS, &hkey)) != ERROR_SUCCESS) {
        /* failed to open key, so try to create it */
        rc = RegCreateKey(hk, keyname, &hkey);
        }
        if(rc != ERROR_SUCCESS) {       
        fprintf(stderr, "\nError: failed to open key '%s'\n", keyname);
        freeRHOMElong(RHome);
        exit(1);
        }
        
        rc = RegSetValueEx(hkey, "InstallPath", 0, REG_SZ,
                   (CONST BYTE *)RHome, lstrlen(RHome)+1);
        rc = RegSetValueEx(hkey, "Current Version", 0, REG_SZ,
                   (CONST BYTE *)version, lstrlen(version)+1);  
        rc = RegCreateKey(hkey, version, &hkey2);
        if (rc == ERROR_SUCCESS) {
        rc = RegSetValueEx(hkey2, "InstallPath", 0, REG_SZ,
                   (CONST BYTE *)RHome, lstrlen(RHome)+1);
        RegCloseKey(hkey2);
        }
        RegCloseKey(hkey);
    }
#endif
    freeRHOMElong(RHome);

    printf("succeeded\n");
    }
    exit(0);
}