modules/openid_provider/openid_provider.module

<?php
// $Id: openid_provider.module,v 1.2 2008/09/07 03:39:57 walkah Exp $

/**
 * OpenID 2.0 Provider implementation for Drupal.
 */

/**
 * Implementation of hook_perm().
 */
function openid_provider_perm() {
  return array('administer openid');
}

/**
 * Implementation of hook_menu().
 */
function openid_provider_menu() {
  $items['admin/settings/openid_provider'] = array(
    'title' => 'OpenID Provider',
    'description' => 'Configure settings for the OpenID Provider.',
    'page callback' => 'drupal_get_form',
    'page arguments' => array('openid_provider_admin_settings'),
    'access arguments' => array('administer openid'),
    'type' => MENU_NORMAL_ITEM,
  );
  
  $items['openid/provider'] = array(
    'title' => 'OpenID Login',
    'page callback' => 'openid_provider_endpoint',
    'access callback' => TRUE,
    'type' => MENU_CALLBACK,
    'file' => 'openid_provider.pages.inc'
  );
  
  $items['openid/provider/send'] = array(
    'title' => 'OpenID Login',
    'page callback' => 'openid_provider_send',
    'access callback' => TRUE,
    'type' => MENU_CALLBACK,
    'file' => 'openid_provider.pages.inc'
  );

  $items['openid/provider/continue'] = array(
    'title' => 'OpenID Login',
    'page callback' => 'openid_provider_continue',
    'access callback' => TRUE,
    'type' => MENU_CALLBACK,
    'file' => 'openid_provider.pages.inc'
  );
  
  $items['user/%user/openid_sites'] = array(
    'title' => 'OpenID Sites',
    'page callback' => 'openid_provider_sites',
    'page arguments' => array(1),
    'access callback' => 'openid_provider_sites_access',
    'access arguments' => array(1),
    'type' => MENU_LOCAL_TASK,
    'file' => 'openid_provider.pages.inc'
  );

  return $items;
}

/**
 * Implementation of hook_init()
 *
 * Add appropriate HTML headers for XRDS and Link discovery.
 */
function openid_provider_init() {
  // Not all OpenID clients may be smart enough to do XRDS.
  drupal_add_link(array('rel' => 'openid2.provider', 'href' => url('openid/provider', array('absolute' => TRUE))));
  drupal_add_link(array('rel' => 'openid.server', 'href' => url('openid/provider', array('absolute' => TRUE))));
}

function openid_provider_sites_access($account) {
  global $user;

  return ($account->uid == $user->uid) || user_access('administer openid');
}

/**
 * Implementation of hook_user.
 */
function openid_provider_user($op, &$edit, &$account, $category = NULL) {
  global $user;
  
  switch ($op) {
    case 'view':
      if ($user->uid == $account->uid) {
        $account->content['openid'] = array(
          '#title' => t('OpenID'),
          '#value' => t('You may login to other OpenID enabled sites using %url', array('%url' => url('user/' . $account->uid, array('absolute' => TRUE)))),
          '#class' => 'openid',
          '#weight' => 10
        );
      }
      break;
  }
}

/**
 * Settings form.
 */
function openid_provider_admin_settings() {
  $form = array();
  $form['openid_provider_assoc_expires_in'] = array(
    '#type' => 'textfield',
    '#title' => t('Associations expire in this many seconds'),
    '#default_value' => variable_get('openid_provider_assoc_expires_in', '3600'),
    '#size' => 10,
    '#maxlength' => 10);
  return system_settings_form($form);
}

/**
 * Return a XRDS for this server to discover it based on the root url
 */
function openid_provider_xrds($account = NULL) {
  module_load_include('inc', 'openid');
  
  if ($account) {
    $types = array(OPENID_NS_2_0 .'/signon');
  }
  else {
    $types = array(OPENID_NS_2_0 .'/server');
  }

  $data = array(
    'Type' => $types,
    'URI' => array(url('openid/provider', array('absolute'=> TRUE))),
  );
  if ($account->uid) {
    $data['LocalID'] = array(url('user/'. $account->uid, array('absolute' => TRUE)));
  }
  
  $xrds['openid_provider'] = array(
    'services' => array(
      array('priority' => 10,
            'data' => $data
      )
    )
  );

  return $xrds;
}

/**
 * Main OpenID Provider form
 */
function openid_provider_form(&$form_state, &$response = array(), $realm = NULL) {
  global $user;

  // Use form_state to store the $response and $realm values 
  if (count($response)) {
    $form_state['storage']['response'] = $response;
  }
  else {
    $response = $form_state['storage']['response'];
  }
  
  if ($realm) {
    $form_state['storage']['realm'] = $realm;
  }
  else {
    $realm = $form_state['storage']['realm'];
  }
  
  $form = array();
  
  $form['intro'] = array(
    '#type' => 'markup',
    '#value' => '<p>'. t('You are being logged into %site, would you like to continue?', array('%site' => $realm)) . '</p>'
  );
  
  $form['#action'] = url('openid/provider/send');

  $form['submit_once'] = array(
    '#type' => 'submit',
    '#value' => t('Yes; just this once'),
  );
  $form['submit_always'] = array(
    '#type' => 'submit',
    '#value' => t('Yes; always'),
    '#submit' => array('openid_provider_form_submit_always')
  );
  $form['cancel'] = array(
    '#type' => 'submit',
    '#value' => t('Cancel'),
    '#submit' => array('openid_provider_form_submit_cancel')
  );

  return $form;
}

/**
 * Once submit handler
 */
function openid_provider_form_submit(&$form, $form_state, $auto_release = FALSE) {
  global $user;

  module_load_include('inc', 'openid');
  module_load_include('inc', 'openid_provider');
  
  $response = _openid_provider_sign($form_state['storage']['response']);
  _openid_provider_rp_save($user->uid, $form_state['storage']['realm'], $auto_release);
  openid_redirect_http($response['openid.return_to'], $response);
}

/**
 * Always submit handler
 */
function openid_provider_form_submit_always(&$form, $form_state) {
  return openid_provider_form_submit($form, $form_state, TRUE);
}
 
/**
 * Cancel submit handler
 */
function openid_provider_form_submit_cancel(&$form, $form_state) {
  module_load_include('inc', 'openid_provider');
  module_load_include('inc', 'openid');
  
  $return_to = $form_state['values']['openid.return_to'];
  $response = openid_provider_cancel_authentication_response($form_state['openid.mode']);
  openid_redirect($return_to, $response);
}
1	<?php
2	// $Id: openid_provider.module,v 1.2 2008/09/07 03:39:57 walkah Exp $
3
4	/**
5	* OpenID 2.0 Provider implementation for Drupal.
6	*/
7
8	/**
9	* Implementation of hook_perm().
10	*/
11	function openid_provider_perm() {
12	return array('administer openid');
13	}
14
15	/**
16	* Implementation of hook_menu().
17	*/
18	function openid_provider_menu() {
19	$items['admin/settings/openid_provider'] = array(
20	'title' => 'OpenID Provider',
21	'description' => 'Configure settings for the OpenID Provider.',
22	'page callback' => 'drupal_get_form',
23	'page arguments' => array('openid_provider_admin_settings'),
24	'access arguments' => array('administer openid'),
25	'type' => MENU_NORMAL_ITEM,
26	);
27
28	$items['openid/provider'] = array(
29	'title' => 'OpenID Login',
30	'page callback' => 'openid_provider_endpoint',
31	'access callback' => TRUE,
32	'type' => MENU_CALLBACK,
33	'file' => 'openid_provider.pages.inc'
34	);
35
36	$items['openid/provider/send'] = array(
37	'title' => 'OpenID Login',
38	'page callback' => 'openid_provider_send',
39	'access callback' => TRUE,
40	'type' => MENU_CALLBACK,
41	'file' => 'openid_provider.pages.inc'
42	);
43
44	$items['openid/provider/continue'] = array(
45	'title' => 'OpenID Login',
46	'page callback' => 'openid_provider_continue',
47	'access callback' => TRUE,
48	'type' => MENU_CALLBACK,
49	'file' => 'openid_provider.pages.inc'
50	);
51
52	$items['user/%user/openid_sites'] = array(
53	'title' => 'OpenID Sites',
54	'page callback' => 'openid_provider_sites',
55	'page arguments' => array(1),
56	'access callback' => 'openid_provider_sites_access',
57	'access arguments' => array(1),
58	'type' => MENU_LOCAL_TASK,
59	'file' => 'openid_provider.pages.inc'
60	);
61
62	return $items;
63	}
64
65	/**
66	* Implementation of hook_init()
67	*
68	* Add appropriate HTML headers for XRDS and Link discovery.
69	*/
70	function openid_provider_init() {
71	// Not all OpenID clients may be smart enough to do XRDS.
72	drupal_add_link(array('rel' => 'openid2.provider', 'href' => url('openid/provider', array('absolute' => TRUE))));
73	drupal_add_link(array('rel' => 'openid.server', 'href' => url('openid/provider', array('absolute' => TRUE))));
74	}
75
76	function openid_provider_sites_access($account) {
77	global $user;
78
79	return ($account->uid == $user->uid) \|\| user_access('administer openid');
80	}
81
82	/**
83	* Implementation of hook_user.
84	*/
85	function openid_provider_user($op, &$edit, &$account, $category = NULL) {
86	global $user;
87
88	switch ($op) {
89	case 'view':
90	if ($user->uid == $account->uid) {
91	$account->content['openid'] = array(
92	'#title' => t('OpenID'),
93	'#value' => t('You may login to other OpenID enabled sites using %url', array('%url' => url('user/' . $account->uid, array('absolute' => TRUE)))),
94	'#class' => 'openid',
95	'#weight' => 10
96	);
97	}
98	break;
99	}
100	}
101
102	/**
103	* Settings form.
104	*/
105	function openid_provider_admin_settings() {
106	$form = array();
107	$form['openid_provider_assoc_expires_in'] = array(
108	'#type' => 'textfield',
109	'#title' => t('Associations expire in this many seconds'),
110	'#default_value' => variable_get('openid_provider_assoc_expires_in', '3600'),
111	'#size' => 10,
112	'#maxlength' => 10);
113	return system_settings_form($form);
114	}
115
116	/**
117	* Return a XRDS for this server to discover it based on the root url
118	*/
119	function openid_provider_xrds($account = NULL) {
120	module_load_include('inc', 'openid');
121
122	if ($account) {
123	$types = array(OPENID_NS_2_0 .'/signon');
124	}
125	else {
126	$types = array(OPENID_NS_2_0 .'/server');
127	}
128
129	$data = array(
130	'Type' => $types,
131	'URI' => array(url('openid/provider', array('absolute'=> TRUE))),
132	);
133	if ($account->uid) {
134	$data['LocalID'] = array(url('user/'. $account->uid, array('absolute' => TRUE)));
135	}
136
137	$xrds['openid_provider'] = array(
138	'services' => array(
139	array('priority' => 10,
140	'data' => $data
141	)
142	)
143	);
144
145	return $xrds;
146	}
147
148	/**
149	* Main OpenID Provider form
150	*/
151	function openid_provider_form(&$form_state, &$response = array(), $realm = NULL) {
152	global $user;
153
154	// Use form_state to store the $response and $realm values
155	if (count($response)) {
156	$form_state['storage']['response'] = $response;
157	}
158	else {
159	$response = $form_state['storage']['response'];
160	}
161
162	if ($realm) {
163	$form_state['storage']['realm'] = $realm;
164	}
165	else {
166	$realm = $form_state['storage']['realm'];
167	}
168
169	$form = array();
170
171	$form['intro'] = array(
172	'#type' => 'markup',
173	'#value' => '<p>'. t('You are being logged into %site, would you like to continue?', array('%site' => $realm)) . '</p>'
174	);
175
176	$form['#action'] = url('openid/provider/send');
177
178	$form['submit_once'] = array(
179	'#type' => 'submit',
180	'#value' => t('Yes; just this once'),
181	);
182	$form['submit_always'] = array(
183	'#type' => 'submit',
184	'#value' => t('Yes; always'),
185	'#submit' => array('openid_provider_form_submit_always')
186	);
187	$form['cancel'] = array(
188	'#type' => 'submit',
189	'#value' => t('Cancel'),
190	'#submit' => array('openid_provider_form_submit_cancel')
191	);
192
193	return $form;
194	}
195
196	/**
197	* Once submit handler
198	*/
199	function openid_provider_form_submit(&$form, $form_state, $auto_release = FALSE) {
200	global $user;
201
202	module_load_include('inc', 'openid');
203	module_load_include('inc', 'openid_provider');
204
205	$response = _openid_provider_sign($form_state['storage']['response']);
206	_openid_provider_rp_save($user->uid, $form_state['storage']['realm'], $auto_release);
207	openid_redirect_http($response['openid.return_to'], $response);
208	}
209
210	/**
211	* Always submit handler
212	*/
213	function openid_provider_form_submit_always(&$form, $form_state) {
214	return openid_provider_form_submit($form, $form_state, TRUE);
215	}
216
217	/**
218	* Cancel submit handler
219	*/
220	function openid_provider_form_submit_cancel(&$form, $form_state) {
221	module_load_include('inc', 'openid_provider');
222	module_load_include('inc', 'openid');
223
224	$return_to = $form_state['values']['openid.return_to'];
225	$response = openid_provider_cancel_authentication_response($form_state['openid.mode']);
226	openid_redirect($return_to, $response);
227	}