Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
李少辉-开发者
gitlab-foss
提交
10b8fd71
G
gitlab-foss
项目概览
李少辉-开发者
/
gitlab-foss
通知
15
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
G
gitlab-foss
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
10b8fd71
编写于
11月 14, 2018
作者:
G
Grzegorz Bizon
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Refactor token authenticatable encrypted strategy
上级
10ea7539
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
20 addition
and
13 deletion
+20
-13
app/models/concerns/token_authenticatable_strategies/base.rb
app/models/concerns/token_authenticatable_strategies/base.rb
+6
-0
app/models/concerns/token_authenticatable_strategies/encrypted.rb
...ls/concerns/token_authenticatable_strategies/encrypted.rb
+14
-13
未找到文件。
app/models/concerns/token_authenticatable_strategies/base.rb
浏览文件 @
10b8fd71
# frozen_string_literal: true
module
TokenAuthenticatableStrategies
attr_reader
:klass
,
:token_field
,
:options
class
Base
def
initialize
(
klass
,
token_field
,
options
)
@klass
=
klass
...
...
@@ -36,6 +38,10 @@ module TokenAuthenticatableStrategies
instance
.
save!
if
Gitlab
::
Database
.
read_write?
end
def
fallback?
options
[
:fallback
]
==
true
end
protected
def
write_new_token
(
instance
)
...
...
app/models/concerns/token_authenticatable_strategies/encrypted.rb
浏览文件 @
10b8fd71
...
...
@@ -7,45 +7,46 @@ module TokenAuthenticatableStrategies
def
find_token_authenticatable
(
token
,
unscoped
=
false
)
return
unless
token
encrypted_value
=
Gitlab
::
CryptoHelper
.
aes256_gcm_encrypt
(
token
)
token_authenticatable
=
relation
(
unscoped
)
.
find_by
(
token_field_name
=>
Gitlab
::
CryptoHelper
.
aes256_gcm_encrypt
(
token
)
)
.
find_by
(
encrypted_field
=>
encrypted_value
)
if
@options
[
:fallback
]
token_authenticatable
||=
fallback_strategy
.
find_token_authenticatable
(
token
)
if
fallback?
token_authenticatable
||=
fallback_strategy
.
find_token_authenticatable
(
token
)
end
token_authenticatable
end
def
get_token
(
instance
)
raw_token
=
instance
.
read_attribute
(
token_field_name
)
raw_token
=
instance
.
read_attribute
(
encrypted_field
)
token
=
Gitlab
::
CryptoHelper
.
aes256_gcm_decrypt
(
raw_token
)
token
||=
fallback_strategy
.
get_token
(
instance
)
if
@options
[
:fallback
]
token
||=
fallback_strategy
.
get_token
(
instance
)
if
fallback?
end
def
set_token
(
instance
,
token
)
raise
ArgumentError
unless
token
raise
ArgumentError
unless
token
.
present?
instance
[
token_field_name
]
=
Gitlab
::
CryptoHelper
.
aes256_gcm_encrypt
(
token
)
# instance[@token_field] = nil if @options[:fallback] # TODO this seems wrong
instance
[
encrypted_field
]
=
Gitlab
::
CryptoHelper
.
aes256_gcm_encrypt
(
token
)
end
protected
def
fallback_strategy
@fallback_strategy
||=
TokenAuthenticatableStrategies
::
Insecure
.
new
(
@klass
,
@token_field
,
@
options
)
.
new
(
klass
,
token_field
,
options
)
end
def
token_set?
(
instance
)
raw_token
=
instance
.
read_attribute
(
token_field_name
)
raw_token
||=
instance
.
read_attribute
(
@token_field
)
if
@options
[
:fallback
]
raw_token
=
instance
.
read_attribute
(
encrypted_field
)
raw_token
||=
instance
.
read_attribute
(
token_field
)
if
fallback?
raw_token
.
present?
end
def
token_field_name
"
#{
@token_field
}
_encrypted"
def
encrypted_field
@encrypted_field
||=
"
#{
@token_field
}
_encrypted"
end
end
end
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录