1 /* 2 * Copyright (c) 1997, 2014, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 /* 27 * @(#)MimeMultipart.java 1.31 03/01/29 28 */ 29 30 31 32 package com.sun.xml.internal.messaging.saaj.packaging.mime.internet; 33 34 import java.io.*; 35 36 import javax.activation.DataSource; 37 38 import com.sun.xml.internal.messaging.saaj.packaging.mime.*; 39 import com.sun.xml.internal.messaging.saaj.packaging.mime.util.*; 40 import com.sun.xml.internal.messaging.saaj.util.FinalArrayList; 41 import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream; 42 import com.sun.xml.internal.messaging.saaj.util.SAAJUtil; 43 44 /** 45 * The MimeMultipart class is an implementation 46 * that uses MIME conventions for the multipart data. <p> 47 * 48 * A MimeMultipart is obtained from a MimeBodyPart whose primary type 49 * is "multipart" (by invoking the part's <code>getContent()</code> method) 50 * or it can be created by a client as part of creating a new MimeMessage. <p> 51 * 52 * The default multipart subtype is "mixed". The other multipart 53 * subtypes, such as "alternative", "related", and so on, can be 54 * implemented as subclasses of MimeMultipart with additional methods 55 * to implement the additional semantics of that type of multipart 56 * content. The intent is that service providers, mail JavaBean writers 57 * and mail clients will write many such subclasses and their Command 58 * Beans, and will install them into the JavaBeans Activation 59 * Framework, so that any JavaMail implementation and its clients can 60 * transparently find and use these classes. Thus, a MIME multipart 61 * handler is treated just like any other type handler, thereby 62 * decoupling the process of providing multipart handlers from the 63 * JavaMail API. Lacking these additional MimeMultipart subclasses, 64 * all subtypes of MIME multipart data appear as MimeMultipart objects. <p> 65 * 66 * An application can directly construct a MIME multipart object of any 67 * subtype by using the <code>MimeMultipart(String subtype)</code> 68 * constructor. For example, to create a "multipart/alternative" object, 69 * use <code>new MimeMultipart("alternative")</code>. 70 * 71 * @version 1.31, 03/01/29 72 * @author John Mani 73 * @author Bill Shannon 74 * @author Max Spivak 75 */ 76 77 //BM MimeMultipart can extend this 78 public class MimeMultipart { 79 80 /** 81 * The DataSource supplying our InputStream. 82 */ 83 protected DataSource ds = null; 84 85 /** 86 * Have we parsed the data from our InputStream yet? 87 * Defaults to true; set to false when our constructor is 88 * given a DataSource with an InputStream that we need to 89 * parse. 90 */ 91 protected boolean parsed = true; 92 93 /** 94 * Vector of MimeBodyPart objects. 95 */ 96 protected FinalArrayList parts = new FinalArrayList(); // Holds BodyParts 97 98 /** 99 * This field specifies the content-type of this multipart 100 * object. It defaults to "multipart/mixed". 101 */ 102 protected ContentType contentType; 103 104 /** 105 * The <code>MimeBodyPart</code> containing this <code>MimeMultipart</code>, 106 * if known. 107 * @since JavaMail 1.1 108 */ 109 protected MimeBodyPart parent; 110 111 protected static final boolean ignoreMissingEndBoundary; 112 static { 113 ignoreMissingEndBoundary = SAAJUtil.getSystemBoolean("saaj.mime.multipart.ignoremissingendboundary"); 114 } 115 116 /** 117 * Default constructor. An empty MimeMultipart object 118 * is created. Its content type is set to "multipart/mixed". 119 * A unique boundary string is generated and this string is 120 * setup as the "boundary" parameter for the 121 * <code>contentType</code> field. <p> 122 * 123 * MimeBodyParts may be added later. 124 */ 125 public MimeMultipart() { 126 this("mixed"); 127 } 128 129 /** 130 * Construct a MimeMultipart object of the given subtype. 131 * A unique boundary string is generated and this string is 132 * setup as the "boundary" parameter for the 133 * <code>contentType</code> field. <p> 134 * 135 * MimeBodyParts may be added later. 136 */ 137 public MimeMultipart(String subtype) { 138 //super(); 139 /* 140 * Compute a boundary string. 141 */ 142 String boundary = UniqueValue.getUniqueBoundaryValue(); 143 contentType = new ContentType("multipart", subtype, null); 144 contentType.setParameter("boundary", boundary); 145 } 146 147 /** 148 * Constructs a MimeMultipart object and its bodyparts from the 149 * given DataSource. <p> 150 * 151 * This constructor handles as a special case the situation where the 152 * given DataSource is a MultipartDataSource object. 153 * 154 * Otherwise, the DataSource is assumed to provide a MIME multipart 155 * byte stream. The <code>parsed</code> flag is set to false. When 156 * the data for the body parts are needed, the parser extracts the 157 * "boundary" parameter from the content type of this DataSource, 158 * skips the 'preamble' and reads bytes till the terminating 159 * boundary and creates MimeBodyParts for each part of the stream. 160 * 161 * @param ds DataSource, can be a MultipartDataSource 162 * @param ct 163 * This must be the same information as {@link DataSource#getContentType()}. 164 * All the callers of this method seem to have this object handy, so 165 * for performance reason this method accepts it. Can be null. 166 */ 167 public MimeMultipart(DataSource ds, ContentType ct) throws MessagingException { 168 // 'ds' was not a MultipartDataSource, we have 169 // to parse this ourself. 170 parsed = false; 171 this.ds = ds; 172 if (ct==null) 173 contentType = new ContentType(ds.getContentType()); 174 else 175 contentType = ct; 176 } 177 178 /** 179 * Set the subtype. This method should be invoked only on a new 180 * MimeMultipart object created by the client. The default subtype 181 * of such a multipart object is "mixed". <p> 182 * 183 * @param subtype Subtype 184 */ 185 public void setSubType(String subtype) { 186 contentType.setSubType(subtype); 187 } 188 189 /** 190 * Return the number of enclosed MimeBodyPart objects. 191 * 192 * @return number of parts 193 */ 194 public int getCount() throws MessagingException { 195 parse(); 196 if (parts == null) 197 return 0; 198 199 return parts.size(); 200 } 201 202 /** 203 * Get the specified MimeBodyPart. BodyParts are numbered starting at 0. 204 * 205 * @param index the index of the desired MimeBodyPart 206 * @return the MimeBodyPart 207 * @exception MessagingException if no such MimeBodyPart exists 208 */ 209 public MimeBodyPart getBodyPart(int index) 210 throws MessagingException { 211 parse(); 212 if (parts == null) 213 throw new IndexOutOfBoundsException("No such BodyPart"); 214 215 return (MimeBodyPart)parts.get(index); 216 } 217 218 /** 219 * Get the MimeBodyPart referred to by the given ContentID (CID). 220 * Returns null if the part is not found. 221 * 222 * @param CID the ContentID of the desired part 223 * @return the MimeBodyPart 224 */ 225 public MimeBodyPart getBodyPart(String CID) 226 throws MessagingException { 227 parse(); 228 229 int count = getCount(); 230 for (int i = 0; i < count; i++) { 231 MimeBodyPart part = getBodyPart(i); 232 String s = part.getContentID(); 233 // Old versions of AXIS2 put angle brackets around the content 234 // id but not the start param 235 String sNoAngle = (s!= null) ? s.replaceFirst("^<", "").replaceFirst(">$", "") 236 :null; 237 if (s != null && (s.equals(CID) || CID.equals(sNoAngle))) 238 return part; 239 } 240 return null; 241 } 242 243 /** 244 * Update headers. The default implementation here just 245 * calls the <code>updateHeaders</code> method on each of its 246 * children BodyParts. <p> 247 * 248 * Note that the boundary parameter is already set up when 249 * a new and empty MimeMultipart object is created. <p> 250 * 251 * This method is called when the <code>saveChanges</code> 252 * method is invoked on the Message object containing this 253 * MimeMultipart. This is typically done as part of the Message 254 * send process, however note that a client is free to call 255 * it any number of times. So if the header updating process is 256 * expensive for a specific MimeMultipart subclass, then it 257 * might itself want to track whether its internal state actually 258 * did change, and do the header updating only if necessary. 259 */ 260 protected void updateHeaders() throws MessagingException { 261 for (int i = 0; i < parts.size(); i++) 262 ((MimeBodyPart)parts.get(i)).updateHeaders(); 263 } 264 265 /** 266 * Iterates through all the parts and outputs each Mime part 267 * separated by a boundary. 268 */ 269 public void writeTo(OutputStream os) 270 throws IOException, MessagingException { 271 parse(); 272 273 String boundary = "--" + contentType.getParameter("boundary"); 274 275 for (int i = 0; i < parts.size(); i++) { 276 OutputUtil.writeln(boundary, os); // put out boundary 277 getBodyPart(i).writeTo(os); 278 OutputUtil.writeln(os); // put out empty line 279 } 280 281 // put out last boundary 282 OutputUtil.writeAsAscii(boundary, os); 283 OutputUtil.writeAsAscii("--", os); 284 os.flush(); 285 } 286 287 /** 288 * Parse the InputStream from our DataSource, constructing the 289 * appropriate MimeBodyParts. The <code>parsed</code> flag is 290 * set to true, and if true on entry nothing is done. This 291 * method is called by all other methods that need data for 292 * the body parts, to make sure the data has been parsed. 293 * 294 * @since JavaMail 1.2 295 */ 296 protected void parse() throws MessagingException { 297 if (parsed) 298 return; 299 300 InputStream in; 301 SharedInputStream sin = null; 302 long start = 0, end = 0; 303 boolean foundClosingBoundary = false; 304 305 try { 306 in = ds.getInputStream(); 307 if (!(in instanceof ByteArrayInputStream) && 308 !(in instanceof BufferedInputStream) && 309 !(in instanceof SharedInputStream)) 310 in = new BufferedInputStream(in); 311 } catch (Exception ex) { 312 throw new MessagingException("No inputstream from datasource"); 313 } 314 if (in instanceof SharedInputStream) 315 sin = (SharedInputStream)in; 316 317 String boundary = "--" + contentType.getParameter("boundary"); 318 byte[] bndbytes = ASCIIUtility.getBytes(boundary); 319 int bl = bndbytes.length; 320 321 ByteOutputStream buf = null; 322 try { 323 // Skip the preamble 324 LineInputStream lin = new LineInputStream(in); 325 String line; 326 while ((line = lin.readLine()) != null) { 327 /* 328 * Strip trailing whitespace. Can't use trim method 329 * because it's too aggressive. Some bogus MIME 330 * messages will include control characters in the 331 * boundary string. 332 */ 333 int i; 334 for (i = line.length() - 1; i >= 0; i--) { 335 char c = line.charAt(i); 336 if (!(c == ' ' || c == '\t')) 337 break; 338 } 339 line = line.substring(0, i + 1); 340 if (line.equals(boundary)) 341 break; 342 } 343 if (line == null) 344 throw new MessagingException("Missing start boundary"); 345 346 /* 347 * Read and process body parts until we see the 348 * terminating boundary line (or EOF). 349 */ 350 boolean done = false; 351 getparts: 352 while (!done) { 353 InternetHeaders headers = null; 354 if (sin != null) { 355 start = sin.getPosition(); 356 // skip headers 357 while ((line = lin.readLine()) != null && line.length() > 0) 358 ; 359 if (line == null) { 360 if (!ignoreMissingEndBoundary) { 361 throw new MessagingException("Missing End Boundary for Mime Package : EOF while skipping headers"); 362 } 363 // assume there's just a missing end boundary 364 break getparts; 365 } 366 } else { 367 // collect the headers for this body part 368 headers = createInternetHeaders(in); 369 } 370 371 if (!in.markSupported()) 372 throw new MessagingException("Stream doesn't support mark"); 373 374 buf = null; 375 // if we don't have a shared input stream, we copy the data 376 if (sin == null) 377 buf = new ByteOutputStream(); 378 int b; 379 boolean bol = true; // beginning of line flag 380 // the two possible end of line characters 381 int eol1 = -1, eol2 = -1; 382 383 /* 384 * Read and save the content bytes in buf. 385 */ 386 for (;;) { 387 if (bol) { 388 /* 389 * At the beginning of a line, check whether the 390 * next line is a boundary. 391 */ 392 int i; 393 in.mark(bl + 4 + 1000); // bnd + "--\r\n" + lots of LWSP 394 // read bytes, matching against the boundary 395 for (i = 0; i < bl; i++) 396 if (in.read() != bndbytes[i]) 397 break; 398 if (i == bl) { 399 // matched the boundary, check for last boundary 400 int b2 = in.read(); 401 if (b2 == '-') { 402 if (in.read() == '-') { 403 done = true; 404 foundClosingBoundary = true; 405 break; // ignore trailing text 406 } 407 } 408 // skip linear whitespace 409 while (b2 == ' ' || b2 == '\t') 410 b2 = in.read(); 411 // check for end of line 412 if (b2 == '\n') 413 break; // got it! break out of the loop 414 if (b2 == '\r') { 415 in.mark(1); 416 if (in.read() != '\n') 417 in.reset(); 418 break; // got it! break out of the loop 419 } 420 } 421 // failed to match, reset and proceed normally 422 in.reset(); 423 424 // if this is not the first line, write out the 425 // end of line characters from the previous line 426 if (buf != null && eol1 != -1) { 427 buf.write(eol1); 428 if (eol2 != -1) 429 buf.write(eol2); 430 eol1 = eol2 = -1; 431 } 432 } 433 434 // read the next byte 435 if ((b = in.read()) < 0) { 436 done = true; 437 break; 438 } 439 440 /* 441 * If we're at the end of the line, save the eol characters 442 * to be written out before the beginning of the next line. 443 */ 444 if (b == '\r' || b == '\n') { 445 bol = true; 446 if (sin != null) 447 end = sin.getPosition() - 1; 448 eol1 = b; 449 if (b == '\r') { 450 in.mark(1); 451 if ((b = in.read()) == '\n') 452 eol2 = b; 453 else 454 in.reset(); 455 } 456 } else { 457 bol = false; 458 if (buf != null) 459 buf.write(b); 460 } 461 } 462 463 /* 464 * Create a MimeBody element to represent this body part. 465 */ 466 MimeBodyPart part; 467 if (sin != null) 468 part = createMimeBodyPart(sin.newStream(start, end)); 469 else 470 part = createMimeBodyPart(headers, buf.getBytes(), buf.getCount()); 471 addBodyPart(part); 472 } 473 } catch (IOException ioex) { 474 throw new MessagingException("IO Error", ioex); 475 } finally { 476 if (buf != null) 477 buf.close(); 478 } 479 480 if (!ignoreMissingEndBoundary && !foundClosingBoundary && sin== null) { 481 throw new MessagingException("Missing End Boundary for Mime Package : EOF while skipping headers"); 482 } 483 parsed = true; 484 } 485 486 /** 487 * Create and return an InternetHeaders object that loads the 488 * headers from the given InputStream. Subclasses can override 489 * this method to return a subclass of InternetHeaders, if 490 * necessary. This implementation simply constructs and returns 491 * an InternetHeaders object. 492 * 493 * @param is the InputStream to read the headers from 494 * @exception MessagingException 495 * @since JavaMail 1.2 496 */ 497 protected InternetHeaders createInternetHeaders(InputStream is) 498 throws MessagingException { 499 return new InternetHeaders(is); 500 } 501 502 /** 503 * Create and return a MimeBodyPart object to represent a 504 * body part parsed from the InputStream. Subclasses can override 505 * this method to return a subclass of MimeBodyPart, if 506 * necessary. This implementation simply constructs and returns 507 * a MimeBodyPart object. 508 * 509 * @param headers the headers for the body part 510 * @param content the content of the body part 511 * @since JavaMail 1.2 512 */ 513 protected MimeBodyPart createMimeBodyPart(InternetHeaders headers, byte[] content, int len) { 514 return new MimeBodyPart(headers, content,len); 515 } 516 517 /** 518 * Create and return a MimeBodyPart object to represent a 519 * body part parsed from the InputStream. Subclasses can override 520 * this method to return a subclass of MimeBodyPart, if 521 * necessary. This implementation simply constructs and returns 522 * a MimeBodyPart object. 523 * 524 * @param is InputStream containing the body part 525 * @exception MessagingException 526 * @since JavaMail 1.2 527 */ 528 protected MimeBodyPart createMimeBodyPart(InputStream is) throws MessagingException { 529 return new MimeBodyPart(is); 530 } 531 532 /** 533 * Setup this MimeMultipart object from the given MultipartDataSource. <p> 534 * 535 * The method adds the MultipartDataSource's MimeBodyPart 536 * objects into this MimeMultipart. This MimeMultipart's contentType is 537 * set to that of the MultipartDataSource. <p> 538 * 539 * This method is typically used in those cases where one 540 * has a multipart data source that has already been pre-parsed into 541 * the individual body parts (for example, an IMAP datasource), but 542 * needs to create an appropriate MimeMultipart subclass that represents 543 * a specific multipart subtype. 544 * 545 * @param mp MimeMultipart datasource 546 */ 547 548 protected void setMultipartDataSource(MultipartDataSource mp) 549 throws MessagingException { 550 contentType = new ContentType(mp.getContentType()); 551 552 int count = mp.getCount(); 553 for (int i = 0; i < count; i++) 554 addBodyPart(mp.getBodyPart(i)); 555 } 556 557 /** 558 * Return the content-type of this MimeMultipart. <p> 559 * 560 * This implementation just returns the value of the 561 * <code>contentType</code> field. 562 * 563 * @return content-type 564 * @see #contentType 565 */ 566 public ContentType getContentType() { 567 return contentType; 568 } 569 570 /** 571 * Remove the specified part from the multipart message. 572 * Shifts all the parts after the removed part down one. 573 * 574 * @param part The part to remove 575 * @return true if part removed, false otherwise 576 * @exception MessagingException if no such MimeBodyPart exists 577 */ 578 public boolean removeBodyPart(MimeBodyPart part) throws MessagingException { 579 if (parts == null) 580 throw new MessagingException("No such body part"); 581 582 boolean ret = parts.remove(part); 583 part.setParent(null); 584 return ret; 585 } 586 587 /** 588 * Remove the part at specified location (starting from 0). 589 * Shifts all the parts after the removed part down one. 590 * 591 * @param index Index of the part to remove 592 * @exception IndexOutOfBoundsException if the given index 593 * is out of range. 594 */ 595 public void removeBodyPart(int index) { 596 if (parts == null) 597 throw new IndexOutOfBoundsException("No such BodyPart"); 598 599 MimeBodyPart part = (MimeBodyPart)parts.get(index); 600 parts.remove(index); 601 part.setParent(null); 602 } 603 604 /** 605 * Adds a MimeBodyPart to the multipart. The MimeBodyPart is appended to 606 * the list of existing Parts. 607 * 608 * @param part The MimeBodyPart to be appended 609 */ 610 public synchronized void addBodyPart(MimeBodyPart part) { 611 if (parts == null) 612 parts = new FinalArrayList(); 613 614 parts.add(part); 615 part.setParent(this); 616 } 617 618 /** 619 * Adds a MimeBodyPart at position <code>index</code>. 620 * If <code>index</code> is not the last one in the list, 621 * the subsequent parts are shifted up. If <code>index</code> 622 * is larger than the number of parts present, the 623 * MimeBodyPart is appended to the end. 624 * 625 * @param part The MimeBodyPart to be inserted 626 * @param index Location where to insert the part 627 */ 628 public synchronized void addBodyPart(MimeBodyPart part, int index) { 629 if (parts == null) 630 parts = new FinalArrayList(); 631 632 parts.add(index,part); 633 part.setParent(this); 634 } 635 636 /** 637 * Return the <code>MimeBodyPart</code> that contains this <code>MimeMultipart</code> 638 * object, or <code>null</code> if not known. 639 * @since JavaMail 1.1 640 */ 641 MimeBodyPart getParent() { 642 return parent; 643 } 644 645 /** 646 * Set the parent of this <code>MimeMultipart</code> to be the specified 647 * <code>MimeBodyPart</code>. Normally called by the <code>Message</code> 648 * or <code>MimeBodyPart</code> <code>setContent(MimeMultipart)</code> method. 649 * <code>parent</code> may be <code>null</code> if the 650 * <code>MimeMultipart</code> is being removed from its containing 651 * <code>MimeBodyPart</code>. 652 * @since JavaMail 1.1 653 */ 654 void setParent(MimeBodyPart parent) { 655 this.parent = parent; 656 } 657 }