source: lab.git/Dev/utvpn/utvpn-unix-v101-7101-public/src/Cedar/Remote.c @ 23aa5e1

trunk
Last change on this file since 23aa5e1 was a1bae3e, checked in by mitty <mitty@…>, 12 years ago
  • copy vendor drop to trunk

git-svn-id: https://lab.mitty.jp/svn/lab/trunk@147 7d2118f6-f56c-43e7-95a2-4bb3031d96e7

  • Property mode set to 100644
File size: 9.8 KB
Line 
1// SoftEther UT-VPN SourceCode
2//
3// Copyright (C) 2004-2010 SoftEther Corporation.
4// Copyright (C) 2004-2010 University of Tsukuba, Japan.
5// Copyright (C) 2003-2010 Daiyuu Nobori.
6// All Rights Reserved.
7//
8// http://utvpn.tsukuba.ac.jp/
9//
10// This program is free software; you can redistribute it and/or
11// modify it under the terms of the GNU General Public License
12// version 2 as published by the Free Software Foundation.
13//
14// This program is distributed in the hope that it will be useful,
15// but WITHOUT ANY WARRANTY; without even the implied warranty of
16// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17// GNU General Public License for more details.
18//
19// You should have received a copy of the GNU General Public License version 2
20// along with this program; if not, write to the Free Software
21// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
22//
23// このファイルは GPL バージョン 2 ライセンスで公開されています。
24// 誰でもこのファイルの内容を複製、改変したり、改変したバージョンを再配布
25// することができます。ただし、原著作物を改変した場合は、原著作物の著作権表示
26// を除去することはできません。改変した著作物を配布する場合は、改変実施者の
27// 著作権表示を原著作物の著作権表示に付随して記載するようにしてください。
28//
29// この SoftEther UT-VPN オープンソース・プロジェクトは、日本国の
30// ソフトイーサ株式会社 (SoftEther Corporation, http://www.softether.co.jp/ )
31// および筑波大学 (University of Tsukuba, http://www.tsukuba.ac.jp/ ) によって
32// ホストされています。
33// 本プログラムの配布者は、本プログラムを、業としての利用以外のため、
34// および、試験または研究のために利用が行われることを想定して配布
35// しています。
36// SoftEther UT-VPN プロジェクトの Web サイトは http://utvpn.tsukuba.ac.jp/ に
37// あります。
38// 本ソフトウェアの不具合の修正、機能改良、セキュリティホールの修復などのコード
39// の改変を行った場合で、その成果物を SoftEther UT-VPN プロジェクトに提出して
40// いただける場合は、 http://utvpn.tsukuba.ac.jp/ までソースコードを送付して
41// ください。SoftEther UT-VPN プロジェクトの本体リリースまたはブランチリリース
42// に組み込みさせていただきます。
43//
44// GPL に基づいて原著作物が提供される本ソフトウェアの改良版を配布、販売する
45// 場合は、そのソースコードを GPL に基づいて誰にでも開示する義務が生じます。
46//
47// 本ソフトウェアに関連する著作権、特許権、商標権はソフトイーサ株式会社
48// (SoftEther Corporation) およびその他の著作権保持者が保有しています。
49// ソフトイーサ株式会社等はこれらの権利を放棄していません。本ソフトウェアの
50// 二次著作物を配布、販売する場合は、これらの権利を侵害しないようにご注意
51// ください。
52//
53// お願い: どのような通信ソフトウェアにも通常は必ず未発見の
54// セキュリティホールが潜んでいます。本ソースコードをご覧いただいた結果、
55// UT-VPN にセキュリティホールを発見された場合は、当該セキュリティホールの
56// 情報を不特定多数に開示される前に、必ず、ソフトイーサ株式会社
57// および脆弱性情報の届出を受け付ける公的機関まで通報いただき、
58// 公益保護にご協力いただきますようお願い申し上げます。
59//
60// ソフトイーサ株式会社は、当該セキュリティホールについて迅速に対処を
61// 行い、UT-VPN および UT-VPN に関連するソフトウェアのユーザー・顧客
62// を保護するための努力を行います。
63//
64// ソフトイーサへの届出先: http://www.softether.co.jp/jp/contact/
65// 日本国内の脆弱性情報届出受付公的機関:
66//         独立行政法人 情報処理推進機構
67//         http://www.ipa.go.jp/security/vuln/report/
68//
69// 上記各事項について不明な点は、ソフトイーサ株式会社までご連絡ください。
70// 連絡先: http://www.softether.co.jp/jp/contact/
71
72// -----------------------------------------------
73// [ChangeLog]
74// 2010.05.20
75//  新規リリース by SoftEther
76// -----------------------------------------------
77
78// Remote.c
79// リモートプロシージャーコール
80
81#include "CedarPch.h"
82
83// RPC の終了
84void EndRpc(RPC *rpc)
85{
86    RpcFree(rpc);
87}
88
89// RPC の解放
90void RpcFree(RPC *rpc)
91{
92    // 引数チェック
93    if (rpc == NULL)
94    {
95        return;
96    }
97
98    Disconnect(rpc->Sock);
99    ReleaseSock(rpc->Sock);
100
101    DeleteLock(rpc->Lock);
102
103    Free(rpc);
104}
105
106// エラーの取得
107UINT RpcGetError(PACK *p)
108{
109    // 引数チェック
110    if (p == NULL)
111    {
112        return ERR_DISCONNECTED;
113    }
114
115    return PackGetInt(p, "error_code");
116}
117
118// エラーチェック
119bool RpcIsOk(PACK *p)
120{
121    // 引数チェック
122    if (p == NULL)
123    {
124        return false;
125    }
126
127    if (PackGetInt(p, "error") == 0)
128    {
129        return true;
130    }
131    else
132    {
133        return false;
134    }
135}
136
137// エラーコード設定
138void RpcError(PACK *p, UINT err)
139{
140    // 引数チェック
141    if (p == NULL)
142    {
143        return;
144    }
145
146    PackAddInt(p, "error", 1);
147    PackAddInt(p, "error_code", err);
148}
149
150// RPC ディスパッチャを起動する
151PACK *CallRpcDispatcher(RPC *r, PACK *p)
152{
153    char func_name[MAX_SIZE];
154    // 引数チェック
155    if (r == NULL || p == NULL)
156    {
157        return NULL;
158    }
159
160    if (PackGetStr(p, "function_name", func_name, sizeof(func_name)) == false)
161    {
162        return NULL;
163    }
164
165    return r->Dispatch(r, func_name, p);
166}
167
168// 次の RPC 呼び出しを待機する
169bool RpcRecvNextCall(RPC *r)
170{
171    UINT size;
172    void *tmp;
173    SOCK *s;
174    BUF *b;
175    PACK *p;
176    PACK *ret;
177    // 引数チェック
178    if (r == NULL)
179    {
180        return false;
181    }
182
183    s = r->Sock;
184
185    if (RecvAll(s, &size, sizeof(UINT), s->SecureMode) == false)
186    {
187        return false;
188    }
189
190    size = Endian32(size);
191
192    if (size > MAX_PACK_SIZE)
193    {
194        return false;
195    }
196
197    tmp = MallocEx(size, true);
198
199    if (RecvAll(s, tmp, size, s->SecureMode) == false)
200    {
201        Free(tmp);
202        return false;
203    }
204
205    b = NewBuf();
206    WriteBuf(b, tmp, size);
207    SeekBuf(b, 0, 0);
208    Free(tmp);
209
210    p = BufToPack(b);
211    FreeBuf(b);
212
213    if (p == NULL)
214    {
215        return false;
216    }
217
218    ret = CallRpcDispatcher(r, p);
219    FreePack(p);
220
221    if (ret == NULL)
222    {
223        ret = PackError(ERR_NOT_SUPPORTED);
224    }
225
226    b = PackToBuf(ret);
227    FreePack(ret);
228
229    size = Endian32(b->Size);
230    SendAdd(s, &size, sizeof(UINT));
231    SendAdd(s, b->Buf, b->Size);
232
233    if (SendNow(s, s->SecureMode) == false)
234    {
235        FreeBuf(b);
236        return false;
237    }
238
239    FreeBuf(b);
240
241    return true;
242}
243
244// RPC サーバー動作
245void RpcServer(RPC *r)
246{
247    SOCK *s;
248    // 引数チェック
249    if (r == NULL)
250    {
251        return;
252    }
253
254    s = r->Sock;
255
256    while (true)
257    {
258        // 次の RPC 呼び出しを待機する
259        if (RpcRecvNextCall(r) == false)
260        {
261            // 通信エラー
262            break;
263        }
264    }
265}
266
267// RPC 呼び出し
268PACK *RpcCall(RPC *r, char *function_name, PACK *p)
269{
270    PACK *ret;
271    UINT num_retry = 0;
272    UINT err = 0;
273    // 引数チェック
274    if (r == NULL || function_name == NULL)
275    {
276        return NULL;
277    }
278
279//  Debug("RpcCall: %s\n", function_name);
280
281    Lock(r->Lock);
282    {
283        if (p == NULL)
284        {
285            p = NewPack();
286        }
287
288        PackAddStr(p, "function_name", function_name);
289
290RETRY:
291        err = 0;
292        ret = RpcCallInternal(r, p);
293
294        if (ret == NULL)
295        {
296            if (r->IsVpnServer && r->Sock != NULL)
297            {
298                if (num_retry < 1)
299                {
300                    num_retry++;
301
302                    // VPN Server に RPC 再接続を試みる
303                    err = AdminReconnect(r);
304
305                    if (err == ERR_NO_ERROR)
306                    {
307                        goto RETRY;
308                    }
309                }
310            }
311        }
312
313        FreePack(p);
314
315        if (ret == NULL)
316        {
317            if (err == 0)
318            {
319                err = ERR_DISCONNECTED;
320            }
321
322            ret = PackError(err);
323            PackAddInt(ret, "error_code", err);
324        }
325    }
326    Unlock(r->Lock);
327
328    return ret;
329}
330
331// RPC 内部呼び出し
332PACK *RpcCallInternal(RPC *r, PACK *p)
333{
334    BUF *b;
335    UINT size;
336    PACK *ret;
337    void *tmp;
338    // 引数チェック
339    if (r == NULL || p == NULL)
340    {
341        return NULL;
342    }
343
344    if (r->Sock == NULL)
345    {
346        return NULL;
347    }
348
349    b = PackToBuf(p);
350
351    size = Endian32(b->Size);
352    SendAdd(r->Sock, &size, sizeof(UINT));
353    SendAdd(r->Sock, b->Buf, b->Size);
354    FreeBuf(b);
355
356    if (SendNow(r->Sock, r->Sock->SecureMode) == false)
357    {
358        return NULL;
359    }
360
361    if (RecvAll(r->Sock, &size, sizeof(UINT), r->Sock->SecureMode) == false)
362    {
363        return NULL;
364    }
365
366    size = Endian32(size);
367    if (size > MAX_PACK_SIZE)
368    {
369        return NULL;
370    }
371
372    tmp = MallocEx(size, true);
373    if (RecvAll(r->Sock, tmp, size, r->Sock->SecureMode) == false)
374    {
375        Free(tmp);
376        return NULL;
377    }
378
379    b = NewBuf();
380    WriteBuf(b, tmp, size);
381    SeekBuf(b, 0, 0);
382    Free(tmp);
383
384    ret = BufToPack(b);
385    if (ret == NULL)
386    {
387        FreeBuf(b);
388        return NULL;
389    }
390
391    FreeBuf(b);
392
393    return ret;
394}
395
396// RPC サーバーの開始
397RPC *StartRpcServer(SOCK *s, RPC_DISPATCHER *dispatch, void *param)
398{
399    RPC *r;
400    // 引数チェック
401    if (s == NULL)
402    {
403        return NULL;
404    }
405
406    r = ZeroMallocEx(sizeof(RPC), true);
407    r->Sock = s;
408    r->Param = param;
409    r->Lock = NewLock();
410    AddRef(s->ref);
411
412    r->ServerMode = true;
413    r->Dispatch = dispatch;
414
415    // 名前の生成
416    Format(r->Name, sizeof(r->Name), "RPC-%u", s->socket);
417
418    return r;
419}
420
421// RPC クライアントの開始
422RPC *StartRpcClient(SOCK *s, void *param)
423{
424    RPC *r;
425    // 引数チェック
426    if (s == NULL)
427    {
428        return NULL;
429    }
430
431    r = ZeroMalloc(sizeof(RPC));
432    r->Sock = s;
433    r->Param = param;
434    r->Lock = NewLock();
435    AddRef(s->ref);
436
437    r->ServerMode = false;
438
439    return r;
440}
441
Note: See TracBrowser for help on using the repository browser.