Initial revision
[kopensolaris-gnu/glibc.git] / nptl / pthread_setspecific.c
1 /* Copyright (C) 2002 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Ulrich Drepper <drepper@redhat.com>, 2002.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, write to the Free
17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
18    02111-1307 USA.  */
19
20 #include <errno.h>
21 #include <stdlib.h>
22 #include "pthreadP.h"
23
24
25 int
26 __pthread_setspecific (key, value)
27      pthread_key_t key;
28      const void *value;
29 {
30   struct pthread *self;
31   unsigned int idx1st;
32   unsigned int idx2nd;
33   struct pthread_key_data *level2;
34   unsigned int seq;
35
36   self = THREAD_SELF;
37
38   /* Special case access to the first 2nd-level block.  This is the
39      usual case.  */
40   if (__builtin_expect (key < PTHREAD_KEY_2NDLEVEL_SIZE, 1))
41     {
42       /* Verify the key is sane.  */
43       if (KEY_UNUSED ((seq = __pthread_keys[key].seq)))
44         /* Not valid.  */
45         return EINVAL;
46
47       level2 = &self->specific_1stblock[key];
48     }
49   else
50     {
51       if (KEY_UNUSED ((seq = __pthread_keys[key].seq))
52           || key >= PTHREAD_KEYS_MAX)
53         /* Not valid.  */
54         return EINVAL;
55
56       idx1st = key / PTHREAD_KEY_2NDLEVEL_SIZE;
57       idx2nd = key % PTHREAD_KEY_2NDLEVEL_SIZE;
58
59       /* This is the second level array.  Allocate it if necessary.  */
60       level2 = THREAD_GETMEM_NC (self, specific[idx1st]);
61       if (level2 == NULL)
62         {
63           if (value == NULL)
64             /* We don't have to do anything.  The value would in any case
65                be NULL.  We can save the memory allocation.  */
66             return 0;
67
68           level2
69             = (struct pthread_key_data *) calloc (PTHREAD_KEY_2NDLEVEL_SIZE,
70                                                   sizeof (*level2));
71           if (level2 == NULL)
72             return ENOMEM;
73
74           THREAD_SETMEM_NC (self, specific[idx1st], level2);
75         }
76
77       /* Pointer to the right array element.  */
78       level2 = &level2[idx2nd];
79     }
80
81   /* Store the data and the sequence number so that we can recognize
82      stale data.  */
83   level2->seq = seq;
84   level2->data = (void *) value;
85
86   /* Remember that we stored at least one set of data.  */
87   THREAD_SETMEM (self, specific_used, true);
88
89   return 0;
90 }
91 strong_alias (__pthread_setspecific, pthread_setspecific)