fix-indent-bug-with-lock-macros-part-two-20040818
[openafs.git] / src / vol / partition.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  *
9  * Portions Copyright (c) 2003 Apple Computer, Inc.
10  */
11
12 /*
13
14         System:         VICE-TWO
15         Module:         partition.c
16         Institution:    The Information Technology Center, Carnegie-Mellon University
17
18  */
19
20 #include <afsconfig.h>
21 #include <afs/param.h>
22
23 RCSID
24     ("$Header$");
25
26 #include <ctype.h>
27 #ifdef AFS_NT40_ENV
28 #include <windows.h>
29 #include <winbase.h>
30 #include <winioctl.h>
31 #else
32 #include <sys/param.h>
33 #include <sys/types.h>
34
35 #if AFS_HAVE_STATVFS || AFS_HAVE_STATVFS64
36 #include <sys/statvfs.h>
37 #endif /* AFS_HAVE_STATVFS */
38 #if defined(AFS_DARWIN_ENV) || defined(AFS_XBSD_ENV)
39 #include <sys/mount.h>
40 #endif
41
42 #if !defined(AFS_SGI_ENV)
43 #ifdef  AFS_OSF_ENV
44 #include <sys/mount.h>
45 #include <ufs/fs.h>
46 #else /* AFS_OSF_ENV */
47 #ifdef AFS_VFSINCL_ENV
48 #define VFS
49 #ifdef  AFS_SUN5_ENV
50 #include <sys/fs/ufs_fs.h>
51 #else
52 #if defined(AFS_DARWIN_ENV) || defined(AFS_XBSD_ENV)
53 #include <ufs/ufs/dinode.h>
54 #include <ufs/ffs/fs.h>
55 #else
56 #include <ufs/fs.h>
57 #endif
58 #endif
59 #else /* AFS_VFSINCL_ENV */
60 #if !defined(AFS_AIX_ENV) && !defined(AFS_LINUX22_ENV) && !defined(AFS_DARWIN_ENV) && !defined(AFS_XBSD_ENV)
61 #include <sys/fs.h>
62 #endif
63 #endif /* AFS_VFSINCL_ENV */
64 #endif /* AFS_OSF_ENV */
65 #include <errno.h>
66 #include <sys/stat.h>
67 #include <stdio.h>
68 #include <sys/file.h>
69 #ifdef  AFS_AIX_ENV
70 #include <sys/vfs.h>
71 #include <sys/lockf.h>
72 #else
73 #ifdef  AFS_HPUX_ENV
74 #include <sys/vfs.h>
75 #include <unistd.h>
76 #include <fcntl.h>
77 #include <checklist.h>
78 #else
79 #if     defined(AFS_SUN_ENV)
80 #include <sys/vfs.h>
81 #ifndef AFS_SUN5_ENV
82 #include <mntent.h>
83 #endif
84 #endif
85 #ifdef AFS_SUN5_ENV
86 #include <unistd.h>
87 #include <sys/mnttab.h>
88 #include <sys/mntent.h>
89 #else
90 #ifdef AFS_LINUX22_ENV
91 #include <mntent.h>
92 #include <sys/statfs.h>
93 #else
94 #include <fstab.h>
95 #endif
96 #endif
97 #endif
98 #endif
99 #endif /* AFS_SGI_ENV */
100 #endif /* AFS_NT40_ENV */
101 #if defined(AFS_SGI_ENV)
102 #include <sys/errno.h>
103 #include <sys/stat.h>
104 #include <stdio.h>
105 #include <sys/file.h>
106 #include <mntent.h>
107 #endif
108
109 #ifdef HAVE_STRING_H
110 #include <string.h>
111 #else
112 #ifdef HAVE_STRINGS_H
113 #include <strings.h>
114 #endif
115 #endif
116
117 #include <rx/xdr.h>
118 #include <afs/afsint.h>
119 #include "nfs.h"
120 #include <afs/errors.h>
121 #include "lock.h"
122 #include "lwp.h"
123 #include <afs/afssyscalls.h>
124 #include "ihandle.h"
125 #ifdef AFS_NAMEI_ENV
126 #ifdef AFS_NT40_ENV
127 #include "ntops.h"
128 #else
129 #include "namei_ops.h"
130 #include <dirent.h>
131 #endif /* AFS_NT40_ENV */
132 #endif /* AFS_NAMEI_ENV */
133 #include "vnode.h"
134 #include "volume.h"
135 #include "partition.h"
136 #ifdef AFS_PTHREAD_ENV
137 #include <assert.h>
138 #else /* AFS_PTHREAD_ENV */
139 #include <afs/assert.h>
140 #endif /* AFS_PTHREAD_ENV */
141
142 #if defined(AFS_HPUX_ENV)
143 #include <sys/types.h>
144 #include <sys/privgrp.h>
145 #endif /* defined(AFS_HPUX_ENV) */
146
147 #ifdef AFS_AIX42_ENV
148 #include <jfs/filsys.h>
149 #endif
150
151 #ifdef O_LARGEFILE
152
153 #define afs_stat        stat64
154 #define afs_open        open64
155 #define afs_fopen       fopen64
156 #ifndef AFS_NT40_ENV
157 #if AFS_HAVE_STATVFS64
158 # define afs_statvfs    statvfs64
159 #else
160 # if AFS_HAVE_STATFS64
161 #  define afs_statfs    statfs64
162 #else
163 #  if AFS_HAVE_STATVFS
164 #   define afs_statvfs  statvfs
165 #  else
166 #   define afs_statfs   statfs
167 #  endif /* !AFS_HAVE_STATVFS */
168 # endif /* !AFS_HAVE_STATFS64 */
169 #endif /* !AFS_HAVE_STATVFS64 */
170 #endif /* !AFS_NT40_ENV */
171
172 #else /* !O_LARGEFILE */
173
174 #define afs_stat        stat
175 #define afs_open        open
176 #define afs_fopen       fopen
177 #ifndef AFS_NT40_ENV
178 #if AFS_HAVE_STATVFS
179 #define afs_statvfs     statvfs
180 #else /* !AFS_HAVE_STATVFS */
181 #define afs_statfs      statfs
182 #endif /* !AFS_HAVE_STATVFS */
183 #endif /* !AFS_NT40_ENV */
184
185 #endif /* !O_LARGEFILE */
186
187 /*@printflike@*/ extern void Log(const char *format, ...);
188
189 int aixlow_water = 8;           /* default 8% */
190 struct DiskPartition *DiskPartitionList;
191
192 #ifdef AFS_SGI_XFS_IOPS_ENV
193 /* Verify that the on disk XFS inodes on the partition are large enough to
194  * hold the AFS attribute. Returns -1 if the attribute can't be set or is
195  * too small to fit in the inode. Returns 0 if the attribute does fit in
196  * the XFS inode.
197  */
198 #include <afs/xfsattrs.h>
199 static int
200 VerifyXFSInodeSize(char *part, char *fstype)
201 {
202     afs_xfs_attr_t junk;
203     int length = SIZEOF_XFS_ATTR_T;
204     int fd = 0;
205     int code = -1;
206     struct fsxattr fsx;
207
208     if (strcmp("xfs", fstype))
209         return 0;
210
211     if (attr_set(part, AFS_XFS_ATTR, &junk, length, ATTR_ROOT) == 0) {
212         if (((fd = open(part, O_RDONLY, 0)) != -1)
213             && (fcntl(fd, F_FSGETXATTRA, &fsx) == 0)) {
214
215             if (fsx.fsx_nextents) {
216                 Log("Partition %s: XFS inodes too small, exiting.\n", part);
217                 Log("Run xfs_size_check utility and remake partitions.\n");
218             } else
219                 code = 0;
220         }
221
222         if (fd > 0)
223             close(fd);
224         (void)attr_remove(part, AFS_XFS_ATTR, ATTR_ROOT);
225     }
226     return code;
227 }
228 #endif
229
230
231 static void
232 VInitPartition_r(char *path, char *devname, Device dev)
233 {
234     struct DiskPartition *dp, *op;
235     dp = (struct DiskPartition *)malloc(sizeof(struct DiskPartition));
236     /* Add it to the end, to preserve order when we print statistics */
237     for (op = DiskPartitionList; op; op = op->next) {
238         if (!op->next)
239             break;
240     }
241     if (op)
242         op->next = dp;
243     else
244         DiskPartitionList = dp;
245     dp->next = 0;
246     dp->name = (char *)malloc(strlen(path) + 1);
247     strncpy(dp->name, path, strlen(path) + 1);
248 #if defined(AFS_NAMEI_ENV) && !defined(AFS_NT40_ENV)
249     /* Create a lockfile for the partition, of the form /vicepa/Lock/vicepa */
250     dp->devName = (char *)malloc(2 * strlen(path) + 6);
251     strcpy(dp->devName, path);
252     strcat(dp->devName, "/");
253     strcat(dp->devName, "Lock");
254     mkdir(dp->devName, 0700);
255     strcat(dp->devName, path);
256     close(afs_open(dp->devName, O_RDWR | O_CREAT, 0600));
257     dp->device = volutil_GetPartitionID(path);
258 #else
259     dp->devName = (char *)malloc(strlen(devname) + 1);
260     strncpy(dp->devName, devname, strlen(devname) + 1);
261     dp->device = dev;
262 #endif
263     dp->lock_fd = -1;
264     dp->flags = 0;
265     dp->f_files = 1;            /* just a default value */
266 #if defined(AFS_NAMEI_ENV) && !defined(AFS_NT40_ENV)
267     if (programType == fileServer)
268         (void)namei_ViceREADME(VPartitionPath(dp));
269 #endif
270     VSetPartitionDiskUsage_r(dp);
271 }
272
273 static void
274 VInitPartition(char *path, char *devname, Device dev)
275 {
276     VOL_LOCK;
277     VInitPartition_r(path, devname, dev);
278     VOL_UNLOCK;
279 }
280
281 #ifndef AFS_NT40_ENV
282 /* VAttachPartitions() finds the vice partitions on this server. Calls
283  * VCheckPartition() to do some basic checks on the partition. If the partition
284  * is a valid vice partition, VCheckPartition will add it to the DiskPartition
285  * list.
286  * Returns the number of errors returned by VCheckPartition. An error in
287  * VCheckPartition means that partition is a valid vice partition but the
288  * fileserver should not start because of the error found on that partition.
289  *
290  * AFS_NAMEI_ENV
291  * No specific user space file system checks, since we don't know what
292  * is being used for vice partitions.
293  *
294  * Use partition name as devname.
295  */
296 int
297 VCheckPartition(char *part, char *devname)
298 {
299     struct afs_stat status;
300 #if !defined(AFS_LINUX20_ENV) && !defined(AFS_NT40_ENV)
301     char AFSIDatPath[MAXPATHLEN];
302 #endif
303
304     /* Only keep track of "/vicepx" partitions since it can get hairy
305      * when NFS mounts are involved.. */
306     if (strncmp(part, VICE_PARTITION_PREFIX, VICE_PREFIX_SIZE)) {
307         return 0;
308     }
309     if (afs_stat(part, &status) < 0) {
310         Log("VInitVnodes: Couldn't find file system %s; ignored\n", part);
311         return 0;
312     }
313 #ifndef AFS_AIX32_ENV
314     if (programType == fileServer) {
315         char salvpath[MAXPATHLEN];
316         strcpy(salvpath, part);
317         strcat(salvpath, "/FORCESALVAGE");
318         if (afs_stat(salvpath, &status) == 0) {
319             Log("VInitVnodes: Found %s; aborting\n", salvpath);
320             return -1;
321         }
322     }
323 #endif
324
325 #if !defined(AFS_LINUX20_ENV) && !defined(AFS_NT40_ENV)
326     strcpy(AFSIDatPath, part);
327     strcat(AFSIDatPath, "/AFSIDat");
328 #ifdef AFS_NAMEI_ENV
329     if (afs_stat(AFSIDatPath, &status) < 0) {
330         DIR *dirp;
331         struct dirent *dp;
332
333         dirp = opendir(part);
334         assert(dirp);
335         while (dp = readdir(dirp)) {
336             if (dp->d_name[0] == 'V') {
337                 Log("This program is compiled with AFS_NAMEI_ENV, but partition %s seems to contain volumes which don't use the namei-interface; aborting\n", part);
338                 closedir(dirp);
339                 return -1;
340             }
341         }
342         closedir(dirp);
343     }
344 #else /* AFS_NAMEI_ENV */
345     if (afs_stat(AFSIDatPath, &status) == 0) {
346         Log("This program is compiled without AFS_NAMEI_ENV, but partition %s seems to contain volumes which use the namei-interface; aborting\n", part);
347         return -1;
348     }
349 #endif /* AFS_NAMEI_ENV */
350 #endif
351
352 #ifdef AFS_SGI_XFS_IOPS_ENV
353     if (VerifyXFSInodeSize(part, status.st_fstype) < 0)
354         return -1;
355 #endif
356
357 #if defined(AFS_DUX40_ENV) && !defined(AFS_NAMEI_ENV)
358     if (status.st_ino != ROOTINO) {
359         Log("%s is not a mounted file system; ignored.\n", part);
360         return 0;
361     }
362 #endif
363
364     VInitPartition(part, devname, status.st_dev);
365
366     return 0;
367 }
368
369 /* VIsAlwaysAttach() checks whether a /vicepX directory should always be
370  * attached (return value 1), or only attached when it is a separately
371  * mounted partition (return value 0).  For non-NAMEI environments, it
372  * always returns 0.
373  */
374 static int
375 VIsAlwaysAttach(char *part)
376 {
377 #ifdef AFS_NAMEI_ENV
378     struct afs_stat st;
379     char checkfile[256];
380     int ret;
381
382     if (strncmp(part, VICE_PARTITION_PREFIX, VICE_PREFIX_SIZE))
383         return 0;
384
385     strncpy(checkfile, part, 100);
386     strcat(checkfile, "/");
387     strcat(checkfile, VICE_ALWAYSATTACH_FILE);
388
389     ret = afs_stat(checkfile, &st);
390     return (ret < 0) ? 0 : 1;
391 #else /* AFS_NAMEI_ENV */
392     return 0;
393 #endif /* AFS_NAMEI_ENV */
394 }
395
396 /* VAttachPartitions2() looks for and attaches /vicepX partitions
397  * where a special file (VICE_ALWAYSATTACH_FILE) exists.  This is
398  * used to attach /vicepX directories which aren't on dedicated
399  * partitions, in the NAMEI fileserver.
400  */
401 void
402 VAttachPartitions2()
403 {
404 #ifdef AFS_NAMEI_ENV
405     DIR *dirp;
406     struct dirent *de;
407     char pname[32];
408
409     dirp = opendir("/");
410     while ((de = readdir(dirp))) {
411         strcpy(pname, "/");
412         strncat(pname, de->d_name, 20);
413         pname[sizeof(pname) - 1] = '\0';
414
415         /* Only keep track of "/vicepx" partitions since automounter
416          * may hose us */
417         if (VIsAlwaysAttach(pname))
418             VCheckPartition(pname, "");
419     }
420     closedir(dirp);
421 #endif /* AFS_NAMEI_ENV */
422 }
423 #endif /* AFS_NT40_ENV */
424
425 #ifdef AFS_SUN5_ENV
426 int
427 VAttachPartitions(void)
428 {
429     int errors = 0;
430     struct mnttab mnt;
431     FILE *mntfile;
432
433     if (!(mntfile = afs_fopen(MNTTAB, "r"))) {
434         Log("Can't open %s\n", MNTTAB);
435         perror(MNTTAB);
436         exit(-1);
437     }
438     while (!getmntent(mntfile, &mnt)) {
439         /* Ignore non ufs or non read/write partitions */
440         if ((strcmp(mnt.mnt_fstype, "ufs") != 0)
441             || (strncmp(mnt.mnt_mntopts, "ro,ignore", 9) == 0))
442             continue;
443
444         /* If we're going to always attach this partition, do it later. */
445         if (VIsAlwaysAttach(mnt.mnt_mountp))
446             continue;
447
448         if (VCheckPartition(mnt.mnt_mountp, mnt.mnt_special) < 0)
449             errors++;
450     }
451
452     (void)fclose(mntfile);
453
454     /* Process the always-attach partitions, if any. */
455     VAttachPartitions2();
456
457     return errors;
458 }
459
460 #endif /* AFS_SUN5_ENV */
461 #if defined(AFS_SGI_ENV) || (defined(AFS_SUN_ENV) && !defined(AFS_SUN5_ENV)) || defined(AFS_HPUX_ENV)
462 int
463 VAttachPartitions(void)
464 {
465     int errors = 0;
466     FILE *mfd;
467     struct mntent *mntent;
468
469     if ((mfd = setmntent(MOUNTED, "r")) == NULL) {
470         Log("Problems in getting mount entries(setmntent)\n");
471         exit(-1);
472     }
473     while (mntent = getmntent(mfd)) {
474         if (!hasmntopt(mntent, MNTOPT_RW))
475             continue;
476
477         /* If we're going to always attach this partition, do it later. */
478         if (VIsAlwaysAttach(mntent->mnt_dir))
479             continue;
480
481         if (VCheckPartition(mntent->mnt_dir, mntent->mnt_fsname) < 0)
482             errors++;
483     }
484
485     endmntent(mfd);
486
487     /* Process the always-attach partitions, if any. */
488     VAttachPartitions2();
489
490     return errors;
491 }
492 #endif
493 #ifdef AFS_AIX_ENV
494 /*
495  * (This function was grabbed from df.c)
496  */
497 int
498 getmount(register struct vmount **vmountpp)
499 {
500     int size;
501     register struct vmount *vm;
502     int nmounts;
503
504     /* set initial size of mntctl buffer to a MAGIC NUMBER */
505     size = BUFSIZ;
506
507     /* try the operation until ok or a fatal error */
508     while (1) {
509         if ((vm = (struct vmount *)malloc(size)) == NULL) {
510             /* failed getting memory for mount status buf */
511             perror("FATAL ERROR: get_stat malloc failed\n");
512             exit(-1);
513         }
514
515         /*
516          * perform the QUERY mntctl - if it returns > 0, that is the
517          * number of vmount structures in the buffer.  If it returns
518          * -1, an error occured.  If it returned 0, then look in
519          * first word of buffer for needed size.
520          */
521         if ((nmounts = mntctl(MCTL_QUERY, size, (caddr_t) vm)) > 0) {
522             /* OK, got it, now return */
523             *vmountpp = vm;
524             return (nmounts);
525
526         } else if (nmounts == 0) {
527             /* the buffer wasn't big enough .... */
528             /* .... get required buffer size */
529             size = *(int *)vm;
530             free(vm);
531
532         } else {
533             /* some other kind of error occurred */
534             free(vm);
535             return (-1);
536         }
537     }
538 }
539
540 int
541 VAttachPartitions(void)
542 {
543     int errors = 0;
544     int nmounts;
545     struct vmount *vmountp;
546
547     if ((nmounts = getmount(&vmountp)) <= 0) {
548         Log("Problems in getting # of mount entries(getmount)\n");
549         exit(-1);
550     }
551     for (; nmounts;
552          nmounts--, vmountp =
553          (struct vmount *)((int)vmountp + vmountp->vmt_length)) {
554         char *part = vmt2dataptr(vmountp, VMT_STUB);
555
556         if (vmountp->vmt_flags & (MNT_READONLY | MNT_REMOVABLE | MNT_REMOTE))
557             continue;           /* Ignore any "special" partitions */
558
559 #ifdef AFS_AIX42_ENV
560 #ifndef AFS_NAMEI_ENV
561         {
562             struct superblock fs;
563             /* The Log statements are non-sequiters in the SalvageLog and don't
564              * even appear in the VolserLog, so restrict them to the FileLog.
565              */
566             if (ReadSuper(&fs, vmt2dataptr(vmountp, VMT_OBJECT)) < 0) {
567                 if (programType == fileServer)
568                     Log("Can't read superblock for %s, ignoring it.\n", part);
569                 continue;
570             }
571             if (IsBigFilesFileSystem(&fs)) {
572                 if (programType == fileServer)
573                     Log("%s is a big files filesystem, ignoring it.\n", part);
574                 continue;
575             }
576         }
577 #endif
578 #endif
579
580         /* If we're going to always attach this partition, do it later. */
581         if (VIsAlwaysAttach(part))
582             continue;
583
584         if (VCheckPartition(part, vmt2dataptr(vmountp, VMT_OBJECT)) < 0)
585             errors++;
586     }
587
588     /* Process the always-attach partitions, if any. */
589     VAttachPartitions2();
590
591     return errors;
592 }
593 #endif
594 #if defined(AFS_DUX40_ENV) || defined(AFS_DARWIN_ENV) || defined(AFS_XBSD_ENV)
595 int
596 VAttachPartitions(void)
597 {
598     int errors = 0;
599     struct fstab *fsent;
600
601     if (setfsent() < 0) {
602         Log("Error listing filesystems.\n");
603         exit(-1);
604     }
605
606     while (fsent = getfsent()) {
607         if (strcmp(fsent->fs_type, "rw") != 0)
608             continue;
609
610         /* If we're going to always attach this partition, do it later. */
611         if (VIsAlwaysAttach(fsent->fs_file))
612             continue;
613
614         if (VCheckPartition(fsent->fs_file, fsent->fs_spec) < 0)
615             errors++;
616     }
617     endfsent();
618
619     /* Process the always-attach partitions, if any. */
620     VAttachPartitions2();
621
622     return errors;
623 }
624 #endif
625
626 #ifdef AFS_NT40_ENV
627 #include <string.h>
628 #include <sys/stat.h>
629 /* VValidVPTEntry
630  *
631  * validate names in vptab.
632  *
633  * Return value:
634  * 1 valid entry
635  * 0 invalid entry
636  */
637
638 int
639 VValidVPTEntry(struct vptab *vpe)
640 {
641     int len = strlen(vpe->vp_name);
642     int i;
643
644     if (len < VICE_PREFIX_SIZE + 1 || len > VICE_PREFIX_SIZE + 2)
645         return 0;
646     if (strncmp(vpe->vp_name, VICE_PARTITION_PREFIX, VICE_PREFIX_SIZE))
647         return 0;
648
649     for (i = VICE_PREFIX_SIZE; i < len; i++) {
650         if (vpe->vp_name[i] < 'a' || vpe->vp_name[i] > 'z') {
651             Log("Invalid partition name %s in registry, ignoring it.\n",
652                 vpe->vp_name);
653             return 0;
654         }
655     }
656     if (len == VICE_PREFIX_SIZE + 2) {
657         i = (int)(vpe->vp_name[VICE_PREFIX_SIZE] - 'a') * 26 +
658             (int)(vpe->vp_name[VICE_PREFIX_SIZE + 1] - 'a');
659         if (i > 255) {
660             Log("Invalid partition name %s in registry, ignoring it.\n",
661                 vpe->vp_name);
662             return 0;
663         }
664     }
665
666     len = strlen(vpe->vp_dev);
667     if (len != 2 || vpe->vp_dev[1] != ':' || vpe->vp_dev[0] < 'A'
668         || vpe->vp_dev[0] > 'Z') {
669         Log("Invalid device name %s in registry, ignoring it.\n",
670             vpe->vp_dev);
671         return 0;
672     }
673
674     return 1;
675 }
676
677 int
678 VCheckPartition(char *partName)
679 {
680     char volRoot[4];
681     char volFsType[64];
682     DWORD dwDummy;
683     int err;
684
685     /* partName is presumed to be of the form "X:" */
686     (void)sprintf(volRoot, "%c:\\", *partName);
687
688     if (!GetVolumeInformation(volRoot,  /* volume root directory */
689                               NULL,     /* volume name buffer */
690                               0,        /* volume name size */
691                               NULL,     /* volume serial number */
692                               &dwDummy, /* max component length */
693                               &dwDummy, /* file system flags */
694                               volFsType,        /* file system name */
695                               sizeof(volFsType))) {
696         err = GetLastError();
697         Log("VCheckPartition: Failed to get partition information for %s, ignoring it.\n", partName);
698         return -1;
699     }
700
701     if (strcmp(volFsType, "NTFS")) {
702         Log("VCheckPartition: Partition %s is not an NTFS partition, ignoring it.\n", partName);
703         return -1;
704     }
705
706     return 0;
707 }
708
709
710 int
711 VAttachPartitions(void)
712 {
713     struct DiskPartition *partP, *prevP, *nextP;
714     struct vpt_iter iter;
715     struct vptab entry;
716
717     if (vpt_Start(&iter) < 0) {
718         Log("No partitions to attach.\n");
719         return 0;
720     }
721
722     while (0 == vpt_NextEntry(&iter, &entry)) {
723         if (!VValidVPTEntry(&entry)) {
724             continue;
725         }
726
727         /* This test for duplicates relies on the fact that the method
728          * of storing the partition names in the NT registry means the same
729          * partition name will never appear twice in the list.
730          */
731         for (partP = DiskPartitionList; partP; partP = partP->next) {
732             if (*partP->devName == *entry.vp_dev) {
733                 Log("Same drive (%s) used for both partition %s and partition %s, ignoring both.\n", entry.vp_dev, partP->name, entry.vp_name);
734                 partP->flags = PART_DUPLICATE;
735                 break;          /* Only one entry will ever be in this list. */
736             }
737         }
738         if (partP)
739             continue;           /* found a duplicate */
740
741         if (VCheckPartition(entry.vp_dev) < 0)
742             continue;
743         /* This test allows for manually inserting the FORCESALVAGE flag
744          * and thereby invoking the salvager. scandisk obviously won't be
745          * doing this for us.
746          */
747         if (programType == fileServer) {
748             struct afs_stat status;
749             char salvpath[MAXPATHLEN];
750             strcpy(salvpath, entry.vp_dev);
751             strcat(salvpath, "\\FORCESALVAGE");
752             if (afs_stat(salvpath, &status) == 0) {
753                 Log("VAttachPartitions: Found %s; aborting\n", salvpath);
754                 exit(1);
755             }
756         }
757         VInitPartition(entry.vp_name, entry.vp_dev, *entry.vp_dev - 'A');
758     }
759     vpt_Finish(&iter);
760
761     /* Run through partition list and clear out the dupes. */
762     prevP = nextP = NULL;
763     for (partP = DiskPartitionList; partP; partP = nextP) {
764         nextP = partP->next;
765         if (partP->flags == PART_DUPLICATE) {
766             if (prevP)
767                 prevP->next = partP->next;
768             else
769                 DiskPartitionList = partP->next;
770             free(partP);
771         } else
772             prevP = partP;
773     }
774
775     return 0;
776 }
777 #endif
778
779 #ifdef AFS_LINUX22_ENV
780 int
781 VAttachPartitions(void)
782 {
783     int errors = 0;
784     FILE *mfd;
785     struct mntent *mntent;
786
787     if ((mfd = setmntent("/proc/mounts", "r")) == NULL) {
788         if ((mfd = setmntent("/etc/mtab", "r")) == NULL) {
789             Log("Problems in getting mount entries(setmntent)\n");
790             exit(-1);
791         }
792     }
793     while ((mntent = getmntent(mfd))) {
794         /* If we're going to always attach this partition, do it later. */
795         if (VIsAlwaysAttach(mntent->mnt_dir))
796             continue;
797
798         if (VCheckPartition(mntent->mnt_dir, mntent->mnt_fsname) < 0)
799             errors++;
800     }
801     endmntent(mfd);
802
803     /* Process the always-attach partitions, if any. */
804     VAttachPartitions2();
805
806     return errors;
807 }
808 #endif /* AFS_LINUX22_ENV */
809
810 /* This routine is to be called whenever the actual name of the partition
811  * is required. The canonical name is still in part->name.
812  */
813 char *
814 VPartitionPath(struct DiskPartition *part)
815 {
816 #ifdef AFS_NT40_ENV
817     return part->devName;
818 #else
819     return part->name;
820 #endif
821 }
822
823 /* get partition structure, abortp tells us if we should abort on failure */
824 struct DiskPartition *
825 VGetPartition_r(char *name, int abortp)
826 {
827     register struct DiskPartition *dp;
828     for (dp = DiskPartitionList; dp; dp = dp->next) {
829         if (strcmp(dp->name, name) == 0)
830             break;
831     }
832     if (abortp)
833         assert(dp != NULL);
834     return dp;
835 }
836
837 struct DiskPartition *
838 VGetPartition(char *name, int abortp)
839 {
840     struct DiskPartition *retVal;
841     VOL_LOCK;
842     retVal = VGetPartition_r(name, abortp);
843     VOL_UNLOCK;
844     return retVal;
845 }
846
847 #ifdef AFS_NT40_ENV
848 void
849 VSetPartitionDiskUsage_r(register struct DiskPartition *dp)
850 {
851     ULARGE_INTEGER free_user, total, free_total;
852     int ufree, tot, tfree;
853
854     if (!GetDiskFreeSpaceEx
855         (VPartitionPath(dp), &free_user, &total, &free_total)) {
856         printf("Failed to get disk space info for %s, error = %d\n", dp->name,
857                GetLastError());
858         return;
859     }
860
861     /* Convert to 1K units. */
862     ufree = (int)Int64ShraMod32(free_user.QuadPart, 10);
863     tot = (int)Int64ShraMod32(total.QuadPart, 10);
864     tfree = (int)Int64ShraMod32(free_total.QuadPart, 10);
865
866     dp->minFree = tfree - ufree;        /* only used in VPrintDiskStats_r */
867     dp->totalUsable = tot;
868     dp->free = tfree;
869 }
870
871 #else
872 void
873 VSetPartitionDiskUsage_r(register struct DiskPartition *dp)
874 {
875     int fd, totalblks, free, used, availblks, bsize, code;
876     int reserved;
877 #ifdef afs_statvfs
878     struct afs_statvfs statbuf;
879 #else
880     struct afs_statfs statbuf;
881 #endif
882
883     if (dp->flags & PART_DONTUPDATE)
884         return;
885     /* Note:  we don't bother syncing because it's only an estimate, update
886      * is syncing every 30 seconds anyway, we only have to keep the disk
887      * approximately 10% from full--you just can't get the stuff in from
888      * the net fast enough to worry */
889 #ifdef afs_statvfs
890     code = afs_statvfs(dp->name, &statbuf);
891 #else
892     code = afs_statfs(dp->name, &statbuf);
893 #endif
894     if (code < 0) {
895         Log("statfs of %s failed in VSetPartitionDiskUsage (errno = %d)\n",
896             dp->name, errno);
897         return;
898     }
899     if (statbuf.f_blocks == -1) {       /* Undefined; skip stats.. */
900         Log("statfs of %s failed in VSetPartitionDiskUsage\n", dp->name);
901         return;
902     }
903     totalblks = statbuf.f_blocks;
904     free = statbuf.f_bfree;
905     reserved = free - statbuf.f_bavail;
906 #ifdef afs_statvfs
907     bsize = statbuf.f_frsize;
908 #else
909     bsize = statbuf.f_bsize;
910 #endif
911     availblks = totalblks - reserved;
912     dp->f_files = statbuf.f_files;      /* max # of files in partition */
913
914     /* Now free and totalblks are in fragment units, but we want them in
915      * 1K units.
916      */
917     if (bsize >= 1024) {
918         free *= (bsize / 1024);
919         totalblks *= (bsize / 1024);
920         availblks *= (bsize / 1024);
921         reserved *= (bsize / 1024);
922     } else {
923         free /= (1024 / bsize);
924         totalblks /= (1024 / bsize);
925         availblks /= (1024 / bsize);
926         reserved /= (1024 / bsize);
927     }
928     /* now compute remaining figures */
929     used = totalblks - free;
930
931     dp->minFree = reserved;     /* only used in VPrintDiskStats_r */
932     dp->totalUsable = availblks;
933     dp->free = availblks - used;        /* this is exactly f_bavail */
934 }
935 #endif /* AFS_NT40_ENV */
936
937 void
938 VSetPartitionDiskUsage(register struct DiskPartition *dp)
939 {
940     VOL_LOCK;
941     VSetPartitionDiskUsage_r(dp);
942     VOL_UNLOCK;
943 }
944
945 void
946 VResetDiskUsage_r(void)
947 {
948     struct DiskPartition *dp;
949     for (dp = DiskPartitionList; dp; dp = dp->next) {
950         VSetPartitionDiskUsage_r(dp);
951 #ifndef AFS_PTHREAD_ENV
952         IOMGR_Poll();
953 #endif /* !AFS_PTHREAD_ENV */
954     }
955 }
956
957 void
958 VResetDiskUsage(void)
959 {
960     VOL_LOCK;
961     VResetDiskUsage_r();
962     VOL_UNLOCK;
963 }
964
965 void
966 VAdjustDiskUsage_r(Error * ec, Volume * vp, afs_sfsize_t blocks,
967                    afs_sfsize_t checkBlocks)
968 {
969     *ec = 0;
970     /* why blocks instead of checkBlocks in the check below?  Otherwise, any check
971      * for less than BlocksSpare would skip the error-checking path, and we
972      * could grow existing files forever, not just for another BlocksSpare
973      * blocks. */
974     if (blocks > 0) {
975 #ifdef  AFS_AIX32_ENV
976         afs_int32 rem, minavail;
977
978         if ((rem = vp->partition->free - checkBlocks) < (minavail =
979                                                          (vp->partition->
980                                                           totalUsable *
981                                                           aixlow_water) /
982                                                          100))
983 #else
984         if (vp->partition->free - checkBlocks < 0)
985 #endif
986             *ec = VDISKFULL;
987         else if (V_maxquota(vp)
988                  && V_diskused(vp) + checkBlocks > V_maxquota(vp))
989             *ec = VOVERQUOTA;
990     }
991     vp->partition->free -= blocks;
992     V_diskused(vp) += blocks;
993 }
994
995 void
996 VAdjustDiskUsage(Error * ec, Volume * vp, afs_sfsize_t blocks,
997                  afs_sfsize_t checkBlocks)
998 {
999     VOL_LOCK;
1000     VAdjustDiskUsage_r(ec, vp, blocks, checkBlocks);
1001     VOL_UNLOCK;
1002 }
1003
1004 int
1005 VDiskUsage_r(Volume * vp, afs_sfsize_t blocks)
1006 {
1007     if (blocks > 0) {
1008 #ifdef  AFS_AIX32_ENV
1009         afs_int32 rem, minavail;
1010
1011         if ((rem = vp->partition->free - blocks) < (minavail =
1012                                                     (vp->partition->
1013                                                      totalUsable *
1014                                                      aixlow_water) / 100))
1015 #else
1016         if (vp->partition->free - blocks < 0)
1017 #endif
1018             return (VDISKFULL);
1019     }
1020     vp->partition->free -= blocks;
1021     return 0;
1022 }
1023
1024 int
1025 VDiskUsage(Volume * vp, afs_sfsize_t blocks)
1026 {
1027     int retVal;
1028     VOL_LOCK;
1029     retVal = VDiskUsage_r(vp, blocks);
1030     VOL_UNLOCK;
1031     return retVal;
1032 }
1033
1034 void
1035 VPrintDiskStats_r(void)
1036 {
1037     struct DiskPartition *dp;
1038     for (dp = DiskPartitionList; dp; dp = dp->next) {
1039         Log("Partition %s: %d available 1K blocks (minfree=%d), ", dp->name,
1040             dp->totalUsable, dp->minFree);
1041         if (dp->free < 0)
1042             Log("overallocated by %d blocks\n", -dp->free);
1043         else
1044             Log("%d free blocks\n", dp->free);
1045     }
1046 }
1047
1048 void
1049 VPrintDiskStats(void)
1050 {
1051     VOL_LOCK;
1052     VPrintDiskStats_r();
1053     VOL_UNLOCK;
1054 }
1055
1056 #ifdef AFS_NT40_ENV
1057 /* Need a separate lock file on NT, since NT only has mandatory file locks. */
1058 #define LOCKFILE "LOCKFILE"
1059 void
1060 VLockPartition_r(char *name)
1061 {
1062     struct DiskPartition *dp = VGetPartition_r(name, 0);
1063     OVERLAPPED lap;
1064
1065     if (!dp)
1066         return;
1067     if (dp->lock_fd == -1) {
1068         char path[64];
1069         int rc;
1070         (void)sprintf(path, "%s\\%s", VPartitionPath(dp), LOCKFILE);
1071         dp->lock_fd =
1072             (int)CreateFile(path, GENERIC_WRITE,
1073                             FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,
1074                             CREATE_ALWAYS, FILE_ATTRIBUTE_HIDDEN, NULL);
1075         assert(dp->lock_fd != (int)INVALID_HANDLE_VALUE);
1076
1077         memset((char *)&lap, 0, sizeof(lap));
1078         rc = LockFileEx((HANDLE) dp->lock_fd, LOCKFILE_EXCLUSIVE_LOCK, 0, 1,
1079                         0, &lap);
1080         assert(rc);
1081     }
1082 }
1083
1084 void
1085 VUnlockPartition_r(char *name)
1086 {
1087     register struct DiskPartition *dp = VGetPartition_r(name, 0);
1088     OVERLAPPED lap;
1089
1090     if (!dp)
1091         return;                 /* no partition, will fail later */
1092     memset((char *)&lap, 0, sizeof(lap));
1093
1094     UnlockFileEx((HANDLE) dp->lock_fd, 0, 1, 0, &lap);
1095     CloseHandle((HANDLE) dp->lock_fd);
1096     dp->lock_fd = -1;
1097 }
1098 #else /* AFS_NT40_ENV */
1099
1100 #if defined(AFS_HPUX_ENV)
1101 #define BITS_PER_CHAR   (8)
1102 #define BITS(type)      (sizeof(type) * BITS_PER_CHAR)
1103
1104 #define LOCKRDONLY_OFFSET       ((PRIV_LOCKRDONLY - 1) / BITS(int))
1105 #endif /* defined(AFS_HPUX_ENV) */
1106
1107 void
1108 VLockPartition_r(char *name)
1109 {
1110     register struct DiskPartition *dp = VGetPartition_r(name, 0);
1111     char *partitionName;
1112     int retries, code;
1113     struct timeval pausing;
1114 #if defined(AFS_HPUX_ENV)
1115     int lockfRtn;
1116     struct privgrp_map privGrpList[PRIV_MAXGRPS];
1117     unsigned int *globalMask;
1118     int globalMaskIndex;
1119 #endif /* defined(AFS_HPUX_ENV) */
1120 #if defined(AFS_DARWIN_ENV)
1121     char lockfile[MAXPATHLEN];
1122 #endif /* defined(AFS_DARWIN_ENV) */
1123 #ifdef AFS_NAMEI_ENV
1124 #ifdef AFS_AIX42_ENV
1125     char LockFileName[MAXPATHLEN + 1];
1126
1127     sprintf((char *)&LockFileName, "%s/AFSINODE_FSLock", name);
1128     partitionName = (char *)&LockFileName;
1129 #endif
1130 #endif
1131
1132     if (!dp)
1133         return;                 /* no partition, will fail later */
1134     if (dp->lock_fd != -1)
1135         return;
1136
1137 #if    defined(AFS_SUN5_ENV) || defined(AFS_AIX41_ENV)
1138 #if !defined(AFS_AIX42_ENV) || !defined(AFS_NAMEI_ENV)
1139     partitionName = dp->devName;
1140 #endif
1141     code = O_RDWR;
1142 #elif defined(AFS_DARWIN_ENV)
1143     strlcpy((partitionName = lockfile), dp->name, sizeof(lockfile));
1144     strlcat(lockfile, "/.lock.afs", sizeof(lockfile));
1145     code = O_RDONLY | O_CREAT;
1146 #else
1147     partitionName = dp->name;
1148     code = O_RDONLY;
1149 #endif
1150
1151     for (retries = 25; retries; retries--) {
1152         dp->lock_fd = afs_open(partitionName, code);
1153         if (dp->lock_fd != -1)
1154             break;
1155         if (errno == ENOENT)
1156             code |= O_CREAT;
1157         pausing.tv_sec = 0;
1158         pausing.tv_usec = 500000;
1159         select(0, NULL, NULL, NULL, &pausing);
1160     }
1161     assert(retries != 0);
1162
1163 #if defined (AFS_HPUX_ENV)
1164
1165     assert(getprivgrp(privGrpList) == 0);
1166
1167     /*
1168      * In general, it will difficult and time-consuming ,if not impossible,
1169      * to try to find the privgroup to which this process belongs that has the
1170      * smallest membership, to minimise the security hole.  So, we use the privgrp
1171      * to which everybody belongs.
1172      */
1173     /* first, we have to find the global mask */
1174     for (globalMaskIndex = 0; globalMaskIndex < PRIV_MAXGRPS;
1175          globalMaskIndex++) {
1176         if (privGrpList[globalMaskIndex].priv_groupno == PRIV_GLOBAL) {
1177             globalMask =
1178                 &(privGrpList[globalMaskIndex].priv_mask[LOCKRDONLY_OFFSET]);
1179             break;
1180         }
1181     }
1182
1183     if (((*globalMask) & privmask(PRIV_LOCKRDONLY)) == 0) {
1184         /* allow everybody to set a lock on a read-only file descriptor */
1185         (*globalMask) |= privmask(PRIV_LOCKRDONLY);
1186         assert(setprivgrp(PRIV_GLOBAL, privGrpList[globalMaskIndex].priv_mask)
1187                == 0);
1188
1189         lockfRtn = lockf(dp->lock_fd, F_LOCK, 0);
1190
1191         /* remove the privilege granted to everybody to lock a read-only fd */
1192         (*globalMask) &= ~(privmask(PRIV_LOCKRDONLY));
1193         assert(setprivgrp(PRIV_GLOBAL, privGrpList[globalMaskIndex].priv_mask)
1194                == 0);
1195     } else {
1196         /* in this case, we should be able to do this with impunity, anyway */
1197         lockfRtn = lockf(dp->lock_fd, F_LOCK, 0);
1198     }
1199
1200     assert(lockfRtn != -1);
1201 #else
1202 #if defined(AFS_AIX_ENV) || defined(AFS_SUN5_ENV)
1203     assert(lockf(dp->lock_fd, F_LOCK, 0) != -1);
1204 #else
1205     assert(flock(dp->lock_fd, LOCK_EX) == 0);
1206 #endif /* defined(AFS_AIX_ENV) || defined(AFS_SUN5_ENV) */
1207 #endif
1208 }
1209
1210 void
1211 VUnlockPartition_r(char *name)
1212 {
1213     register struct DiskPartition *dp = VGetPartition_r(name, 0);
1214     if (!dp)
1215         return;                 /* no partition, will fail later */
1216     close(dp->lock_fd);
1217     dp->lock_fd = -1;
1218 }
1219
1220 #endif /* AFS_NT40_ENV */
1221
1222 void
1223 VLockPartition(char *name)
1224 {
1225     VOL_LOCK;
1226     VLockPartition_r(name);
1227     VOL_UNLOCK;
1228 }
1229
1230 void
1231 VUnlockPartition(char *name)
1232 {
1233     VOL_LOCK;
1234     VUnlockPartition_r(name);
1235     VOL_UNLOCK;
1236 }