Merge pull request #3 from mrwoowoo/c-version

add service prarms

src/main.cpp

@@ -1,46 +1,52 @@
-#include <iostream>
+#include <iostream>
 #include <string>
 #include "httplib.h"
 
 int getLoginData(httplib::Result& res, std::string& loginURL, std::string& loginPath, std::string& queryString);
 
 int main(int argc, char** argv)
 {
-	if (argc != 3)
+	if (argc < 3)
 	{
-		std::cout << "使用方法:" << std::endl
-			<< "程序名 你的账号 你的密码" << std::endl
-			<< "例如:./ruijie_login 123456 123456" << std::endl;
+		std::cout << "使用方法:" << std::endl
+			<< "程序名 你的账号 你的密码 service(可选)" << std::endl
+			<< "例如:./ruijie_login 123456 123456" << std::endl;
 		return -1;
 	}
 	std::string username = argv[1];
 	std::string password = argv[2];
+	std::string service;
+	if (argc == 4)
+	{
+		service = argv[3];
+	}
 	httplib::Result checkResult = httplib::Client("http://www.google.cn").Get("/generate_204");
 	if (!checkResult)
 	{
-		std::cout << "登陆状态检查失败" << std::endl;
+		std::cout << "登陆状态检查失败" << std::endl;
 		return -1;
 	}
 	if (checkResult->status == 204)
 	{
-		std::cout << "你已连接网络,无需登录" << std::endl;
+		std::cout << "你已连接网络,无需登录" << std::endl;
 		return 0;
 	}
 	std::string loginURL, loginPath, queryString;
 	getLoginData(checkResult, loginURL, loginPath, queryString);
-	queryString = "userId=" + username + "&password=" + password + "&service=&queryString=" + queryString + "&operatorPwd=&operatorUserId=&validcode=&passwordEncrypt=false";
+	queryString = "userId=" + username + "&password=" + password + "&service=" + service + "&queryString=" + queryString +
+		"&operatorPwd=&operatorUserId=&validcode=&passwordEncrypt=false";
 	//std::cout << loginURL << std::endl;
 	//std::cout << loginPath << std::endl;
 	//std::cout << queryString << std::endl;
 	httplib::Result loginResult = httplib::Client(loginURL).Post(loginPath.c_str(), queryString, "application/x-www-form-urlencoded");
 	size_t result = loginResult->body.find("\"result\":\"success\"");
 	if (result == std::string::npos)
 	{
-		std::cout << "登录失败" << std::endl;
+		std::cout << "登录失败" << std::endl;
 	}
 	else 
 	{
-		std::cout << "登录成功" << std::endl;
+		std::cout << "登录成功" << std::endl;
 	}
 	return 0;
 }
@@ -53,7 +59,8 @@ int getLoginData(httplib::Result& res, std::string& loginURL, std::string& login
 	index3 = res->body.find("index.jsp");
 	index4 = res->body.find_first_of("?");
 	index5 = res->body.find_last_of("'");
-	if (index1 == std::string::npos || index2 == std::string::npos || index3 == std::string::npos || index4 == std::string::npos || index5 == std::string::npos)
+	if (index1 == std::string::npos || index2 == std::string::npos || index3 == std::string::npos || 
+		index4 == std::string::npos || index5 == std::string::npos)
 	{
 		return 0;
 	}