Wt examples  4.0.2
Session.C
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2011 Emweb bvba, Heverlee, Belgium.
3  *
4  * See the LICENSE file for terms of use.
5  */
6 
7 #include "Session.h"
8 
9 #include "Wt/Auth/AuthService.h"
10 #include "Wt/Auth/HashFunction.h"
11 #include "Wt/Auth/PasswordService.h"
12 #include "Wt/Auth/PasswordStrengthValidator.h"
13 #include "Wt/Auth/PasswordVerifier.h"
14 #include "Wt/Auth/GoogleService.h"
15 #include "Wt/Auth/Dbo/AuthInfo.h"
16 #include "Wt/Auth/Dbo/UserDatabase.h"
17 
18 #include <Wt/WApplication.h>
19 #include <Wt/WLogger.h>
20 
21 #ifndef WT_WIN32
22 #include <unistd.h>
23 #endif
24 
25 #if !defined(WT_WIN32) && !defined(__CYGWIN__) && !defined(ANDROID)
26 #define HAVE_CRYPT
27 #include <crypt.h>
28 #endif
29 
30 namespace dbo = Wt::Dbo;
31 
32 namespace {
33 
34 #ifdef HAVE_CRYPT
35 class UnixCryptHashFunction : public Auth::HashFunction
36  {
37  public:
38  virtual std::string compute(const std::string& msg,
39  const std::string& salt) const
40  {
41  std::string md5Salt = "$1$" + salt;
42  return crypt(msg.c_str(), md5Salt.c_str());
43  }
44 
45  virtual bool verify(const std::string& msg,
46  const std::string& salt,
47  const std::string& hash) const
48  {
49  return crypt(msg.c_str(), hash.c_str()) == hash;
50  }
51 
52  virtual std::string name () const {
53  return "crypt";
54  }
55  };
56 #endif // HAVE_CRYPT
57 
58  class MyOAuth : public std::vector<const Auth::OAuthService *>
59  {
60  public:
61  ~MyOAuth()
62  {
63  for (unsigned i = 0; i < size(); ++i)
64  delete (*this)[i];
65  }
66  };
67 
68  Auth::AuthService myAuthService;
69  Auth::PasswordService myPasswordService(myAuthService);
70  MyOAuth myOAuthServices;
71 }
72 
74 {
75  myAuthService.setAuthTokensEnabled(true, "hangmancookie");
76  myAuthService.setEmailVerificationEnabled(true);
77 
78  std::unique_ptr<Auth::PasswordVerifier> verifier
79  = cpp14::make_unique<Auth::PasswordVerifier>();
80  verifier->addHashFunction(cpp14::make_unique<Auth::BCryptHashFunction>(7));
81 
82 #ifdef HAVE_CRYPT
83  // We want to still support users registered in the pre - Wt::Auth
84  // version of the hangman example
85  verifier->addHashFunction(cpp14::make_unique<UnixCryptHashFunction>());
86 #endif
87 
88  myPasswordService.setVerifier(std::move(verifier));
89  myPasswordService.setStrengthValidator(cpp14::make_unique<Auth::PasswordStrengthValidator>());
90  myPasswordService.setAttemptThrottlingEnabled(true);
91 
92  if (Auth::GoogleService::configured())
93  myOAuthServices.push_back(new Auth::GoogleService(myAuthService));
94 }
95 
97 {
98  auto sqlite3 = cpp14::make_unique<Dbo::backend::Sqlite3>(WApplication::instance()->appRoot() + "hangman.db");
99  sqlite3->setProperty("show-queries", "true");
100  session_.setConnection(std::move(sqlite3));
101 
102  session_.mapClass<User>("user");
103  session_.mapClass<AuthInfo>("auth_info");
104  session_.mapClass<AuthInfo::AuthIdentityType>("auth_identity");
105  session_.mapClass<AuthInfo::AuthTokenType>("auth_token");
106 
107  users_ = cpp14::make_unique<UserDatabase>(session_);
108 
109  dbo::Transaction transaction(session_);
110  try {
111  session_.createTables();
112 
113  /*
114  * Add a default guest/guest account
115  */
116  Auth::User guestUser = users_->registerNew();
117  guestUser.addIdentity(Auth::Identity::LoginName, "guest");
118  myPasswordService.updatePassword(guestUser, "guest");
119 
120  log("info") << "Database created";
121  } catch (...) {
122  log("info") << "Using existing database";
123  }
124 
125  transaction.commit();
126 }
127 
129 {
130 }
131 
132 dbo::ptr<User> Session::user() const
133 {
134  if (login_.loggedIn()) {
135  dbo::ptr<AuthInfo> authInfo = users_->find(login_.user());
136  dbo::ptr<User> user = authInfo->user();
137 
138  if (!user) {
139  user = session_.add(Wt::cpp14::make_unique<User>());
140  authInfo.modify()->setUser(user);
141  }
142 
143  return user;
144  } else
145  return dbo::ptr<User>();
146 }
147 
148 std::string Session::userName() const
149 {
150  if (login_.loggedIn())
151  return login_.user().identity(Auth::Identity::LoginName).toUTF8();
152  else
153  return std::string();
154 }
155 
157 {
158  dbo::Transaction transaction(session_);
159 
160  dbo::ptr<User> u = user();
161  if (u) {
162  u.modify()->score += s;
163  ++u.modify()->gamesPlayed;
164  u.modify()->lastGame = WDateTime::currentDateTime();
165  }
166 
167  transaction.commit();
168 }
169 
170 std::vector<User> Session::topUsers(int limit)
171 {
172  dbo::Transaction transaction(session_);
173 
174  Users top = session_.find<User>().orderBy("score desc").limit(limit);
175 
176  std::vector<User> result;
177  for (Users::const_iterator i = top.begin(); i != top.end(); ++i) {
178  dbo::ptr<User> user = *i;
179  result.push_back(*user);
180 
181  dbo::ptr<AuthInfo> auth = *user->authInfos.begin();
182  std::string name = auth->identity(Auth::Identity::LoginName).toUTF8();
183 
184  result.back().name = name;
185  }
186 
187  transaction.commit();
188 
189  return result;
190 }
191 
193 {
194  dbo::Transaction transaction(session_);
195 
196  dbo::ptr<User> u = user();
197  int ranking = -1;
198 
199  if (u)
200  ranking = session_.query<int>("select distinct count(score) from user")
201  .where("score > ?").bind(u->score);
202 
203  transaction.commit();
204 
205  return ranking + 1;
206 }
207 
208 Auth::AbstractUserDatabase& Session::users()
209 {
210  return *users_;
211 }
212 
213 const Auth::AuthService& Session::auth()
214 {
215  return myAuthService;
216 }
217 
218 const Auth::AbstractPasswordService& Session::passwordAuth()
219 {
220  return myPasswordService;
221 }
222 
223 const std::vector<const Auth::OAuthService *>& Session::oAuth()
224 {
225  return myOAuthServices;
226 }
Auth::Dbo::AuthInfo< User > AuthInfo
Definition: User.h:22
dbo::collection< dbo::ptr< User > > Users
Definition: User.h:24
std::string userName() const
Definition: Session.C:148
Auth::AbstractUserDatabase & users()
Definition: Session.C:208
Session()
Definition: Session.C:96
int findRanking()
Definition: Session.C:192
Definition: User.h:26
static const Auth::AuthService & auth()
Definition: Session.C:213
dbo::collection< dbo::ptr< AuthInfo > > authInfos
Definition: User.h:35
static const std::vector< const Auth::OAuthService * > & oAuth()
Definition: Session.C:223
static void configureAuth()
Definition: Session.C:73
Dbo::ptr< User > user() const
Definition: Session.C:132
std::vector< User > topUsers(int limit)
Definition: Session.C:170
~Session()
Definition: Session.C:128
static const Auth::AbstractPasswordService & passwordAuth()
Definition: Session.C:218
void addToScore(int s)
Definition: Session.C:156

Generated on Wed May 30 2018 for the C++ Web Toolkit (Wt) by doxygen 1.8.12