]> arthur.barton.de Git - netatalk.git/blob - etc/afpd/mangle.c
Do some bounds checking to insure that the extension doesn't overrun
[netatalk.git] / etc / afpd / mangle.c
1 /* 
2  * $Id: mangle.c,v 1.8 2002-07-04 18:14:38 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             LOG(log_error, logtype_default, "demangle: %s is not a mangled filename", mfilename);
30             return mfilename;
31         }
32
33         if ((ext = strrchr(mfilename, '.')) != NULL) {
34             ext_len = strlen(ext);
35         }
36         if (strlen(mangle) != strlen(MANGLE_CHAR) + MANGLE_LENGTH + ext_len) {
37             LOG(log_error, logtype_default, "demangle: %s is not long enough to be a mangled filename", mfilename);
38             return mfilename;
39         }
40
41         filename = cnid_mangle_get(vol->v_db, mfilename);
42
43         /* No unmangled filename was found. */
44         if (filename == NULL) {
45             LOG(log_error, logtype_default, "demangle: Unable to lookup %s in the mangle database", mfilename);
46             return mfilename;
47         }
48
49         return filename;
50 }
51
52 char *
53 mangle(const struct vol *vol, char *filename) {
54     char *ext = NULL;
55     char *tf = NULL;
56     char *m = NULL;
57     static char mfilename[MAX_LENGTH + 1];
58     char mangle_suffix[MANGLE_LENGTH + 1];
59     int ext_len = 0;
60     int mangle_suffix_int = 0;
61
62     /* Do we really need to mangle this filename? */
63     if (strlen(filename) <= MAX_LENGTH) {
64         return filename;
65     }
66
67     /* First, attmept to locate a file extension. */
68     if ((ext = strrchr(filename, '.')) != NULL) {
69         ext_len = strlen(ext);
70         if (ext_len > MAX_EXT_LENGTH) {
71             /* Do some bounds checking to prevent an extension overflow. */
72             ext_len = MAX_EXT_LENGTH;
73         }
74     }
75
76     /* Check to see if we already have a mangled filename by this name. */
77     while (1) {
78         m = mfilename;
79         strncpy(m, filename, MAX_LENGTH - strlen(MANGLE_CHAR) - MANGLE_LENGTH - ext_len);
80         m[MAX_LENGTH - strlen(MANGLE_CHAR) - MANGLE_LENGTH - ext_len] = '\0';
81
82         strcat(m, MANGLE_CHAR);
83         (void)sprintf(mangle_suffix, "%03d", mangle_suffix_int);
84         strcat(m, mangle_suffix);
85
86         if (ext) {
87                 strncat(m, ext, ext_len);
88         }
89
90         tf = cnid_mangle_get(vol->v_db, m);
91         if (tf == NULL || (strcmp(tf, filename)) == 0) {
92             break;
93         }
94         else {
95             if (++mangle_suffix_int > MAX_MANGLE_SUFFIX_LENGTH) {
96                 LOG(log_error, logtype_default, "mangle: Failed to find a free mangle suffix; returning original filename");
97                 return filename;
98             }
99         }
100     }
101
102     if (cnid_mangle_add(vol->v_db, m, filename) < 0) {
103         return filename;
104     }
105
106     return m;
107 }
108 #endif /* FILE_MANGLING */