com.ddling.client.smtp.SmtpClient.java Source code

Introduction

Here is the source code for com.ddling.client.smtp.SmtpClient.java

Source

/*
 * Copyright (C) 2014 lingdongdong
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.ddling.client.smtp;

import com.ddling.client.mailmanage.Mail;
import com.ddling.client.usermanage.UserService;
import com.ddling.client.utils.Constants;
import com.ddling.client.utils.LoggerFactory;
import org.apache.commons.codec.binary.Base64;
import org.apache.log4j.Logger;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.StringTokenizer;

/**
 * Created by lingdongdong on 14/12/25.
 */
public class SmtpClient {

    // log
    public Logger logger = LoggerFactory.getLogger(SmtpClient.class);

    // ?Socket
    private Socket socket = null;
    // socket???
    private BufferedReader in = null;
    // socket??
    private PrintWriter out = null;
    // ???
    private String server = null;
    // ????
    private int port;
    // ????
    private Mail mail = null;

    public SmtpClient() {
        logger.info("Start a smtp client");
    }

    public SmtpClient(Mail mail) {
        this.mail = mail;
        initilizeTheServer();
    }

    public void setServerAddressAndPort(String server, int port) {
        this.server = server;
        this.port = port;
    }

    /**
     * ???????
     */
    private void initilizeTheServer() {
        setServerAddressAndPort(Constants.LOCAL_SMTP_SERVER_ADDRESS, Constants.LOCAL_SMTP_SERVER_PORT);
    }

    /**
     * ??
     * @return ???true ??false
     */
    public boolean sendEmail() {
        boolean sendEmailOK = true;

        try {
            initClient();
            ehlo();
            auth();
            sendEmailHeader();
            sendEmailContent();
            quit();
        } catch (Exception e) {
            logger.error(e);
            sendEmailOK = false;
        } finally {
            this.closeTheClient();
        }

        return sendEmailOK;
    }

    private void auth() throws IOException {
        sendData("Auth login");

        int response = getResponse();

        if (response != 334) {
            throw new IOException("Auth login Fail!");
        }

        String username = Base64.encodeBase64String(UserService.getLoginUser().getUsername().getBytes());

        sendData(username);

        response = getResponse();

        if (response != 334) {
            throw new IOException("Username Auth fail!");
        }

        String password = Base64.encodeBase64String(UserService.getLoginUser().getPassword().getBytes());

        sendData(password);
        response = getResponse();

        if (response != 235) {
            throw new IOException("Auth password fail!");
        }
    }

    /**
     * ??
     * @throws java.io.IOException
     */
    private void quit() throws IOException {
        sendData("QUIT");

        int response = getResponse();

        if (response != 221) {
            throw new IOException("Quit fail");
        }

        logger.info("Quit Done!");
    }

    /**
     * ??
     * @throws java.io.IOException
     */
    private void sendEmailContent() throws IOException {

        sendData("DATA");

        int response = getResponse();

        if (response != 354) {
            throw new IOException("Send Email content fail");
        }

        sendData(mail.getContent());
        sendData(".");

        response = getResponse();
        if (response != 250) {
            throw new IOException("Send Email Content fail");
        }

        logger.info("Send Email Content done!");
    }

    /**
     * ???(mail from, rcpt to?subject)
     * @throws java.io.IOException
     */
    private void sendEmailHeader() throws IOException {

        sendData("MAIL FROM:<" + mail.getMail_from() + ">");

        int response = getResponse();

        if (response != 250) {
            throw new IOException("Send Email Header fail!");
        }

        sendData("RCPT TO:<" + mail.getMail_to() + ">");

        response = getResponse();

        if (response != 250) {
            throw new IOException("Send Email Header fail!");
        }

        logger.info("Send Email Header done!");
    }

    /**
     * ????helo
     * @throws java.io.IOException
     */
    private void ehlo() throws IOException {
        sendData("HELO " + server);

        int response = getResponse();

        if (response != 250) {
            throw new IOException("ehlo fail!");
        }

        logger.info("ehlo done!");
    }

    /**
     * ???
     * @param s ???
     */
    private void sendData(String s) {
        out.println(s);
        out.flush();
    }

    /**
     * ?smtp
     */
    private void initClient() {
        logger.info("Connect to " + server + " At port " + port);

        try {
            socket = new Socket(server, port);

            in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
            out = new PrintWriter(socket.getOutputStream(), true);

            int response = getResponse();

            if (response == 220) {
                logger.info("Connect to " + server + " done!");
            }

        } catch (IOException e) {
            logger.error(e);
        }
    }

    /**
     * ???
     * @return
     */
    private int getResponse() {

        String line = "";
        try {
            line = in.readLine();
        } catch (IOException e) {
            logger.error(e);
        }

        logger.debug(line);
        StringTokenizer get = new StringTokenizer(line, " ");
        return Integer.parseInt(get.nextToken());
    }

    /**
     * smtp
     */
    public void closeTheClient() {
        if (socket != null) {
            try {
                socket.close();
            } catch (IOException e) {
                logger.error(e);
            }
        }
    }

    public static void main(String[] args) {
    }
}