1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/* ``The contents of this file are subject to the Erlang Public License,
* Version 1.1, (the "License"); you may not use this file except in
* compliance with the License. You should have received a copy of the
* Erlang Public License along with this software. If not, it can be
* retrieved via the world wide web at http://www.erlang.org/.
*
* Software distributed under the License is distributed on an "AS IS"
* basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
* the License for the specific language governing rights and limitations
* under the License.
*
* The Initial Developer of the Original Code is Ericsson Utvecklings AB.
* Portions created by Ericsson are Copyright 1999, Ericsson Utvecklings
* AB. All Rights Reserved.''
*
* $Id$
*/
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include "erl_driver.h"
ErlDrvData start(ErlDrvPort port, char *command);
int control(ErlDrvData drv_data, unsigned int command, char *buf,
int len, char **rbuf, int rlen);
static int call(ErlDrvData drv_data,
unsigned int command,
char *buf, int len,
char **rbuf, int rlen,
unsigned int *flags);
static ErlDrvEntry thr_alloc_drv_entry = {
NULL /* init */,
start,
NULL /* stop */,
NULL /* output */,
NULL /* ready_input */,
NULL /* ready_output */,
"thr_alloc_drv",
NULL /* finish */,
NULL /* handle */,
control,
NULL /* timeout */,
NULL /* outputv */,
NULL /* ready_async */,
NULL /* flush */,
NULL /* call */,
NULL /* event */,
ERL_DRV_EXTENDED_MARKER,
ERL_DRV_EXTENDED_MAJOR_VERSION,
ERL_DRV_EXTENDED_MINOR_VERSION,
ERL_DRV_FLAG_USE_PORT_LOCKING,
NULL /* handle2 */,
NULL /* handle_monitor */
};
DRIVER_INIT(thr_alloc_drv)
{
return &thr_alloc_drv_entry;
}
void *
test_thread(void *vsize)
{
int i;
int size = (int) (long) vsize;
void *mem;
mem = driver_alloc(size);
if (mem)
driver_free(mem);
}
ErlDrvData start(ErlDrvPort port, char *command)
{
return (ErlDrvData) port;
}
int control(ErlDrvData drv_data, unsigned int command, char *buf,
int len, char **rbuf, int rlen)
{
ErlDrvPort port = (ErlDrvPort) drv_data;
char *result = "failure";
int result_len;
if (len <= 20) {
int res;
ErlDrvTid tid;
char ibuf[21];
int size;
memcpy((void *) ibuf, buf, len);
ibuf[len] = '\0';
size = atoi(ibuf);
if (size > 0) {
res = erl_drv_thread_create("test_thread",
&tid,
test_thread,
(void *) (long) size,
NULL);
if (res == 0) {
res = erl_drv_thread_join(tid, NULL);
if (res == 0)
result = "ok";
}
if (res != 0)
driver_failure_posix(port, res);
}
}
result_len = strlen(result);
if (result_len <= rlen) {
memcpy(*rbuf, result, result_len);
return result_len;
}
else {
*rbuf = driver_alloc(result_len);
if (!*rbuf) {
driver_failure_posix(port, ENOMEM);
return 0;
}
else {
memcpy(*rbuf, result, result_len);
return result_len;
}
}
}
|