用java实现的邮件系统三（收取邮件服务器）

package com.wepull.pop3;

import java.net.Socket;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintStream;
import com.wepull.smtp.*;

public class POPSession extends Thread {

/**
* 获得一次会话
*/
private Socket s;

/**
* 读取网络
*/
private BufferedReader br;

/**
* 写网络
*/
private PrintStream ps;

private String usermsg;

private String usermsgpath;

private Set set = new HashSet();

/**
* 标记要删除的邮件
*/
private Map<String ,String> map = new HashMap<String,String>();

/**
* 把第几封邮件和邮件名对应起来
*/
private Map<String ,String> filemap = new HashMap<String,String>();

/**
* @param s
* 初始化网络输入和网络输出流
*/
public POPSession(Socket s) {
this.s = s;

try {
br = new BufferedReader(new InputStreamReader(s.getInputStream()));

ps = new PrintStream(s.getOutputStream());

} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}

/*
* (non-Javadoc)
*
* @see java.lang.Thread#run() 多线程的run方法
*/

@Override
public void run() {

ps.println("+OK POP3 server ready");
String line = "";
try {
while (!s.isClosed() && (line = br.readLine()) != null) {

System.out.println("客服端发送的命令：" + line);
/**
* 处理从客户端发来的命令
*/
doAction(line);
if ("quit".equalsIgnoreCase(line)) {
/**
* 处理quit命令(关闭流)
*/
doQuit();
break;
}

}

} catch (IOException e) {
if ("Connection reset".equalsIgnoreCase(e.getMessage())) {

} else {
e.printStackTrace();
}
} finally {

try {
br.close();
ps.close();
s.close();

} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

}

}

/**
* @param line
* 处理从客户端发来的命令
*/
private void doAction(String line) {
String msg = line.split(" ")[0];

if ("user".equalsIgnoreCase(msg)) {
/**
* 处理user命令(验证用户是否合法)
*/
doUser(line);

} else if ("pass".equalsIgnoreCase(msg)) {
/**
* 处理pass命令(验证用户密码是否正确)
*/
doPass(line);

} else if ("stat".equalsIgnoreCase(msg)) {
/**
* 处理stat命令(列出用户邮件数和邮件大小)
*/
doStat();
}else if ("noop".equalsIgnoreCase(msg)) {
/**
* 处理top(n,m)命令(输出第n封邮件的前m行)
*/
ps.println("+OK");
} else if ("top".equalsIgnoreCase(msg)) {
/**
* 处理top(n,m)命令(输出第n封邮件的前m行)
*/
doTop(line);
} else if ("list".equalsIgnoreCase(msg)) {
/**
* 处理list命令(列出每封邮件的大小)
*/
doList(line);
} else if ("retr".equalsIgnoreCase(msg)) {
/**
* 处理retr命令(给客服端发送指定邮件的内容)
*/
doRetr(line);
} else if ("dele".equalsIgnoreCase(msg)) {
/**
* 处理dele命令(标记指定要删除的邮件)
*/
doDele(line);
} else if ("rest".equalsIgnoreCase(msg)) {
/**
* 处理rest命令(取消指定删除的邮件的标记)
*/
doRest(line);
} else if ("uidl".equalsIgnoreCase(msg)) {
/**
* 处理uidl命令(验证用户是否合法)
*/
doUidl();
}
}

/**
* 处理user命令(验证用户是否合法)
*/
private void doUser(String msg) {
usermsg = msg.split(" ")[1];
usermsgpath = Common.RPATH + "//" + usermsg;
// msg = Common.RPATH + "//" + msg;
// System.out.println();
System.out.println(usermsg);
CopeDeleteFile.copy(new File(Common.PATHS + "//" + usermsg), new File(
Common.RPATH));
usermsgpath += "//" + "revd";
System.out.println(usermsgpath);
File userfile = new File(usermsgpath);
File[] userfiles = userfile.listFiles();
int count = 0;
String sum="";
for (int i = 0; i < userfiles.length; i++) {
count++;
sum=""+count;
filemap.put(sum, userfiles[i].getAbsolutePath());
}

ps.println("+OK is a real hoopy frood");
}

/**
* 处理pass命令(验证用户密码是否正确)
*/
private void doPass(String msg) {

ps.println("+OK " + List() + " message(s) [" + total() + " byte(s)]");
}

/**
* 处理stat命令(列出用户邮件数和邮件大小)
*/
private void doStat() {
ps.println("+OK " + List() + " " + total());

}

private long total() {
// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();
// long count = 0;
// for (int i = 0; i < userfiles.length; i++) {
// count += userfiles[i].length();
// }

Set set = filemap.keySet();
long count = 0;
Iterator ita = set.iterator();
while (ita.hasNext()) {
String key = (String) ita.next();
String filepath = (String) filemap.get(key);
File messagefile = new File(filepath);
count += messagefile.length();
}

return count;
}

private int List() {
// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();

// return userfiles.length;
return filemap.size();
}

private void doList(String line) {

// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();
// int count = 0;
// for (int i = 0; i < userfiles.length; i++) {
// ps.println(++count + " " + userfiles[i].length());
// }

if(line.split(" ").length==2){
line =line.split(" ")[1];
String filepath = (String) filemap.get(line);
File messagefile = new File(filepath);
ps.println("+OK "+line+" "+ messagefile.length());

}else{
ps.println("+OK " + List() + " " + "messages");
Set set = filemap.keySet();

Iterator ita = set.iterator();
while (ita.hasNext()) {
String key = (String) ita.next();
String filepath = (String) filemap.get(key);
File messagefile = new File(filepath);
ps.println(key + " " + messagefile.length());
}
}

ps.println(".");

}

private void doRetr(String msg) {
msg = msg.split(" ")[1];
File file = new File((String) filemap.get(msg));

ps.println("+OK" + " " + file.length() + " " + "octets");

BufferedReader brfile = null;
try {
brfile = new BufferedReader(new InputStreamReader(
new FileInputStream(file)));
String line = "";
while (true) {
line = brfile.readLine();
ps.println(line);
if (".".equalsIgnoreCase(line)) {
break;
}
}
} catch (NumberFormatException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} finally {
try {
brfile.close();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}

// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();
//
// ps.println("+OK" + " " + userfiles[Integer.parseInt(msg) -
// 1].length()
// + " " + "octets");
// BufferedReader brfile = null;
// try {
// brfile = new BufferedReader(new InputStreamReader(
// new FileInputStream(userfiles[Integer.parseInt(msg) - 1])));
// String line = "";
// while (true) {
// line = brfile.readLine();
// ps.println(line);
// if (".".equalsIgnoreCase(line)) {
// break;
// }
// }
// } catch (NumberFormatException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// } catch (FileNotFoundException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// } catch (IOException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// } finally {
// try {
// brfile.close();
// } catch (IOException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// }
// }

}

private void doDele(String msg) {
msg = msg.split(" ")[1];

// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();

// map.put(msg, userfiles[Integer.parseInt(msg) - 1].getAbsolutePath());
map.put(msg, filemap.get(msg));
filemap.remove(msg);
ps.println("+OK");

}

private void doRest(String msg) {
msg = msg.split(" ")[1];
// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();

filemap.put(msg, map.get(msg));
map.remove(msg);
ps.println("+ok");

}

private void doUidl() {
ps.println("+OK " + List() + " " + "messages");
Set set = filemap.keySet();
Iterator ita = set.iterator();
while (ita.hasNext()) {

String key = (String) ita.next();
String filepath = (String) filemap.get(key);
File messagefile = new File(filepath);
ps.println(key + " " + messagefile.getName());
System.out.println("come"+key+messagefile.getName());
}

// File userfile = new File(usermsgpath);
// File[] userfiles = userfile.listFiles();
// BufferedReader brfile = null;
// String line = "";
// for (int i = 0; i < userfiles.length; i++) {
// try {
// brfile = new BufferedReader(new InputStreamReader(
// new FileInputStream(userfiles[i])));
//
// line = brfile.readLine();
//
// } catch (FileNotFoundException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// } catch (IOException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// } finally {
// try {
// brfile.close();
// } catch (IOException e) {
// // TODO Auto-generated catch block
// e.printStackTrace();
// }
// }
// ps.println(++i + " " + line.split("<")[1].split(">")[0]);
// }

ps.println(".");

}

private void doTop(String msg) {

ps.println("+OK");
String n = "";
String m = "";
String[] str = msg.split(" ");
n = str[1];
File file = new File((String) filemap.get(n));
BufferedReader br = null;
String line = "";
try {
br = new BufferedReader(new InputStreamReader(new FileInputStream(
file)));
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
if (str.length == 2) {
while (true) {
try {
line = br.readLine();
ps.println(line);

if ("".equalsIgnoreCase(line)) {
ps.println(".");
break;
}

} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}

} else if (str.length == 3) {
m = str[2];
int count = 0;

if ("0".equalsIgnoreCase(m)) {
try {
while ((line = br.readLine())!=null) {
ps.println(line);
ps.flush();
if ("".equalsIgnoreCase(line)) {
//ps.println();
br.close();
break;
}
}
ps.println(".");
} catch (IOException e) {
e.printStackTrace();
}

} else {
while (true) {
try {
count++;
line = br.readLine();
ps.println(line);

if (count == Integer.parseInt(m)) {
ps.println(".");
break;
}

} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}

}

}

}

private void doQuit() {

ps.println("+OK dewey POP3 server signing off");
dodelefile();

}

private void dodelefile() {

Set set = map.keySet();

Iterator ita = set.iterator();

while (ita.hasNext()) {

String key = (String) ita.next();

String filename = (String) map.get(key);

File deltefile = new File(filename);
deltefile.delete();
}

}
}