]> arthur.barton.de Git - netatalk.git/blob - etc/afpd/mangle.c
812f9d5bdcfccb0daec0f5936c029a6048a88d69
[netatalk.git] / etc / afpd / mangle.c
1 /* 
2  * $Id: mangle.c,v 1.13 2002-12-18 00:17:03 jmarcus Exp $ 
3  *
4  * Copyright (c) 2002. Joe Marcus Clarke (marcus@marcuscom.com)
5  * All Rights Reserved.  See COPYRIGHT.
6  *
7  * mangle, demangle (filename):
8  * mangle or demangle filenames if they are greater than the max allowed
9  * characters for a given version of AFP.
10  */
11
12 #ifdef HAVE_CONFIG_H
13 #include "config.h"
14 #endif /* HAVE_CONFIG_H */
15
16 #ifdef FILE_MANGLING
17 #include "mangle.h"
18
19 char *
20 demangle(const struct vol *vol, char *mfilename) {
21         char *filename = NULL;
22         char *ext = NULL;
23         int ext_len = 0;
24         char *mangle;
25
26         /* Is this really a mangled file? */
27         mangle = strstr(mfilename, MANGLE_CHAR);
28         if (!mangle) {
29             return mfilename;
30         }
31
32         if ((ext = strrchr(mfilename, '.')) != NULL) {
33             ext_len = strlen(ext);
34         }
35         if (strlen(mangle) != strlen(MANGLE_CHAR) + MANGLE_LENGTH + ext_len) {
36             return mfilename;
37         }
38
39         filename = cnid_mangle_get(vol->v_db, mfilename);
40
41         /* No unmangled filename was found. */
42         if (filename == NULL) {
43             LOG(log_error, logtype_afpd, "demangle: Unable to lookup %s in the mangle database", mfilename);
44             return mfilename;
45         }
46
47         return filename;
48 }
49
50 char *
51 mangle(const struct vol *vol, char *filename) {
52     char *ext = NULL;
53     char *tf = NULL;
54     char *m = NULL;
55     static char mfilename[MAX_LENGTH + 1];
56     char mangle_suffix[MANGLE_LENGTH + 1];
57     int ext_len = 0;
58     int mangle_suffix_int = 0;
59
60     /* Do we really need to mangle this filename? */
61     if (strlen(filename) <= vol->max_filename) {
62         return filename;
63     }
64
65     /* First, attmept to locate a file extension. */
66     if ((ext = strrchr(filename, '.')) != NULL) {
67         ext_len = strlen(ext);
68         if (ext_len > MAX_EXT_LENGTH) {
69             /* Do some bounds checking to prevent an extension overflow. */
70             ext_len = MAX_EXT_LENGTH;
71         }
72     }
73
74     /* Check to see if we already have a mangled filename by this name. */
75     while (1) {
76         m = mfilename;
77         strncpy(m, filename, MAX_LENGTH - strlen(MANGLE_CHAR) - MANGLE_LENGTH - ext_len);
78         m[MAX_LENGTH - strlen(MANGLE_CHAR) - MANGLE_LENGTH - ext_len] = '\0';
79
80         strcat(m, MANGLE_CHAR);
81         (void)sprintf(mangle_suffix, "%03d", mangle_suffix_int);
82         strcat(m, mangle_suffix);
83
84         if (ext) {
85                 strncat(m, ext, ext_len);
86         }
87
88         tf = cnid_mangle_get(vol->v_db, m);
89         if (tf == NULL || (strcmp(tf, filename)) == 0) {
90             break;
91         }
92         else {
93             if (++mangle_suffix_int > MAX_MANGLE_SUFFIX_LENGTH) {
94                 LOG(log_error, logtype_afpd, "mangle: Failed to find a free mangle suffix; returning original filename");
95                 return filename;
96             }
97         }
98     }
99
100     if (cnid_mangle_add(vol->v_db, m, filename) < 0) {
101         return filename;
102     }
103
104     return m;
105 }
106 #endif /* FILE_MANGLING */