drivedetect.c 9.95 KB
Newer Older
Mike Hearn's avatar
Mike Hearn committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * Drive autodetection code
 *
 * Copyright 2004 Mike Hearn
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
18
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
Mike Hearn's avatar
Mike Hearn committed
19 20 21
 *
 */

22
#include "config.h"
23
#include "wine/port.h"
24

Mike Hearn's avatar
Mike Hearn committed
25 26 27 28 29 30
#include <wine/debug.h>
#include <wine/library.h>

#include "winecfg.h"

#include <stdio.h>
31
#ifdef HAVE_MNTENT_H
Mike Hearn's avatar
Mike Hearn committed
32
#include <mntent.h>
33
#endif
Mike Hearn's avatar
Mike Hearn committed
34 35 36 37 38 39 40 41 42 43 44 45
#include <stdlib.h>
#include <errno.h>

#include <sys/stat.h>

#include <winbase.h>

WINE_DEFAULT_DEBUG_CHANNEL(winecfg);

BOOL gui_mode = TRUE;
static long working_mask = 0;

46 47 48 49 50 51
typedef struct
{
  const char *szNode;
  int nType;
} DEV_NODES;

52
#ifdef HAVE_MNTENT_H
53

54
static const DEV_NODES sDeviceNodes[] = {
55
  {"/dev/fd", DRIVE_REMOVABLE},
56 57 58
  {"/dev/pf", DRIVE_REMOVABLE},
  {"/dev/aztcd", DRIVE_CDROM},
  {"/dev/bpcd", DRIVE_CDROM},
59
  {"/dev/cdrom", DRIVE_CDROM},
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
  {"/dev/cdu535", DRIVE_CDROM},
  {"/dev/cdwriter", DRIVE_CDROM},
  {"/dev/cm205cd", DRIVE_CDROM},
  {"/dev/cm206cd", DRIVE_CDROM},
  {"/dev/gscd", DRIVE_CDROM},
  {"/dev/hitcd", DRIVE_CDROM},
  {"/dev/iseries/vcd", DRIVE_CDROM},
  {"/dev/lmscd", DRIVE_CDROM},
  {"/dev/mcd", DRIVE_CDROM},
  {"/dev/optcd", DRIVE_CDROM},
  {"/dev/pcd", DRIVE_CDROM},
  {"/dev/sbpcd", DRIVE_CDROM},
  {"/dev/scd", DRIVE_CDROM},
  {"/dev/sjcd", DRIVE_CDROM},
  {"/dev/sonycd", DRIVE_CDROM},
  {"/dev/sr", DRIVE_CDROM},
76 77 78
  {"",0}
};

79
static const char * const ignored_fstypes[] = {
Mike Hearn's avatar
Mike Hearn committed
80 81 82 83 84 85 86
    "devpts",
    "tmpfs",
    "proc",
    "sysfs",
    "swap",
    "usbdevfs",
    "rpc_pipefs",
87
    "binfmt_misc",
Mike Hearn's avatar
Mike Hearn committed
88 89 90
    NULL
};

91
static const char * const ignored_mnt_dirs[] = {
92 93 94 95
    "/boot",
    NULL
};

96 97 98 99 100 101 102 103 104 105 106 107 108 109
static int try_dev_node(char *dev)
{
    const DEV_NODES *pDevNodes = sDeviceNodes;
    
    while(pDevNodes->szNode[0])
    {
        if(!strncmp(dev,pDevNodes->szNode,strlen(pDevNodes->szNode)))
            return pDevNodes->nType;
        ++pDevNodes;
    }
    
    return DRIVE_FIXED;
}

Mike Hearn's avatar
Mike Hearn committed
110 111
static BOOL should_ignore_fstype(char *type)
{
112
    const char * const *s;
Mike Hearn's avatar
Mike Hearn committed
113 114 115 116 117 118 119
    
    for (s = ignored_fstypes; *s; s++)
        if (!strcmp(*s, type)) return TRUE;

    return FALSE;
}

120 121
static BOOL should_ignore_mnt_dir(char *dir)
{
122
    const char * const *s;
123 124 125 126 127 128 129

    for (s = ignored_mnt_dirs; *s; s++)
        if (!strcmp(*s, dir)) return TRUE;

    return FALSE;
}

Mike Hearn's avatar
Mike Hearn committed
130 131 132 133 134 135 136 137 138 139 140
static BOOL is_drive_defined(char *path)
{
    int i;
    
    for (i = 0; i < 26; i++)
        if (drives[i].in_use && !strcmp(drives[i].unixpath, path)) return TRUE;

    return FALSE;
}

/* returns Z + 1 if there are no more available letters */
141
static char allocate_letter(int type)
Mike Hearn's avatar
Mike Hearn committed
142
{
143
    char letter, start;
Mike Hearn's avatar
Mike Hearn committed
144

145 146 147 148 149 150
    if (type == DRIVE_REMOVABLE)
	start = 'A';
    else
	start = 'C';

    for (letter = start; letter <= 'Z'; letter++)
Mike Hearn's avatar
Mike Hearn committed
151 152 153 154
        if ((DRIVE_MASK_BIT(letter) & working_mask) != 0) break;

    return letter;
}
155
#endif
Mike Hearn's avatar
Mike Hearn committed
156 157 158 159 160

#define FSTAB_OPEN 1
#define NO_MORE_LETTERS 2
#define NO_ROOT 3
#define NO_DRIVE_C 4
161
#define NO_HOME 5
Mike Hearn's avatar
Mike Hearn committed
162 163 164 165 166 167 168 169 170 171 172

static void report_error(int code)
{
    char *buffer;
    int len;
    
    switch (code)
    {
        case FSTAB_OPEN:
            if (gui_mode)
            {
173
                static const char s[] = "Could not open your mountpoint description table.\n\nOpening of /etc/fstab failed: %s";
Mike Hearn's avatar
Mike Hearn committed
174 175 176 177 178 179 180 181
                len = snprintf(NULL, 0, s, strerror(errno));
                buffer = HeapAlloc(GetProcessHeap(), 0, len + 1);
                snprintf(buffer, len, s, strerror(errno));
                MessageBox(NULL, s, "", MB_OK | MB_ICONEXCLAMATION);
                HeapFree(GetProcessHeap(), 0, buffer);
            }
            else
            {
182
                fprintf(stderr, "winecfg: could not open fstab: %s\n", strerror(errno));
Mike Hearn's avatar
Mike Hearn committed
183 184 185 186 187
            }
            break;

        case NO_MORE_LETTERS:
            if (gui_mode) MessageBox(NULL, "No more letters are available to auto-detect available drives with.", "", MB_OK | MB_ICONEXCLAMATION);
188
            fprintf(stderr, "winecfg: no more available letters while scanning /etc/fstab\n");
Mike Hearn's avatar
Mike Hearn committed
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
            break;

        case NO_ROOT:
            if (gui_mode) MessageBox(NULL, "Could not ensure that the root directory was mapped.\n\n"
                                     "This can happen if you run out of drive letters. "
                                     "It's important to have the root directory mapped, otherwise Wine"
                                     "will not be able to always find the programs you want to run. "
                                     "Try unmapping a drive letter then trying again.", "",
                                     MB_OK | MB_ICONEXCLAMATION);
            else fprintf(stderr, "winecfg: unable to map root drive\n");
            break;

        case NO_DRIVE_C:
            if (gui_mode)
                MessageBox(NULL, "No virtual drive C mapped\n\nTry running wineprefixcreate", "", MB_OK | MB_ICONEXCLAMATION);
            else
                fprintf(stderr, "winecfg: no drive_c directory\n");
206 207 208 209 210 211 212 213 214 215

        case NO_HOME:
            if (gui_mode)
                MessageBox(NULL, "Could not ensure that your home directory was mapped.\n\n"
                                 "This can happen if you run out of drive letters. "
                                 "Try unmapping a drive letter then try again.", "",
                                 MB_OK | MB_ICONEXCLAMATION);
            else 
                fprintf(stderr, "winecfg: unable to map home drive\n");
            break;
Mike Hearn's avatar
Mike Hearn committed
216 217 218
    }
}

219
static void ensure_root_is_mapped(void)
Mike Hearn's avatar
Mike Hearn committed
220 221 222 223 224 225 226 227 228 229 230 231 232 233
{
    int i;
    BOOL mapped = FALSE;

    for (i = 0; i < 26; i++)
        if (drives[i].in_use && !strcmp(drives[i].unixpath, "/")) mapped = TRUE;

    if (!mapped)
    {
        /* work backwards from Z, trying to map it */
        char letter;

        for (letter = 'Z'; letter >= 'A'; letter--)
        {
234 235 236 237 238 239
            if (!drives[letter - 'A'].in_use) 
            {
                add_drive(letter, "/", "System", "0", DRIVE_FIXED);
                WINE_TRACE("allocated drive %c as the root drive\n", letter);
                break;
            }
Mike Hearn's avatar
Mike Hearn committed
240 241 242 243 244 245
        }

        if (letter == ('A' - 1)) report_error(NO_ROOT);
    }
}

246
static void ensure_home_is_mapped(void)
247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273
{
    int i;
    BOOL mapped = FALSE;
    char *home = getenv("HOME");

    if (!home) return;

    for (i = 0; i < 26; i++)
        if (drives[i].in_use && !strcmp(drives[i].unixpath, home)) mapped = TRUE;

    if (!mapped)
    {
        char letter;

        for (letter = 'H'; letter <= 'Z'; letter++)
        {
            if (!drives[letter - 'A'].in_use)
            {
                add_drive(letter, home, "Home", "0", DRIVE_FIXED);
                WINE_TRACE("allocated drive %c as the user's home directory\n", letter);
                break;
            }
        }
        if (letter == ('Z' + 1)) report_error(NO_HOME);
    }        
}

274
static void ensure_drive_c_is_mapped(void)
Mike Hearn's avatar
Mike Hearn committed
275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297
{
    struct stat buf;
    const char *configdir = wine_get_config_dir();
    int len;
    char *drive_c_dir;
    
    if (drives[2].in_use) return;

    len = snprintf(NULL, 0, "%s/../drive_c", configdir);
    drive_c_dir = HeapAlloc(GetProcessHeap(), 0, len);
    snprintf(drive_c_dir, len, "%s/../drive_c", configdir);
    HeapFree(GetProcessHeap(), 0, drive_c_dir);

    if (stat(drive_c_dir, &buf) == 0)
    {
        add_drive('C', "../drive_c", "Virtual Windows Drive", "0", DRIVE_FIXED);
    }
    else
    {
        report_error(NO_DRIVE_C);
    }
}

298
int autodetect_drives(void)
Mike Hearn's avatar
Mike Hearn committed
299
{
300
#ifdef HAVE_MNTENT_H
Mike Hearn's avatar
Mike Hearn committed
301 302
    struct mntent *ent;
    FILE *fstab;
303
#endif
Mike Hearn's avatar
Mike Hearn committed
304 305 306 307 308 309 310 311 312 313 314 315

    /* we want to build a list of autodetected drives, then ensure each entry
       exists in the users setup. so, we superimpose the autodetected drives
       onto whatever is pre-existing.

       for now let's just rummage around inside the fstab.
     */

    load_drives();

    working_mask = drive_available_mask('\0');

316
#ifdef HAVE_MNTENT_H
Mike Hearn's avatar
Mike Hearn committed
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
    fstab = fopen("/etc/fstab", "r");
    if (!fstab)
    {
        report_error(FSTAB_OPEN);
        return FALSE;
    }

    while ((ent = getmntent(fstab)))
    {
        char letter;
        char label[256];
        int type;
        
        WINE_TRACE("ent->mnt_dir=%s\n", ent->mnt_dir);

        if (should_ignore_fstype(ent->mnt_type)) continue;
333
        if (should_ignore_mnt_dir(ent->mnt_dir)) continue;
Mike Hearn's avatar
Mike Hearn committed
334 335
        if (is_drive_defined(ent->mnt_dir)) continue;

336 337 338 339 340 341 342 343 344
        if (!strcmp(ent->mnt_type, "nfs")) type = DRIVE_REMOTE;
        else if (!strcmp(ent->mnt_type, "nfs4")) type = DRIVE_REMOTE;
        else if (!strcmp(ent->mnt_type, "smbfs")) type = DRIVE_REMOTE;
        else if (!strcmp(ent->mnt_type, "cifs")) type = DRIVE_REMOTE;
        else if (!strcmp(ent->mnt_type, "coda")) type = DRIVE_REMOTE;
        else if (!strcmp(ent->mnt_type, "iso9660")) type = DRIVE_CDROM;
        else if (!strcmp(ent->mnt_type, "ramfs")) type = DRIVE_RAMDISK;
        else type = try_dev_node(ent->mnt_fsname);
        
Mike Hearn's avatar
Mike Hearn committed
345
        /* allocate a drive for it */
346
        letter = allocate_letter(type);
Mike Hearn's avatar
Mike Hearn committed
347 348 349 350 351 352 353
        if (letter == ']')
        {
            report_error(NO_MORE_LETTERS);
            fclose(fstab);
            return FALSE;
        }
        
354
        strcpy(label, "Drive X");
Mike Hearn's avatar
Mike Hearn committed
355
        label[6] = letter;
356 357
        
        WINE_TRACE("adding drive %c for %s, type %s with label %s\n", letter, ent->mnt_dir, ent->mnt_type,label);
Mike Hearn's avatar
Mike Hearn committed
358 359 360

        add_drive(letter, ent->mnt_dir, label, "0", type);
        
361 362
        /* working_mask is a map of the drive letters still available. */
        working_mask &= ~DRIVE_MASK_BIT(letter);
Mike Hearn's avatar
Mike Hearn committed
363 364 365
    }

    fclose(fstab);
366
#endif
Mike Hearn's avatar
Mike Hearn committed
367 368 369 370 371

    ensure_root_is_mapped();

    ensure_drive_c_is_mapped();

372 373
    ensure_home_is_mapped();

Mike Hearn's avatar
Mike Hearn committed
374 375
    return TRUE;
}