OSDN Git Service

16ae00e9bd4e7a8f75feabbcb21bff9c910be38d
[webchat/WebChat.git] / chatServer.js
1 /*\r
2  * 設定\r
3  */\r
4 $max_room_number = 3;   //最大ルーム数\r
5 $spilt_size = 1024 * 512;       //分割するサイズ\r
6 $reset_password_diff = 1000 * 60 * 60;\r
7 $block_message = "メッセージの送信に失敗しました";       //ブロック時のメッセージ\r
8 $not_match_password = "パスワードが一致しませんでした";  //パスワードが一致しない場合に表示されるメッセージ\r
9 $password_setted_message = "パスワードを設定しました";      //パスワードが設定されたときに表示されるメッセージ\r
10 $password_resetted_message = "パスワードをリセットしました";      //パスワードが再設定されたときに表示されるメッセージ\r
11 $failed_set_password_message = "パスワードの設定に失敗しました"; //パスワードが再設定されたときに表示されるメッセージ\r
12 $ip_ban_list_file_name = "ipbanlist.txt";       //アクセスを禁止するIPが記録されているファイル\r
13 $room_configure_file_name = "roomlist.txt";     //ルームの設定が記録されているファイル\r
14 $port = process.env.port || 3000;       //ポート\r
15 $username = "admin";    //管理者用のページにアクセスできるユーザ名\r
16 $password = "admin";    //管理者用のページにアクセスするのに必要なパスワード\r
17 $token_length = 32;     //トークンの長さ\r
18 $redisHost = "localhost";       //redisサーバのアドレス\r
19 $redisPort = 6379;      //redisサーバのポート\r
20 $redisPassword = "";    //redisサーバのパスワード\r
21 $system_name = "system";        //システム発言を表す名前\r
22 $log_directory = "log"; //ログファイルを置くフォルダー\r
23 $log_file_name = "logfile%d.txt";       //ログファイル名(%dはそのままにしておくこと)\r
24 $splited_log_file_name = "logfile%d_%s.txt"     //分割後のファイル名(%dと%sはそのままにしておくこと)\r
25 $pastlogfile_pattern = "logfile%d(_+.*)?\.txt"; //過去ログと判定する正規表現\r
26 $logfile_pattern = "logfile[0-9]+(_*.*)?\.txt"  //過去ログと判定する正規表現\r
27 \r
28 /**\r
29  * Module dependencies.\r
30  */\r
31 \r
32 // Server\r
33 var express = require('express');\r
34 \r
35 var app = module.exports = express.createServer();\r
36 \r
37 var util = require("util");\r
38 \r
39 var lazy = require("lazy");\r
40 \r
41 var fs = require("fs");\r
42 \r
43 var parseCookie = require("connect").utils.parseCookie;\r
44 \r
45 var RedisStore = require("connect-redis")(express);\r
46 var sessionStore = new RedisStore({host:$redisHost,port:$redisPort,pass:$redisPassword});\r
47 \r
48 var async = require("async");\r
49 \r
50 var path = require("path");\r
51 \r
52 // Configuration\r
53 \r
54 app.configure(function(){\r
55         app.disabled("view cache");\r
56         app.set("view options", { layout: false })\r
57         app.set("views", __dirname + "/public");\r
58         app.set("view engine", "ejs");\r
59         app.use(express.bodyParser());\r
60         app.use(express.methodOverride());\r
61         app.use(express.cookieParser());\r
62         app.use(express.session({\r
63                 store:sessionStore,\r
64                 secret: "5514EA2B-C9B2-4D65-8D81-1F33A180A0C2",\r
65                 cookie: { httpOnly: false }\r
66         }));\r
67         app.use(app.router);\r
68         app.use(express.static(__dirname + "/public"));\r
69 });\r
70 \r
71 app.configure('development', function(){\r
72   app.use(express.errorHandler({ dumpExceptions: true, showStack: true })); \r
73 });\r
74 \r
75 app.configure('production', function(){\r
76   app.use(express.errorHandler()); \r
77 });\r
78 \r
79 // Routes\r
80 app.get("/chat", function(req, res){\r
81         var auth_string = getRandomString($token_length);\r
82         req.session.items = {token:auth_string};\r
83 \r
84         var room_number = 0;\r
85         if(typeof(req.query.rno) != "undefined")\r
86                 room_number = req.query.rno;\r
87         res.render("chat",{rno:room_number,token:auth_string});\r
88 });\r
89 \r
90 app.all("/log/" + $logfile_pattern,express.basicAuth(function (user, pass) {\r
91         return user === $username && pass === $password;\r
92 }));\r
93 \r
94 app.get("/log/" + $logfile_pattern, function(req, res){\r
95         res.sendfile(__dirname + req.url);\r
96 });\r
97 \r
98 app.all("/admin",express.basicAuth(function (user, pass) {\r
99         return user === $username && pass === $password;\r
100 }));\r
101 \r
102 app.get("/admin", function(req, res){\r
103         renderAdmin(req,res);\r
104 });\r
105 \r
106 app.post("/admin",function(req,res){\r
107         if(req.session.items.token != req.body.token)\r
108         {\r
109                 res.send($invaild_token_message);\r
110                 return;\r
111         }\r
112         if(typeof(req.body.erase) != "undefined")\r
113         {\r
114                 removeLog(req.body.file,function(){\r
115                         renderAdmin(req,res);\r
116                 });\r
117         }\r
118         if(typeof(req.body.registor) != "undefined")\r
119         {\r
120                 ipbanlist.Update(req.body.newbanlist,function(){\r
121                         renderAdmin(req,res);\r
122                 });\r
123         }\r
124         if(typeof(req.body.updateroom) != "undefined")\r
125         {\r
126                 $rooms.Update(req.body.newroomlist,function(){\r
127                         renderAdmin(req,res);\r
128                 });\r
129         }\r
130 });\r
131 \r
132 function renderAdmin(req,res)\r
133 {\r
134         var auth_string = getRandomString($token_length);\r
135         req.session.items = {token:auth_string};\r
136         var iplist = ipbanlist.GetText();\r
137 \r
138         fs.readdir($log_directory,function(err,list){\r
139                 res.render("admin", {\r
140                         files: list,\r
141                         log_directory:$log_directory,\r
142                         ipbanlist:iplist,\r
143                         token:auth_string,\r
144                         roomlist:$rooms.GetString()\r
145                 });\r
146         });\r
147 }\r
148 \r
149 function getRandomString(length)\r
150 {\r
151         var RandomString = "";\r
152         var BaseString ="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!#$%&'()=~|-^\@[;:],./\`{+*}<>?_";\r
153         for(var i=0; i<length; i++) {\r
154                 RandomString += BaseString.charAt( Math.floor( Math.random() * BaseString.length));\r
155         }\r
156         return RandomString\r
157 }\r
158 \r
159 function removeLog(files,callback)\r
160 {\r
161         async.map(files,\r
162         function(item,callback){\r
163                 fs.unlink($log_directory + "/" + item,callback);\r
164         },\r
165         function(err,results){\r
166                 if(typeof(callback) == "function")\r
167                         callback();\r
168         });\r
169 }\r
170 \r
171 app.listen($port);\r
172 console.log("Express server listening on port %d in %s mode", app.address().port, app.settings.env);\r
173 \r
174 /*\r
175  * サーバー部分\r
176  */\r
177 \r
178 var io = require("socket.io").listen(app);\r
179 io.configure('production', function(){\r
180   io.enable('browser client minification');  // minified されたクライアントファイルを送信する\r
181   io.enable('browser client etag');          // バージョンによって etag によるキャッシングを有効にする\r
182   io.set('log level', 1);                    // ログレベルを設定(デフォルトより下げている)\r
183 });\r
184 var clients = new Array();\r
185 \r
186 var ipbanlist = new IpBanCollecion();\r
187 var $rooms = new RoomInfomationCollection();\r
188 \r
189 createLogDirectory();\r
190 \r
191 for(var i = 0; i < $max_room_number; i++)\r
192 {\r
193         clients[i] =io\r
194         .of(GetNameFromRoomNumber(i))\r
195         .authorization(ParseAuthorization)\r
196         .on("connection", function (socket) {\r
197                 console.log("connected from %s",GetClientIPAdress(socket));\r
198 \r
199                 var rno = GetRoomNumberFromName(socket.namespace.name);\r
200                 var roomconfig = {};\r
201                 if($rooms.IsFixedPassword(rno))\r
202                         roomconfig.type = 2;\r
203                 else if($rooms.IsContains(rno))\r
204                         roomconfig.type = 1;\r
205                 else\r
206                         roomconfig.type = 0;\r
207                 roomconfig.IsOwned = !$rooms.IsFirstAuth(rno);\r
208                 socket.json.emit("send roominfo",roomconfig);\r
209 \r
210                 socket.on("get pastLogList", function (msg) {\r
211                         ParseGetPastLogList(socket,msg);\r
212                 });\r
213                 socket.on("get pastLog", function (msg) {\r
214                         ParseGetPastLog(socket,msg);\r
215                 });\r
216                 socket.on("join",function(msg){\r
217                         ParseJoin(socket,msg);\r
218                 });\r
219                 socket.on("quit",function(msg){\r
220                         ParseQuit(socket,msg);\r
221                 });\r
222                 socket.on("set password",function(msg){\r
223                         ParseSetPassword(socket,msg);\r
224                 });\r
225                 socket.on("send msg", function (msg) {\r
226                         ParseSendMsg(socket,msg);\r
227                 });\r
228                 socket.on("disconnect", function (msg) {\r
229                         ParseDisconnect(socket,msg);\r
230                 });\r
231         });\r
232 }\r
233 \r
234 function createLogDirectory()\r
235 {\r
236         path.exists($log_directory,function(exists){\r
237                 if(exists == false)\r
238                         fs.mkdirSync($log_directory);\r
239         });\r
240 }\r
241 \r
242 function ParseAuthorization(handshakeData, callback)\r
243 {\r
244         if(handshakeData.headers.cookie) {\r
245                 var cookie = handshakeData.headers.cookie;\r
246                 var sessionID = parseCookie(cookie)["connect.sid"];\r
247                 sessionStore.get(sessionID, function (err, session) {\r
248                         var result = null;\r
249                         if (err || ipbanlist.IsBaned(handshakeData.address.address))\r
250                                 result = "failed get from session store";\r
251                         else if(handshakeData.query.token != session.items.token)\r
252                                 result = "invaild token";\r
253                         sessionStore.destroy(sessionID);\r
254                         callback(result,result == null && !err);\r
255                 });\r
256         } else {\r
257                 return callback("failed get cookie", false);\r
258         }\r
259 }\r
260 \r
261 function ParseDisconnect(socket,msg)\r
262 {\r
263         console.log("disconnected");\r
264 }\r
265 \r
266 function ParseSetPassword(socket,msg)\r
267 {\r
268         var rno = GetRoomNumberFromName(socket.namespace.name);\r
269         var newMeg = {\r
270                 name:$system_name,\r
271                 message:null,\r
272         };\r
273         if($rooms.SetPassword(rno,msg.owner,msg.password))\r
274                 newMeg.message = $password_setted_message;\r
275         else\r
276                 newMeg.message = $failed_set_password_message;\r
277         ParseSendMsg(socket,newMeg);\r
278 }\r
279 \r
280 function ParseJoin(socket,msg)\r
281 {\r
282         var ip = GetClientIPAdress(socket);\r
283 \r
284         if(ipbanlist.IsBlockedToWrite(ip))\r
285         {\r
286                 socket.emit("error",$block_message);\r
287                 return;\r
288         }\r
289 \r
290         var rno = GetRoomNumberFromName(socket.namespace.name);\r
291         if($rooms.IsContains(rno))\r
292         {\r
293                 if($rooms.IsTimeout(rno) ||\r
294                         $rooms.IsFirstAuth(rno))\r
295                 {\r
296                         $rooms.Reset(rno,msg.name);\r
297                         ParseGetPastLog(socket,util.format($log_file_name,rno));\r
298                 }\r
299                 else if($rooms.Auth(rno,msg.name,msg.password))\r
300                 {\r
301                         ParseGetPastLog(socket,util.format($log_file_name,rno));\r
302                 }\r
303                 else\r
304                 {\r
305                         socket.emit("error",$not_match_password);\r
306                         return;\r
307                 }\r
308         }\r
309 \r
310         var newMeg = {\r
311                 name:$system_name,\r
312                 message:util.format("/enteredby %s %s %s",msg.name,msg.color,msg.mailto),\r
313         };\r
314         ParseSendMsg(socket,newMeg);\r
315 }\r
316 \r
317 function ParseQuit(socket,msg)\r
318 {\r
319         var ip = GetClientIPAdress(socket);\r
320 \r
321         if(ipbanlist.IsBlockedToWrite(ip))\r
322         {\r
323                 socket.emit("error",$block_message);\r
324                 return;\r
325         }\r
326 \r
327         var rno = GetRoomNumberFromName(socket.namespace.name);\r
328 \r
329         var newMeg = {\r
330                 name:$system_name,\r
331                 message:$password_resetted_message,\r
332         };\r
333         if($rooms.IsContains(rno))\r
334         {\r
335                 if($rooms.IsOwner(rno,msg.name))\r
336                 {\r
337                         $rooms.Reset(rno,null);\r
338                         ParseSendMsg(socket,newMeg);\r
339                 }\r
340                 if(!$rooms.IsFirstAuth(rno) &&\r
341                         !$rooms.IsAuthed(rno,msg.name))\r
342                         return;\r
343                 else\r
344                         $rooms.RemoveAuth(rno,msg.name);\r
345         }\r
346 \r
347         newMeg.message = util.format("/quitedby %s",msg.name);\r
348         ParseSendMsg(socket,newMeg);\r
349 }\r
350 \r
351 //socket 接続中のソケット\r
352 //msg msgクラス\r
353 function ParseSendMsg(socket,msg)\r
354 {\r
355         var ip = GetClientIPAdress(socket);\r
356 \r
357         if(ip in ipbanlist)\r
358         {\r
359                 socket.emit("error",$block_message);\r
360                 return;\r
361         }\r
362 \r
363         var rno = GetRoomNumberFromName(socket.namespace.name);\r
364 \r
365         if(msg.name != $system_name && \r
366                 $rooms.IsContains(rno) &&\r
367                 !$rooms.IsAuthed(rno,msg.name) &&\r
368                 !$rooms.IsOwner(rno,msg.name))\r
369         {\r
370                 return;\r
371         }\r
372 \r
373         var date = new Date();\r
374 \r
375         var repacked_msg = CreateMessage(msg.name,date,msg.message);\r
376 \r
377         socket.json.emit("req msg", repacked_msg);\r
378 \r
379         socket.json.broadcast.emit("req msg", repacked_msg);\r
380 \r
381         var path = $log_directory + "/" + util.format($log_file_name,rno);\r
382         var log = new ChatLog(path);\r
383         log.Save(repacked_msg,ip,rno);\r
384 }\r
385 \r
386 function GetNameFromRoomNumber(number)\r
387 {\r
388         return "/" + number;\r
389 }\r
390 \r
391 function GetRoomNumberFromName(name)\r
392 {\r
393         if(name.charAt(0) == "/")\r
394                 return parseInt(name.substr(1));\r
395         throw "GetRoomNumberFromName error";\r
396 }\r
397 \r
398 function ParseGetPastLogList(socket,msg)\r
399 {\r
400         var list = fs.readdir($log_directory,function(err,files){\r
401                 var text = "";\r
402                 var rno = GetRoomNumberFromName(socket.namespace.name);\r
403                 var pattern = $pastlogfile_pattern.replace("%d",rno);\r
404                 for(var i = 0; i < files.length; i++)\r
405                 {\r
406                         var logname = files[i];\r
407                         if(logname.match(pattern))\r
408                                 text += files[i] + "\n";\r
409                 }\r
410                 socket.emit("req pastloglist",text);\r
411         });\r
412 }\r
413 \r
414 function ParseGetPastLog(socket,file)\r
415 {\r
416         if(file == "")\r
417                 return;\r
418         var path = $log_directory + "/" + file;\r
419         var log = new ChatLog(path);\r
420         log.ToArray(function(array){\r
421                 socket.json.emit("req pastlog",array);\r
422         });\r
423 }\r
424 \r
425 function ChatLog(path)\r
426 {\r
427         this.ToArray = function(callback)\r
428         {\r
429                 var state = fs.stat(path,function(err,state){\r
430                         if(err)\r
431                                 return;\r
432                         var array = new Array();\r
433                         var stream = fs.createReadStream(path);\r
434                         new lazy(stream)\r
435                                 .lines\r
436                                 .forEach(function(line){\r
437                                         array.push(CreateMessageFromText(line.toString()));\r
438                                 })\r
439                                 .join(function(){\r
440                                         callback(array);\r
441                                 });\r
442                 });\r
443         }\r
444 \r
445         this.Save = function(msg,ip,rno){\r
446                 var text = GetTextFromMessage(msg,ip);\r
447 \r
448                 SplitLog(rno,function(){\r
449                         WritePastLog(path,text);\r
450                 });\r
451         };\r
452 \r
453         function GetTextFromMessage(msg,ip)\r
454         {\r
455                 var text = msg.name + "<>" +\r
456                                 msg.date + "<>" +\r
457                                 ip + "<>" +\r
458                                 msg.message +\r
459                                 "\n";\r
460                 return text;\r
461         }\r
462 \r
463         function SplitLog(rno,callback)\r
464         {\r
465                 var state = fs.stat(path,function(err,state){\r
466                         if(err && typeof(callback) == "function")\r
467                         {\r
468                                 callback();\r
469                                 return;\r
470                         }\r
471                         if(state.size > $spilt_size)\r
472                         {\r
473                                 var date = new Date();\r
474                                 var dateString = ""+date.getFullYear()+date.getMonth()+date.getDate()+date.getHours()+date.getMinutes()+date.getSeconds();\r
475 \r
476                                 var newpath = $log_directory + "/" +\r
477                                         util.format($splited_log_file_name,rno,dateString);\r
478                                 fs.rename(path,newpath,callback);\r
479                         }else{\r
480                                 if(typeof(callback) == "function")\r
481                                         callback();\r
482                         }\r
483                 });\r
484         }\r
485 \r
486         function WritePastLog(path,text)\r
487         {\r
488                 async.waterfall([\r
489                         function(callback){\r
490                                 fs.open(path,"a",callback);\r
491                         },\r
492                         function(fd,callback){\r
493                                 var buf = new Buffer(text);\r
494                                 fs.write(fd,buf,0,Buffer.byteLength(text),null,function(){\r
495                                         callback(null,fd);\r
496                                 });\r
497                         },\r
498                         function(fd){\r
499                                 fs.close(fd);\r
500                         }\r
501                 ]);\r
502         }\r
503 }\r
504 \r
505 function GetClientIPAdress(socket)\r
506 {\r
507         return socket.handshake.headers["x-forwarded-for"] || socket.handshake.address.address;\r
508 }\r
509 \r
510 // Message クラス\r
511 function CreateMessage(name,date,message)\r
512 {\r
513         var result = {name:name,\r
514                 date:date,\r
515                 message:message};\r
516         return result;\r
517 }\r
518 function CreateMessageFromText(text)\r
519 {\r
520         var data = text.split("<>");\r
521         var msg = {name:data[0],\r
522                 date:data[1],\r
523                 message:data[3]};\r
524         return msg;\r
525 }\r
526 \r
527 //RoomInfomationCollecionクラス\r
528 function RoomInfomationCollection()\r
529 {\r
530         var collection = {};\r
531         this.Reset = function(rno,owner){\r
532                 var date = new Date();\r
533                 var time = date.getTime();\r
534                 collection[rno].password = null;\r
535                 collection[rno].authed_list = {};\r
536                 collection[rno].owner = owner;\r
537                 collection[rno].time = time;\r
538                 console.log(util.format("password is reseted in %s",rno));\r
539         };\r
540         this.IsContains = function(rno){\r
541                 return rno in collection;\r
542         };\r
543         this.IsFirstAuth = function(rno){\r
544                 if(!this.IsContains(rno) || typeof(collection[rno].owner) == "undefined")\r
545                         return false;\r
546                 return collection[rno].owner == null;\r
547         }\r
548         this.IsAuthed = function(rno,name){\r
549                 if(!this.IsContains(rno))\r
550                         return false;\r
551                 return name == collection[rno].owner ||\r
552                         name in collection[rno].authed_list;\r
553         };\r
554         this.IsFixedPassword = function(rno){\r
555                 if(!this.IsContains(rno))\r
556                         return false;\r
557                 return collection[rno].owner == $system_name;\r
558         };\r
559         this.IsOwner = function(rno,name){\r
560                 return this.IsContains(rno) &&\r
561                         typeof(collection[rno].owner) != "undefined" &&\r
562                         collection[rno].owner == name;\r
563         }\r
564         this.IsTimeout = function(rno){\r
565                 var date = new Date();\r
566                 var time = date.getTime();\r
567                 return this.IsContains(rno) &&\r
568                         !this.IsFixedPassword(rno) &&\r
569                         (typeof(collection[rno].time) != "undefined" &&\r
570                         time - collection[rno].time >= $reset_password_diff);\r
571         };\r
572         this.RemoveAuth = function(rno,name)\r
573         {\r
574                 delete collection[rno].authed_list[name];\r
575         }\r
576         this.Auth = function(rno,name,password){\r
577                 if(typeof(collection[rno].password) != "undefined" &&\r
578                         collection[rno].password != password)\r
579                         return false;\r
580                 var date = new Date();\r
581                 var time = date.getTime();\r
582                 collection[rno].time = time;\r
583                 collection[rno].authed_list[name] = "";\r
584                 return true;\r
585         }\r
586         this.SetPassword = function(rno,owner,password){\r
587                 if(this.IsContains(rno) && \r
588                         owner == collection[rno].owner &&\r
589                         !this.IsFixedPassword(rno))\r
590                 {\r
591                         var date = new Date();\r
592                         collection[rno].time = date.getTime();\r
593                         collection[rno].password = password;\r
594                         \r
595                         console.log(util.format("password is seted to %s in %s",password,rno));\r
596                         return true;\r
597                 }\r
598                 return false;\r
599         };\r
600         this.GetString = function(){\r
601                 var retval = "";\r
602                 for(var rno in collection)\r
603                 {\r
604                         var pass = collection[rno].password;\r
605                         if(pass == null)\r
606                                 pass = "";\r
607                         retval += rno + ":" + pass + "\r\n";\r
608                 }\r
609                 return retval;\r
610         };\r
611         this.GetKeys = function(){\r
612                 var retval = {};\r
613                 for(var rno in collection)\r
614                 {\r
615                         retval[rno] = {};\r
616                 }\r
617                 return retval;\r
618         }\r
619         this.Update = function(text,callfunc){\r
620                 async.waterfall([\r
621                         function(callback){\r
622                                 fs.open($room_configure_file_name,"w",callback);\r
623                         },\r
624                         function(fd,callback){\r
625                                 var buf = new Buffer(text);\r
626                                 fs.write(fd,buf,0,Buffer.byteLength(text),null,function(){\r
627                                         callback(null,fd);\r
628                                 });\r
629                         },\r
630                         function(fd,callback){\r
631                                 fs.close(fd,function(){\r
632                                         GetRoomList(callfunc);\r
633                                 });\r
634                         }\r
635                 ]);\r
636         }\r
637         function GetRoomList(callback){\r
638                 Clear();\r
639                 path.exists($room_configure_file_name,function(exists){\r
640                         if(exists == false)\r
641                         {\r
642                                 if(typeof(callback) == "function")\r
643                                         callback();\r
644                                 return;\r
645                         }\r
646                         var stream = fs.createReadStream($room_configure_file_name);\r
647                         new lazy(stream)\r
648                                 .lines\r
649                                 .forEach(function(line){\r
650                                         var token = line.toString().replace(/(\r|\n|\r\n)/gm, "").split(":");\r
651                                         if(token.length == 1)\r
652                                         {\r
653                                                 Add(token[0],null);\r
654                                         }\r
655                                         else if(token.length == 2)\r
656                                         {\r
657                                                 var rno = token[0];\r
658                                                 var pass = token[1];\r
659                                                 if(pass == "")\r
660                                                         pass = null;\r
661                                                 Add(rno, pass);\r
662                                         }\r
663                                 })\r
664                                 .join(function(){\r
665                                         if(typeof(callback) == "function")\r
666                                                 callback();\r
667                                 });\r
668                 });\r
669         }\r
670         function Clear(){\r
671                 collection = {};\r
672         };\r
673         function Add(rno,pass){\r
674                 collection[rno] = {time : null,\r
675                         password : pass,\r
676                         owner : null,\r
677                         authed_list : {}};\r
678                 if(pass != null)\r
679                         collection[rno].owner = $system_name;\r
680         };\r
681 \r
682         GetRoomList();\r
683 }\r
684 \r
685 //IPBANクラス\r
686 function IpBanCollecion()\r
687 {\r
688         var collection = {};\r
689         this.IsBaned = function(ip){\r
690                 return collection[ip] == "r";\r
691         }\r
692         this.IsBlockedToWrite = function(ip){\r
693                 return ip in collection;\r
694         }\r
695         this.GetText = function(){\r
696                 var text = "";\r
697                 for(var key in collection)\r
698                 {\r
699                         if(collection[key] == "")\r
700                                 text += key + "\r\n";\r
701                         else\r
702                                 text += key + ":" + collection[key] + "\r\n";\r
703                 }\r
704                 return text;\r
705         }\r
706         this.Update = function(text,callfunc){\r
707                 async.waterfall([\r
708                         function(callback){\r
709                                 fs.open($ip_ban_list_file_name,"w",callback);\r
710                         },\r
711                         function(fd,callback){\r
712                                 var buf = new Buffer(text);\r
713                                 fs.write(fd,buf,0,Buffer.byteLength(text),null,function(){\r
714                                         callback(null,fd);\r
715                                 });\r
716                         },\r
717                         function(fd,callback){\r
718                                 fs.close(fd,function(){\r
719                                         GetIpBanList(callfunc);\r
720                                 });\r
721                         }\r
722                 ]);\r
723         }\r
724         function GetIpBanList(callback)\r
725         {\r
726                 collection = {};\r
727                 path.exists($ip_ban_list_file_name,function(exists){\r
728                         if(exists == false)\r
729                         {\r
730                                 if(typeof(callback) == "function")\r
731                                         callback();\r
732                                 return;\r
733                         }\r
734                         var stream = fs.createReadStream($ip_ban_list_file_name);\r
735                         new lazy(stream)\r
736                                 .lines\r
737                                 .forEach(function(line){\r
738                                         var token = line.toString().replace(/(\r|\n|\r\n)/gm, "").split(":");\r
739                                         var ip = token[0];\r
740                                         if(token.length == 1)\r
741                                                 collection[ip] = "";\r
742                                         else\r
743                                                 collection[ip] = token[1];\r
744                                 })\r
745                                 .join(function(){\r
746                                         if(typeof(callback) == "function")\r
747                                                 callback();\r
748                                 });\r
749                 });\r
750         }\r
751         GetIpBanList();\r
752 }\r
753 \r