sts.js 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180
  1. // 临时密钥服务例子
  2. var bodyParser = require('body-parser');
  3. var STS = require('qcloud-cos-sts');
  4. var express = require('express');
  5. var crypto = require('crypto');
  6. var pathLib = require('path');
  7. var fs = require('fs');
  8. // 配置参数
  9. var config = {
  10. secretId: process.env.SecretId,
  11. secretKey: process.env.SecretKey,
  12. proxy: process.env.Proxy,
  13. durationSeconds: 1800,
  14. bucket: process.env.Bucket,
  15. region: process.env.Region,
  16. // 允许操作(上传)的对象前缀,可以根据自己网站的用户登录态判断允许上传的目录,例子: user1/* 或者 * 或者a.jpg
  17. // 请注意当使用 * 时,可能存在安全风险,详情请参阅:https://cloud.tencent.com/document/product/436/40265
  18. allowPrefix: '_ALLOW_DIR_/*',
  19. // 密钥的权限列表
  20. allowActions: [
  21. // 所有 action 请看文档 https://cloud.tencent.com/document/product/436/31923
  22. // 简单上传
  23. 'name/cos:PutObject',
  24. 'name/cos:PostObject',
  25. // 分片上传
  26. 'name/cos:InitiateMultipartUpload',
  27. 'name/cos:ListMultipartUploads',
  28. 'name/cos:ListParts',
  29. 'name/cos:UploadPart',
  30. 'name/cos:CompleteMultipartUpload'
  31. ],
  32. };
  33. // 创建临时密钥服务和用于调试的静态服务
  34. var app = express();
  35. var replaceBucketRegion = (filePath) => {
  36. return (req, res, next) => {
  37. var content = fs.readFileSync(filePath).toString()
  38. .replace(/(var config = {\r?\n *Bucket: ')test-1250000000(',\r?\n *Region: ')ap-guangzhou(',?\r?\n};?)/,
  39. '$1' + config.bucket + '$2' + config.region +'$3');
  40. res.header('Content-Type', 'application/javascript');
  41. res.send(content);
  42. };
  43. };
  44. app.use('/demo/demo.js', replaceBucketRegion(pathLib.resolve(__dirname, '../demo/demo.js')));
  45. app.use('/test/test.js', replaceBucketRegion(pathLib.resolve(__dirname, '../test/test.js')));
  46. app.use('/dist/', express.static(pathLib.resolve(__dirname, '../dist')));
  47. app.use('/demo/', express.static(pathLib.resolve(__dirname, '../demo')));
  48. app.use('/test/', express.static(pathLib.resolve(__dirname, '../test')));
  49. app.all('/', (req, res, next) => res.redirect('/demo/'));
  50. app.use(bodyParser.json());
  51. // 格式一:临时密钥接口
  52. app.all('/sts', function (req, res, next) {
  53. // TODO 这里根据自己业务需要做好放行判断
  54. if (config.allowPrefix === '_ALLOW_DIR_/*') {
  55. res.send({error: '请修改 allowPrefix 配置项,指定允许上传的路径前缀'});
  56. return;
  57. }
  58. // 获取临时密钥
  59. var LongBucketName = config.bucket;
  60. var ShortBucketName = LongBucketName.substr(0, LongBucketName.lastIndexOf('-'));
  61. var AppId = LongBucketName.substr(LongBucketName.lastIndexOf('-') + 1);
  62. var policy = {
  63. 'version': '2.0',
  64. 'statement': [{
  65. 'action': config.allowActions,
  66. 'effect': 'allow',
  67. 'resource': [
  68. 'qcs::cos:' + config.region + ':uid/' + AppId + ':prefix//' + AppId + '/' + ShortBucketName + '/' + config.allowPrefix,
  69. ],
  70. }],
  71. };
  72. var startTime = Math.round(Date.now() / 1000);
  73. STS.getCredential({
  74. secretId: config.secretId,
  75. secretKey: config.secretKey,
  76. proxy: config.proxy,
  77. region: config.region,
  78. durationSeconds: config.durationSeconds,
  79. policy: policy,
  80. }, function (err, tempKeys) {
  81. if (tempKeys) tempKeys.startTime = startTime;
  82. res.send(err || tempKeys);
  83. });
  84. });
  85. // // 格式二:临时密钥接口,支持细粒度权限控制
  86. // // 判断是否允许获取密钥
  87. // var allowScope = function (scope) {
  88. // var allow = (scope || []).every(function (item) {
  89. // return config.allowActions.includes(item.action) &&
  90. // item.bucket === config.bucket &&
  91. // item.region === config.region &&
  92. // (item.prefix || '').startsWith(config.allowPrefix);
  93. // });
  94. // return allow;
  95. // };
  96. // app.all('/sts-scope', function (req, res, next) {
  97. // var scope = req.body;
  98. //
  99. // // TODO 这里根据自己业务需要做好放行判断
  100. // if (config.allowPrefix === '_ALLOW_DIR_/*') {
  101. // res.send({error: '请修改 allowPrefix 配置项,指定允许上传的路径前缀'});
  102. // return;
  103. // }
  104. // // TODO 这里可以判断 scope 细粒度控制权限
  105. // if (!scope || !scope.length || !allowScope(scope)) return res.send({error: 'deny'});
  106. //
  107. // // 获取临时密钥
  108. // var policy = STS.getPolicy(scope);
  109. // var startTime = Math.round(Date.now() / 1000);
  110. // STS.getCredential({
  111. // secretId: config.secretId,
  112. // secretKey: config.secretKey,
  113. // proxy: config.proxy,
  114. // durationSeconds: config.durationSeconds,
  115. // policy: policy,
  116. // }, function (err, tempKeys) {
  117. // if (tempKeys) tempKeys.startTime = startTime;
  118. // res.send(err || tempKeys);
  119. // });
  120. // });
  121. //
  122. // // 用于 PostObject 签名保护
  123. // app.all('/post-policy', function (req, res, next) {
  124. // var query = req.query;
  125. // var now = Math.round(Date.now() / 1000);
  126. // var exp = now + 900;
  127. // var qKeyTime = now + ';' + exp;
  128. // var qSignAlgorithm = 'sha1';
  129. // var policy = JSON.stringify({
  130. // 'expiration': new Date(exp * 1000).toISOString(),
  131. // 'conditions': [
  132. // // {'acl': query.ACL},
  133. // // ['starts-with', '$Content-Type', 'image/'],
  134. // // ['starts-with', '$success_action_redirect', redirectUrl],
  135. // // ['eq', '$x-cos-server-side-encryption', 'AES256'],
  136. // {'q-sign-algorithm': qSignAlgorithm},
  137. // {'q-ak': config.secretId},
  138. // {'q-sign-time': qKeyTime},
  139. // {'bucket': config.bucket},
  140. // {'key': query.key},
  141. // ]
  142. // });
  143. //
  144. // // 签名算法说明文档:https://www.qcloud.com/document/product/436/7778
  145. // // 步骤一:生成 SignKey
  146. // var signKey = crypto.createHmac('sha1', config.secretKey).update(qKeyTime).digest('hex');
  147. //
  148. // // 步骤二:生成 StringToSign
  149. // var stringToSign = crypto.createHash('sha1').update(policy).digest('hex');
  150. //
  151. // // 步骤三:生成 Signature
  152. // var qSignature = crypto.createHmac('sha1', signKey).update(stringToSign).digest('hex');
  153. //
  154. // console.log(policy);
  155. // res.send({
  156. // policyObj: JSON.parse(policy),
  157. // policy: Buffer.from(policy).toString('base64'),
  158. // qSignAlgorithm: qSignAlgorithm,
  159. // qAk: config.secretId,
  160. // qKeyTime: qKeyTime,
  161. // qSignature: qSignature,
  162. // // securityToken: securityToken, // 如果使用临时密钥,要返回在这个资源 sessionToken 的值
  163. // });
  164. // });
  165. //
  166. app.all('*', function (req, res, next) {
  167. res.send({code: -1, message: '404 Not Found'});
  168. });
  169. // 启动签名服务
  170. app.listen(3000);
  171. console.log('app is listening at http://127.0.0.1:3000');