bool mime_builder::save_as(acl::ofstream& fp) { acl::string buf; if (attachs_.empty()) { //header_.set_type(MIME_CTYPE_TEXT, body_html_ ? // MIME_STYPE_HTML : MIME_STYPE_PLAIN); header_.set_type("text", body_html_ ? "html" : "plain"); header_.add_header("Content-Transfer-Encoding", "base64"); } else { //header_.set_type(MIME_CTYPE_MULTIPART, // MIME_STYPE_MIXED); header_.set_type("multipart", "mixed"); delimeter_.format("------=_Part_%d_%ld.%ld", getpid(), acl_pthread_self(), time(NULL)); header_.set_boundary(delimeter_.c_str()); } header_.build_head(buf, false); if (fp.write(buf) == -1) { logger_error("write head to file %s error %s", fp.file_path(), acl::last_serror()); return false; } if (!delimeter_.empty()) { if (add_boundary(fp) == false) return false; } if (add_body(fp) == false) return false; if (delimeter_.empty()) return true; std::vector<char*>::const_iterator cit = attachs_.begin(); for (; cit != attachs_.end(); ++cit) { if (add_boundary(fp) == false) return false; if (add_attach(fp, *cit) == false) return false; } // 添加最后一个分隔符 return add_boundary(fp, true); }
bool mime_builder::add_boundary(acl::ofstream& fp, bool end /* = false */) { if (fp.format("--%s", delimeter_.c_str()) == -1) { logger_error("write boundary error(%s) to %s", acl::last_serror(), fp.file_path()); return false; } else if (end) { if (fp.format("--\r\n\r\n") == -1) { logger_error("write boundary endline error(%s) to %s", acl::last_serror(), fp.file_path()); return false; } else return true; } else if (fp.format("\r\n") == -1) { logger_error("write boundary end error(%s) to %s", acl::last_serror(), fp.file_path()); return false; } else return true; }
bool mime_builder::add_body(acl::ofstream& fp) { acl::string buf; const char* ptr; size_t len; if (body_html_) ptr = body_html_; else ptr = body_text_; if (ptr == NULL) { logger_error("no body!"); return false; } int ret = fp.format("Content-Type: text/%s; charset=\"utf-8\"\r\n" "Content-Transfer-Encoding: base64\r\n\r\n", body_html_ ? "html" : "plain"); if (ret == -1) { logger_error("write body header to %s error %s", fp.file_path(), acl::last_serror()); return false; } len = strlen(ptr); acl::mime_base64::encode(ptr, (int) len, &buf); if (fp.write(buf) == -1) { logger_error("write body to %s error %s", fp.file_path(), acl::last_serror()); return false; } else if (fp.format("\r\n") == -1) { logger_error("write body endline to %s error %s", fp.file_path(), acl::last_serror()); return false; } else return true; }
bool mime_builder::add_attach(acl::ofstream& fp, const char* filepath) { ////////////////////////////////////////////////////////////////////////// // 创建头部分信息并写入邮件中 acl::string filebuf; global::get_filename(filepath, filebuf); acl::string filename; acl::rfc2047::encode(filebuf.c_str(), (int) filebuf.length(), &filename, "utf-8", 'B', false); acl::string header; header.format("Content-Type: application/octet-stream;\r\n" "\tname=\"%s\r\n" "Content-Disposition: attachment;\r\n" "\tfilename=\"%s\"\r\n" "Content-Transfer-Encoding: base64\r\n\r\n", filename.c_str(), filename.c_str()); if (fp.write(header) == -1) { logger_error("write header to file %s error %s", fp.file_path(), acl::last_serror()); return false; } ////////////////////////////////////////////////////////////////////////// // 读取附件数据并写入邮件中 acl::ifstream in; if (in.open_read(filepath) == false) { logger_error("open file %s error %s", filepath, acl::last_serror()); return false; } acl::mime_base64 coder; char inbuf[8192]; acl::string outbuf; int ret; while (!in.eof()) { ret = in.read(inbuf, sizeof(inbuf) - 1, false); if (ret == -1) break; coder.encode_update(inbuf, ret, &outbuf); if (!outbuf.empty() && fp.write(outbuf) == -1) { logger_error("write to %s error %s", fp.file_path(), acl::last_serror()); return false; } outbuf.clear(); } coder.encode_finish(&outbuf); if (outbuf.empty() == false && fp.write(outbuf) == -1) { logger_error("write to %s error %s", fp.file_path(), acl::last_serror()); return false; } if (fp.format("\r\n") == -1) { logger_error("write endline to %s error %s", fp.file_path(), acl::last_serror()); return false; } else return true; }