2 changed files with 142 additions and 4 deletions
@ -0,0 +1,97 @@ |
|||||||
|
/* |
||||||
|
* Spreed WebRTC. |
||||||
|
* Copyright (C) 2016 struktur AG |
||||||
|
* |
||||||
|
* This file is part of Spreed WebRTC. |
||||||
|
* |
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU Affero General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU Affero General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU Affero General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
package channelling |
||||||
|
|
||||||
|
import ( |
||||||
|
"crypto/rand" |
||||||
|
"encoding/base64" |
||||||
|
"testing" |
||||||
|
) |
||||||
|
|
||||||
|
func getRandom(n int) ([]byte, error) { |
||||||
|
result := make([]byte, n) |
||||||
|
if _, err := rand.Read(result); err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
return result, nil |
||||||
|
} |
||||||
|
|
||||||
|
func Test_ReverseBase64(t *testing.T) { |
||||||
|
for i := 0; i < 1000; i++ { |
||||||
|
data, err := getRandom(64) |
||||||
|
if err != nil { |
||||||
|
t.Errorf("Could not get random data: %v", err) |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
s := base64.URLEncoding.EncodeToString(data) |
||||||
|
reversed, err := reverseBase64String(s) |
||||||
|
if err != nil { |
||||||
|
t.Errorf("Could not reverse %s: %v", s, err) |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
if s == reversed { |
||||||
|
t.Errorf("Reversing should be different for %s", s) |
||||||
|
} |
||||||
|
|
||||||
|
original, err := reverseBase64String(reversed) |
||||||
|
if err != nil { |
||||||
|
t.Errorf("Could not reverse back %s: %v", reversed, err) |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
if s != original { |
||||||
|
t.Errorf("Reversing back should have restored %s from %s but got %s", s, reversed, original) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func Test_Sessions(t *testing.T) { |
||||||
|
sessionSecret, err := getRandom(64) |
||||||
|
if err != nil { |
||||||
|
t.Fatalf("Could not create session secret: %v", err) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
encryptionSecret, err := getRandom(32) |
||||||
|
if err != nil { |
||||||
|
t.Fatalf("Could not create encryption secret: %v", err) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
tickets := NewTickets(sessionSecret, encryptionSecret, "test") |
||||||
|
silentOutput = true |
||||||
|
for i := 0; i < 1000; i++ { |
||||||
|
st := tickets.DecodeSessionToken("") |
||||||
|
if st == nil { |
||||||
|
t.Error("Could not create session") |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
if !tickets.ValidateSession(st.Id, st.Sid) { |
||||||
|
t.Errorf("Session is invalid: %v", st) |
||||||
|
continue |
||||||
|
} |
||||||
|
} |
||||||
|
silentOutput = false |
||||||
|
} |
Loading…
Reference in new issue