File: NGMimeContentDispositionHeaderFieldGenerator.m

package info (click to toggle)
sope 3.2.6-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 16,408 kB
  • ctags: 2,779
  • sloc: objc: 167,855; sh: 3,646; ansic: 3,399; python: 318; makefile: 181
file content (118 lines) | stat: -rw-r--r-- 3,317 bytes parent folder | download | duplicates (7)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/*
  Copyright (C) 2000-2005 SKYRIX Software AG

  This file is part of SOPE.

  SOPE 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, or (at your option) any
  later version.

  SOPE 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 SOPE; see the file COPYING.  If not, write to the
  Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#include "NGMimeHeaderFieldGenerator.h"
#include "NGMimeHeaderFields.h"
#include "common.h"
#include <string.h>

@implementation NGMimeContentDispositionHeaderFieldGenerator

- (NSData *)generateDataForHeaderFieldNamed:(NSString *)_headerField
  value:(id)_value
{
  NGMimeContentDispositionHeaderField *field;
  NSString      *tmp;
  NSMutableData *data;
  
  field = _value;
  if (field == nil) {
    [self logWithFormat:@"WARNING(%s): Content-Disposition field is empty",
          __PRETTY_FUNCTION__];
    return [NSData data];
  }
  
  if ([_value isKindOfClass:[NSString class]])
    return [_value dataUsingEncoding:NSUTF8StringEncoding];
  
  // TODO: move the stuff below to some NSString or NSData category?
  
  data = [NSMutableData dataWithCapacity: 64];
  tmp  = [field type];
  [data appendBytes:[tmp cString] length:[tmp length]];
  tmp = [field filename];
  if (tmp != nil) {
    [data appendBytes:"; " length:2];
    [data appendBytes:"filename=\"" length:10];
    
    const char* bytes;
    unsigned length;
    int  cnt;
    BOOL doEnc;

    //d = [tmp dataUsingEncoding: NSUTF8StringEncoding];
    //bytes = [d bytes];
    //length = [d length];
    bytes = [tmp cStringUsingEncoding: NSUTF8StringEncoding];
    length = strlen(bytes);
    
    cnt = 0;
    doEnc = NO;
    while (cnt < length) {
      if ((unsigned char)bytes[cnt] > 127) {
	doEnc = YES;
	break;
      }
      cnt++;
    }

    if (doEnc)
      {
	char        iso[]     = "=?utf-8?q?";
	unsigned    isoLen    = 10;
	char        isoEnd[]  = "?=";
	unsigned    isoEndLen = 2;
	int    desLen;
	char        *des;
	
	bytes = [[tmp stringByReplacingOccurrencesOfString:@"\\\"" withString:@"\""] cStringUsingEncoding: NSUTF8StringEncoding];
	length = strlen(bytes);

	desLen = length * 3 + 20;
	
	des = calloc(desLen + 2, sizeof(char));
	
	memcpy(des, iso, isoLen);
	desLen = NGEncodeQuotedPrintableMime((unsigned char *)bytes, length,
					     (unsigned char *)(des + isoLen),
					     desLen - isoLen);
	if (desLen != -1) {
	  memcpy(des + isoLen + desLen, isoEnd, isoEndLen);
	  [data appendBytes:des length:(isoLen + desLen + isoEndLen)];
	}
	else {
          [self logWithFormat:@"WARNING(%s:%i): An error occour during "
		@"quoted-printable decoding",
                __PRETTY_FUNCTION__, __LINE__];
	  if (des != NULL) free(des);
	}
      }
    else
      {
	[data appendBytes:[tmp cString] length:[tmp length]];
      }

    [data appendBytes:"\"" length:1];
  }
  return data;
}
  
@end /* NGMimeContentDispositionHeaderFieldGenerator */