slirp fix for -smb command (Initial patch by Juergen Keil)
[qemu] / slirp / mbuf.c
1 /*
2  * Copyright (c) 1995 Danny Gasparovski
3  *
4  * Please read the file COPYRIGHT for the
5  * terms and conditions of the copyright.
6  */
7
8 /*
9  * mbuf's in SLiRP are much simpler than the real mbufs in
10  * FreeBSD.  They are fixed size, determined by the MTU,
11  * so that one whole packet can fit.  Mbuf's cannot be
12  * chained together.  If there's more data than the mbuf
13  * could hold, an external malloced buffer is pointed to
14  * by m_ext (and the data pointers) and M_EXT is set in
15  * the flags
16  */
17
18 #include <slirp.h>
19
20 struct  mbuf *mbutl;
21 char    *mclrefcnt;
22 int mbuf_alloced = 0;
23 struct mbuf m_freelist, m_usedlist;
24 int mbuf_thresh = 30;
25 int mbuf_max = 0;
26 int msize;
27
28 void
29 m_init()
30 {
31         m_freelist.m_next = m_freelist.m_prev = &m_freelist;
32         m_usedlist.m_next = m_usedlist.m_prev = &m_usedlist;
33         msize_init();
34 }
35
36 void
37 msize_init()
38 {
39         /*
40          * Find a nice value for msize
41          * XXX if_maxlinkhdr already in mtu
42          */
43         msize = (if_mtu>if_mru?if_mtu:if_mru) + 
44                         if_maxlinkhdr + sizeof(struct m_hdr ) + 6;
45 }
46
47 /*
48  * Get an mbuf from the free list, if there are none
49  * malloc one
50  * 
51  * Because fragmentation can occur if we alloc new mbufs and
52  * free old mbufs, we mark all mbufs above mbuf_thresh as M_DOFREE,
53  * which tells m_free to actually free() it
54  */
55 struct mbuf *
56 m_get()
57 {
58         register struct mbuf *m;
59         int flags = 0;
60         
61         DEBUG_CALL("m_get");
62         
63         if (m_freelist.m_next == &m_freelist) {
64                 m = (struct mbuf *)malloc(msize);
65                 if (m == NULL) goto end_error;
66                 mbuf_alloced++;
67                 if (mbuf_alloced > mbuf_thresh)
68                         flags = M_DOFREE;
69                 if (mbuf_alloced > mbuf_max)
70                         mbuf_max = mbuf_alloced;
71         } else {
72                 m = m_freelist.m_next;
73                 remque(m);
74         }
75         
76         /* Insert it in the used list */
77         insque(m,&m_usedlist);
78         m->m_flags = (flags | M_USEDLIST);
79         
80         /* Initialise it */
81         m->m_size = msize - sizeof(struct m_hdr);
82         m->m_data = m->m_dat;
83         m->m_len = 0;
84         m->m_nextpkt = 0;
85         m->m_prevpkt = 0;
86 end_error:
87         DEBUG_ARG("m = %lx", (long )m);
88         return m;
89 }
90
91 void
92 m_free(m)
93         struct mbuf *m;
94 {
95         
96   DEBUG_CALL("m_free");
97   DEBUG_ARG("m = %lx", (long )m);
98         
99   if(m) {
100         /* Remove from m_usedlist */
101         if (m->m_flags & M_USEDLIST)
102            remque(m);
103         
104         /* If it's M_EXT, free() it */
105         if (m->m_flags & M_EXT)
106            free(m->m_ext);
107
108         /*
109          * Either free() it or put it on the free list
110          */
111         if (m->m_flags & M_DOFREE) {
112                 free(m);
113                 mbuf_alloced--;
114         } else if ((m->m_flags & M_FREELIST) == 0) {
115                 insque(m,&m_freelist);
116                 m->m_flags = M_FREELIST; /* Clobber other flags */
117         }
118   } /* if(m) */
119 }
120
121 /*
122  * Copy data from one mbuf to the end of
123  * the other.. if result is too big for one mbuf, malloc()
124  * an M_EXT data segment
125  */
126 void
127 m_cat(m, n)
128         register struct mbuf *m, *n;
129 {
130         /*
131          * If there's no room, realloc
132          */
133         if (M_FREEROOM(m) < n->m_len)
134                 m_inc(m,m->m_size+MINCSIZE);
135         
136         memcpy(m->m_data+m->m_len, n->m_data, n->m_len);
137         m->m_len += n->m_len;
138
139         m_free(n);
140 }
141
142
143 /* make m size bytes large */
144 void
145 m_inc(m, size)
146         struct mbuf *m;
147         int size;
148 {
149         /* some compiles throw up on gotos.  This one we can fake. */
150         if(m->m_size>size) return;
151
152         if (m->m_flags & M_EXT) {
153           /* datasize = m->m_data - m->m_ext; */
154           m->m_ext = (char *)realloc(m->m_ext,size);
155 /*              if (m->m_ext == NULL)
156  *                      return (struct mbuf *)NULL;
157  */             
158           /* m->m_data = m->m_ext + datasize; */
159         } else {
160           int datasize;
161           char *dat;
162           datasize = m->m_data - m->m_dat;
163           dat = (char *)malloc(size);
164 /*              if (dat == NULL)
165  *                      return (struct mbuf *)NULL;
166  */
167           memcpy(dat, m->m_dat, m->m_size);
168           
169           m->m_ext = dat;
170           m->m_data = m->m_ext + datasize;
171           m->m_flags |= M_EXT;
172         }
173  
174         m->m_size = size;
175
176 }
177
178
179
180 void
181 m_adj(m, len)
182         struct mbuf *m;
183         int len;
184 {
185         if (m == NULL)
186                 return;
187         if (len >= 0) {
188                 /* Trim from head */
189                 m->m_data += len;
190                 m->m_len -= len;
191         } else {
192                 /* Trim from tail */
193                 len = -len;
194                 m->m_len -= len;
195         }
196 }
197
198
199 /*
200  * Copy len bytes from m, starting off bytes into n
201  */
202 int
203 m_copy(n, m, off, len)
204         struct mbuf *n, *m;
205         int off, len;
206 {
207         if (len > M_FREEROOM(n))
208                 return -1;
209
210         memcpy((n->m_data + n->m_len), (m->m_data + off), len);
211         n->m_len += len;
212         return 0;
213 }
214
215
216 /*
217  * Given a pointer into an mbuf, return the mbuf
218  * XXX This is a kludge, I should eliminate the need for it
219  * Fortunately, it's not used often
220  */
221 struct mbuf *
222 dtom(dat)
223         void *dat;
224 {
225         struct mbuf *m;
226         
227         DEBUG_CALL("dtom");
228         DEBUG_ARG("dat = %lx", (long )dat);
229
230         /* bug corrected for M_EXT buffers */
231         for (m = m_usedlist.m_next; m != &m_usedlist; m = m->m_next) {
232           if (m->m_flags & M_EXT) {
233             if( (char *)dat>=m->m_ext && (char *)dat<(m->m_ext + m->m_size) )
234               return m;
235           } else {
236             if( (char *)dat >= m->m_dat && (char *)dat<(m->m_dat + m->m_size) )
237               return m;
238           }
239         }
240         
241         DEBUG_ERROR((dfd, "dtom failed"));
242         
243         return (struct mbuf *)0;
244 }
245