mbuf.c 5.0 KB
Newer Older
B
bellard 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * Copyright (c) 1995 Danny Gasparovski
 *
 * Please read the file COPYRIGHT for the
 * terms and conditions of the copyright.
 */

/*
 * mbuf's in SLiRP are much simpler than the real mbufs in
 * FreeBSD.  They are fixed size, determined by the MTU,
 * so that one whole packet can fit.  Mbuf's cannot be
 * chained together.  If there's more data than the mbuf
 * could hold, an external malloced buffer is pointed to
 * by m_ext (and the data pointers) and M_EXT is set in
 * the flags
 */

P
Peter Maydell 已提交
18
#include "qemu/osdep.h"
B
bellard 已提交
19 20
#include <slirp.h>

21 22 23 24 25
#define MBUF_THRESH 30

/*
 * Find a nice value for msize
 */
26 27
#define SLIRP_MSIZE\
    (offsetof(struct mbuf, m_dat) + IF_MAXLINKHDR + TCPIPHDR_DELTA + IF_MTU)
B
bellard 已提交
28 29

void
30
m_init(Slirp *slirp)
B
bellard 已提交
31
{
32 33
    slirp->m_freelist.qh_link = slirp->m_freelist.qh_rlink = &slirp->m_freelist;
    slirp->m_usedlist.qh_link = slirp->m_usedlist.qh_rlink = &slirp->m_usedlist;
B
bellard 已提交
34 35
}

36 37 38 39
void m_cleanup(Slirp *slirp)
{
    struct mbuf *m, *next;

40 41
    m = (struct mbuf *) slirp->m_usedlist.qh_link;
    while ((struct quehead *) m != &slirp->m_usedlist) {
42 43 44 45 46 47 48
        next = m->m_next;
        if (m->m_flags & M_EXT) {
            free(m->m_ext);
        }
        free(m);
        m = next;
    }
49 50
    m = (struct mbuf *) slirp->m_freelist.qh_link;
    while ((struct quehead *) m != &slirp->m_freelist) {
51 52 53 54 55 56
        next = m->m_next;
        free(m);
        m = next;
    }
}

B
bellard 已提交
57 58 59
/*
 * Get an mbuf from the free list, if there are none
 * malloc one
60
 *
B
bellard 已提交
61 62 63 64 65
 * Because fragmentation can occur if we alloc new mbufs and
 * free old mbufs, we mark all mbufs above mbuf_thresh as M_DOFREE,
 * which tells m_free to actually free() it
 */
struct mbuf *
66
m_get(Slirp *slirp)
B
bellard 已提交
67 68 69
{
	register struct mbuf *m;
	int flags = 0;
70

B
bellard 已提交
71
	DEBUG_CALL("m_get");
72

73
	if (slirp->m_freelist.qh_link == &slirp->m_freelist) {
74
		m = (struct mbuf *)malloc(SLIRP_MSIZE);
B
bellard 已提交
75
		if (m == NULL) goto end_error;
76 77
		slirp->mbuf_alloced++;
		if (slirp->mbuf_alloced > MBUF_THRESH)
B
bellard 已提交
78
			flags = M_DOFREE;
79
		m->slirp = slirp;
B
bellard 已提交
80
	} else {
81
		m = (struct mbuf *) slirp->m_freelist.qh_link;
B
bellard 已提交
82 83
		remque(m);
	}
84

B
bellard 已提交
85
	/* Insert it in the used list */
86
	insque(m,&slirp->m_usedlist);
B
bellard 已提交
87
	m->m_flags = (flags | M_USEDLIST);
88

B
bellard 已提交
89
	/* Initialise it */
A
Anthony Liguori 已提交
90
	m->m_size = SLIRP_MSIZE - offsetof(struct mbuf, m_dat);
B
bellard 已提交
91 92
	m->m_data = m->m_dat;
	m->m_len = 0;
93 94
        m->m_nextpkt = NULL;
        m->m_prevpkt = NULL;
95
        m->resolution_requested = false;
F
Fabien Chouteau 已提交
96
        m->expiration_date = (uint64_t)-1;
B
bellard 已提交
97
end_error:
98
	DEBUG_ARG("m = %p", m);
B
bellard 已提交
99 100 101 102
	return m;
}

void
103
m_free(struct mbuf *m)
B
bellard 已提交
104
{
105

B
bellard 已提交
106
  DEBUG_CALL("m_free");
107
  DEBUG_ARG("m = %p", m);
108

B
bellard 已提交
109 110 111 112
  if(m) {
	/* Remove from m_usedlist */
	if (m->m_flags & M_USEDLIST)
	   remque(m);
113

B
bellard 已提交
114 115 116 117 118 119 120 121
	/* If it's M_EXT, free() it */
	if (m->m_flags & M_EXT)
	   free(m->m_ext);

	/*
	 * Either free() it or put it on the free list
	 */
	if (m->m_flags & M_DOFREE) {
122
		m->slirp->mbuf_alloced--;
M
Mark McLoughlin 已提交
123
		free(m);
B
bellard 已提交
124
	} else if ((m->m_flags & M_FREELIST) == 0) {
125
		insque(m,&m->slirp->m_freelist);
B
bellard 已提交
126 127 128 129 130 131 132 133 134 135 136
		m->m_flags = M_FREELIST; /* Clobber other flags */
	}
  } /* if(m) */
}

/*
 * Copy data from one mbuf to the end of
 * the other.. if result is too big for one mbuf, malloc()
 * an M_EXT data segment
 */
void
137
m_cat(struct mbuf *m, struct mbuf *n)
B
bellard 已提交
138 139 140 141 142 143
{
	/*
	 * If there's no room, realloc
	 */
	if (M_FREEROOM(m) < n->m_len)
		m_inc(m,m->m_size+MINCSIZE);
144

B
bellard 已提交
145 146 147 148 149 150 151 152 153
	memcpy(m->m_data+m->m_len, n->m_data, n->m_len);
	m->m_len += n->m_len;

	m_free(n);
}


/* make m size bytes large */
void
154
m_inc(struct mbuf *m, int size)
B
bellard 已提交
155
{
B
bellard 已提交
156 157
	int datasize;

B
bellard 已提交
158 159 160 161
	/* some compiles throw up on gotos.  This one we can fake. */
        if(m->m_size>size) return;

        if (m->m_flags & M_EXT) {
B
bellard 已提交
162
	  datasize = m->m_data - m->m_ext;
B
bellard 已提交
163
	  m->m_ext = (char *)realloc(m->m_ext,size);
B
bellard 已提交
164
	  m->m_data = m->m_ext + datasize;
B
bellard 已提交
165 166 167 168 169
        } else {
	  char *dat;
	  datasize = m->m_data - m->m_dat;
	  dat = (char *)malloc(size);
	  memcpy(dat, m->m_dat, m->m_size);
170

B
bellard 已提交
171 172 173 174
	  m->m_ext = dat;
	  m->m_data = m->m_ext + datasize;
	  m->m_flags |= M_EXT;
        }
175

B
bellard 已提交
176 177 178 179 180 181 182
        m->m_size = size;

}



void
183
m_adj(struct mbuf *m, int len)
B
bellard 已提交
184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
{
	if (m == NULL)
		return;
	if (len >= 0) {
		/* Trim from head */
		m->m_data += len;
		m->m_len -= len;
	} else {
		/* Trim from tail */
		len = -len;
		m->m_len -= len;
	}
}


/*
 * Copy len bytes from m, starting off bytes into n
 */
int
203
m_copy(struct mbuf *n, struct mbuf *m, int off, int len)
B
bellard 已提交
204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
{
	if (len > M_FREEROOM(n))
		return -1;

	memcpy((n->m_data + n->m_len), (m->m_data + off), len);
	n->m_len += len;
	return 0;
}


/*
 * Given a pointer into an mbuf, return the mbuf
 * XXX This is a kludge, I should eliminate the need for it
 * Fortunately, it's not used often
 */
struct mbuf *
220
dtom(Slirp *slirp, void *dat)
B
bellard 已提交
221 222
{
	struct mbuf *m;
223

B
bellard 已提交
224
	DEBUG_CALL("dtom");
225
	DEBUG_ARG("dat = %p", dat);
B
bellard 已提交
226 227

	/* bug corrected for M_EXT buffers */
228 229
	for (m = (struct mbuf *) slirp->m_usedlist.qh_link;
	     (struct quehead *) m != &slirp->m_usedlist;
230
	     m = m->m_next) {
B
bellard 已提交
231 232 233 234 235 236 237 238
	  if (m->m_flags & M_EXT) {
	    if( (char *)dat>=m->m_ext && (char *)dat<(m->m_ext + m->m_size) )
	      return m;
	  } else {
	    if( (char *)dat >= m->m_dat && (char *)dat<(m->m_dat + m->m_size) )
	      return m;
	  }
	}
239

B
bellard 已提交
240
	DEBUG_ERROR((dfd, "dtom failed"));
241

B
bellard 已提交
242 243
	return (struct mbuf *)0;
}