OAuthProxyAuthenticator.php 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. <?php
  2. namespace Base\OAuthClientBundle\Security;
  3. use Symfony\Component\HttpFoundation\Request;
  4. use Symfony\Component\HttpFoundation\Response;
  5. use Symfony\Component\Security\Core\Authentication\Token\PreAuthenticatedToken;
  6. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  7. use Symfony\Component\Security\Core\Exception\AuthenticationException;
  8. use Symfony\Component\Security\Core\Exception\CustomUserMessageAuthenticationException;
  9. use Symfony\Component\Security\Core\Exception\BadCredentialsException;
  10. use Symfony\Component\Security\Core\User\UserProviderInterface;
  11. use Symfony\Component\Security\Http\Authentication\SimplePreAuthenticatorInterface;
  12. use Symfony\Component\Security\Http\Authentication\AuthenticationFailureHandlerInterface;
  13. use Buzz\Listener\BasicAuthListener;
  14. use Buzz\Message;
  15. use HWI\Bundle\OAuthBundle\OAuth\Response\UserResponseInterface;
  16. use Symfony\Component\Security\Core\Authentication\Token\AnonymousToken;
  17. class OAuthProxyAuthenticator implements SimplePreAuthenticatorInterface, AuthenticationFailureHandlerInterface
  18. {
  19. public function __construct($client_id, $client_secret, $access_token_url, $user_info_url)
  20. {
  21. $this->client_id = $client_id;
  22. $this->client_secret = $client_secret;
  23. $this->access_token_url = $access_token_url;
  24. $this->user_info_url = $user_info_url;
  25. }
  26. public function createToken(Request $request, $providerKey)
  27. {
  28. if($request->headers->has("php-auth-user") and $request->headers->has("php-auth-pw"))
  29. return new PreAuthenticatedToken($request->headers->get("php-auth-user"), $request->headers->get("php-auth-pw"), $providerKey);
  30. return new AnonymousToken("anon.", "anon.");
  31. }
  32. public function supportsToken(TokenInterface $token, $providerKey)
  33. {
  34. return $token instanceof PreAuthenticatedToken && $token->getProviderKey() === $providerKey;
  35. }
  36. public function authenticateToken(TokenInterface $token, UserProviderInterface $userProvider, $providerKey)
  37. {
  38. $password = $token->getCredentials();
  39. $username = $token->getUsername();
  40. $token = @json_decode(file_get_contents("/tmp/.".base64_encode($username. ":" . $password)), true);
  41. if(!isset($token["access_token"])){
  42. $browser = new \Buzz\Browser();
  43. $listener = new BasicAuthListener($this->client_id, $this->client_secret);
  44. $browser->addListener($listener);
  45. $body = ['grant_type' => 'password',
  46. 'username' => $username,
  47. 'password' => $password,
  48. ];
  49. $response = $browser->post($this->access_token_url, ['Content-Type' => 'application/x-www-form-urlencoded'], http_build_query($body));
  50. $token = json_decode($response->getContent(), true);
  51. if($token['expires_in'])
  52. $token["expires_at"] = time() + $token['expires_in'];
  53. else
  54. $token["expires_at"] = time() + 3600;
  55. file_put_contents("/tmp/.".base64_encode($username. ":" . $password), json_encode($token));
  56. }
  57. if(isset($token["expires_at"]) and $token["expires_at"] >= time()){
  58. $browser = new \Buzz\Browser();
  59. $listener = new BasicAuthListener($this->client_id, $this->client_secret);
  60. $browser->addListener($listener);
  61. $body = ['grant_type' => 'refresh_token',
  62. 'refresh_token' => $token['refresh_token']
  63. ];
  64. $response = $browser->post($this->access_token_url, ['Content-Type' => 'application/x-www-form-urlencoded'], http_build_query($body));
  65. $token = json_decode($response->getContent(), true);
  66. if($token['expires_in'])
  67. $token["expires_at"] = time() + $token['expires_in'];
  68. else
  69. $token["expires_at"] = time() + 3600;
  70. file_put_contents("/tmp/.".base64_encode($username. ":" . $password), json_encode($token));
  71. }
  72. if(!isset($token["user_info"])){
  73. $oauth_headers = [
  74. "Authorization" => ucfirst($token["token_type"])." ".$token["access_token"],
  75. ];
  76. $browser = new \Buzz\Browser();
  77. $response = $browser->get($this->user_info_url, $oauth_headers);
  78. $auth_info = json_decode($response->getContent(), true);
  79. $token["user_info"] = $auth_info;
  80. file_put_contents("/tmp/.".base64_encode($username. ":" . $password), json_encode($token));
  81. }else{
  82. $auth_info = $token["user_info"];
  83. }
  84. $user = $userProvider->loadUserByUsername($auth_info["username"]);
  85. $user->setRoles($auth_info["roles"]);
  86. $user->setTenancyCurrent($auth_info["tenancyCurrent"]);
  87. return new PreAuthenticatedToken($user, array(), $providerKey, $user->getRoles());
  88. }
  89. public function onAuthenticationFailure(Request $request, AuthenticationException $exception)
  90. {
  91. return new Response(
  92. // this contains information about *why* authentication failed
  93. // use it, or return your own message
  94. strtr($exception->getMessageKey(), $exception->getMessageData()), 401);
  95. }
  96. }