使用httpclient模拟登录(Cookies使用)

原文地址：http://www.oschina.net/code/snippet_179804_46120

[1].[代码] [Java]代码 跳至[1]

?

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

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

import java.io.BufferedReader;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.util.ArrayList;

import java.util.List;

 

import org.apache.http.HttpEntity;

import org.apache.http.HttpResponse;

import org.apache.http.NameValuePair;

import org.apache.http.client.CookieStore;

import org.apache.http.client.HttpClient;

import org.apache.http.client.entity.UrlEncodedFormEntity;

import org.apache.http.client.methods.HttpGet;

import org.apache.http.client.methods.HttpPost;

import org.apache.http.cookie.Cookie;

import org.apache.http.impl.client.AbstractHttpClient;

import org.apache.http.impl.client.BasicCookieStore;

import org.apache.http.impl.client.DefaultHttpClient;

import org.apache.http.impl.cookie.BasicClientCookie;

import org.apache.http.message.BasicNameValuePair;

import org.apache.http.util.EntityUtils;

import org.junit.Test;

 

public class HttpClientLoginTest {

    @Test

    // 获取一个HTML页面的内容，一个简单的get应用

    publicvoid grabPageHTML() throws Exception {

        HttpClient httpclient =new DefaultHttpClient();

        HttpGet httpget =new HttpGet("http://www.baidu.com/");

        HttpResponse response = httpclient.execute(httpget);

        HttpEntity entity = response.getEntity();

        String html = EntityUtils.toString(entity,"GBK");

 

        // releaseConnection等同于reset，作用是重置request状态位，为下次使用做好准备。

        // 其实就是用一个HttpGet获取多个页面的情况下有效果；否则可以忽略此方法。

        httpget.releaseConnection();

 

        System.out.println(html);

    }

 

    // 下载一个文件到本地（本示范中为一个验证码图片）

    @Test

    publicvoid downloadFile() throws Exception {

        String url ="http://www.lashou.com/account/captcha";

        File dir =new File("D:\\TDDOWNLOAD");

        if(!dir.exists()) {

            dir.mkdirs();

        }

        String destfilename ="D:\\TDDOWNLOAD\\yz.png";

        HttpClient httpclient =new DefaultHttpClient();

        HttpGet httpget =new HttpGet(url);

        File file =new File(destfilename);

        if(file.exists()) {

            file.delete();

        }

 

        HttpResponse response = httpclient.execute(httpget);

        HttpEntity entity = response.getEntity();

        InputStream in = entity.getContent();

        try{

            FileOutputStream fout =new FileOutputStream(file);

            intl = -1;

            byte[] tmp =new byte[2048];

            while((l = in.read(tmp)) != -1) {

                fout.write(tmp);

            }

            fout.close();

        }finally {

            // 在用InputStream处理HttpEntity时，切记要关闭低层流。

            in.close();

        }

 

        httpget.releaseConnection();

    }

 

    @Test

    // Post方法，模拟表单提交参数登录到网站。

    // 结合了上面两个方法：grabPageHTML/downloadFile，同时增加了Post的代码。

    publicvoid login2Lashou() throws Exception {

        // 第一步：先下载验证码到本地

        String url ="http://www.lashou.com/account/captcha";

        String destfilename ="D:\\TDDOWNLOAD\\yz.png";

        HttpClient httpclient =new DefaultHttpClient();

        HttpGet httpget =new HttpGet(url);

        File file =new File(destfilename);

        if(file.exists()) {

            file.delete();

        }

 

        HttpResponse response = httpclient.execute(httpget);

        HttpEntity entity = response.getEntity();

        InputStream in = entity.getContent();

        try{

            FileOutputStream fout =new FileOutputStream(file);

            intl = -1;

            byte[] tmp =new byte[2048];

            while((l = in.read(tmp)) != -1) {

                fout.write(tmp);

            }

            fout.close();

        }finally {

            in.close();

        }

        httpget.releaseConnection();

 

        // 第二步：用Post方法带若干参数尝试登录，需要手工输入下载验证码中显示的字母、数字

        BufferedReader br =new BufferedReader(newInputStreamReader(System.in));

        System.out.println("请输入下载下来的验证码中显示的数字...");

        String yan = br.readLine();

 

        HttpPost httppost =new HttpPost("http://www.lashou.com/account/login/");

        List<NameValuePair> params =new ArrayList<NameValuePair>();

        params.add(newBasicNameValuePair("user_id","testuser007"));

        params.add(newBasicNameValuePair("pwd","asdfg123"));

        params.add(newBasicNameValuePair("yan", yan));

        params.add(newBasicNameValuePair("save_user","on"));

        params.add(newBasicNameValuePair("save_pwd","on"));

        params.add(newBasicNameValuePair("sub","登录"));

        httppost.setEntity(newUrlEncodedFormEntity(params));

 

        response = httpclient.execute(httppost);

        entity = response.getEntity();

        // 在这里可以用Jsoup之类的工具对返回结果进行分析，以判断登录是否成功

        String postResult = EntityUtils.toString(entity,"GBK");

        // 我们这里只是简单的打印出当前Cookie值以判断登录是否成功。

        CookieStore cookieStore = ((AbstractHttpClient) httpclient).getCookieStore();

        List<Cookie> cookies = ((AbstractHttpClient) httpclient)

                .getCookieStore().getCookies();

        for(Cookie cookie : cookies)

            System.out.println("cookie begin***\n"+ cookie + "\n cookie end");

        httppost.releaseConnection();

 

        // 第三步：打开会员页面以判断登录成功（未登录用户是打不开会员页面的）

        String memberpage ="http://www.lashou.com/account/orders/";

        httpget =new HttpGet(memberpage);

        response = httpclient.execute(httpget);// 必须是同一个HttpClient！

        entity = response.getEntity();

        String html = EntityUtils.toString(entity,"GBK");

        httpget.releaseConnection();

 

        System.out.println(html);

    }

 

    @Test

    publicvoid testSystemIn() throws IOException {

        BufferedReader reader =new BufferedReader(newInputStreamReader(

                System.in));

        String hello = reader.readLine();

        System.out.println(hello);

    }

 

    @Test

    // 设置已获取的cookie，查看需要登录后才能查看的页面

    publicvoid testGetinfoByLoginCookie()throws Exception, IOException {

        DefaultHttpClient httpclient =new DefaultHttpClient();

        CookieStore cookieStore =new BasicCookieStore();

        BasicClientCookie cookie1 =new BasicClientCookie("ThinkID","5s4tmqem08gh091v3egoa7sqf7");

        cookie1.setDomain(".lashou.com");

        BasicClientCookie cookie2 =new BasicClientCookie("city_b","2419");

        cookie2.setDomain("lashou.com");

        BasicClientCookie cookie3 =new BasicClientCookie("client_key","1425104707wd157b4b24ff70adcb875a");

        cookie3.setDomain("lashou.com");

        BasicClientCookie cookie4 =new BasicClientCookie("login_name2","testuser007");

        cookie4.setDomain("lashou.com");

        BasicClientCookie cookie5 =new BasicClientCookie("pwd2","4c88a4062736c26572d3ec382868fa2b");

        cookie5.setDomain("lashou.com");

        cookieStore.addCookie(cookie1);

        cookieStore.addCookie(cookie2);

        cookieStore.addCookie(cookie3);

        cookieStore.addCookie(cookie4);

        cookieStore.addCookie(cookie5);

        List<Cookie> cookies =new ArrayList<Cookie>();

        httpclient.setCookieStore(cookieStore);

         

        List<Cookie> cookieList = httpclient.getCookieStore().getCookies();

        for(inti=0;i<cookieList.size();i++){

            System.out.println("cookie"+i+":"+cookieList.get(i));

        }

         

        // 设置已登录的cookie

        String memberpage ="http://www.lashou.com/account/orders/";

        HttpGet httpget =new HttpGet(memberpage);

        HttpResponse response = httpclient.execute(httpget);// 必须是同一个HttpClient！

        HttpEntity entity = response.getEntity();

        entity = response.getEntity();

        String html = EntityUtils.toString(entity,"GBK");

        httpget.releaseConnection();

 

        System.out.println(html);

    }

}