diff options
Diffstat (limited to 'sys/netsmb/smb_subr.h')
-rw-r--r-- | sys/netsmb/smb_subr.h | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/sys/netsmb/smb_subr.h b/sys/netsmb/smb_subr.h new file mode 100644 index 0000000..16ce212 --- /dev/null +++ b/sys/netsmb/smb_subr.h @@ -0,0 +1,198 @@ +/* + * Copyright (c) 2000-2001, Boris Popov + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. All advertising materials mentioning features or use of this software + * must display the following acknowledgement: + * This product includes software developed by Boris Popov. + * 4. Neither the name of the author nor the names of any co-contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + * + * $FreeBSD$ + */ +#ifndef _NETSMB_SMB_SUBR_H_ +#define _NETSMB_SMB_SUBR_H_ + +#ifndef _KERNEL +#error "This file shouldn't be included from userland programs" +#endif + +#ifdef MALLOC_DECLARE +MALLOC_DECLARE(M_SMBTEMP); +#endif + +#if __FreeBSD_version > 500000 +#define FB_CURRENT +#else +# if __FreeBSD_version > 400000 +# define FB_RELENG4 +# else +# error "Unsupported version of FreeBSD" +# endif +#endif + +#define SMBERROR(format, args...) printf("%s: "format, __FUNCTION__ ,## args) +#define SMBPANIC(format, args...) printf("%s: "format, __FUNCTION__ ,## args) + +#ifdef SMB_SOCKET_DEBUG +#define SMBSDEBUG(format, args...) printf("%s: "format, __FUNCTION__ ,## args) +#else +#define SMBSDEBUG(format, args...) +#endif + +#ifdef SMB_IOD_DEBUG +#define SMBIODEBUG(format, args...) printf("%s: "format, __FUNCTION__ ,## args) +#else +#define SMBIODEBUG(format, args...) +#endif + +#ifdef SMB_SOCKETDATA_DEBUG +void m_dumpm(struct mbuf *m); +#else +#define m_dumpm(m) +#endif + +#if __FreeBSD_version > 400009 +#define SMB_SIGMASK(set) \ + (SIGISMEMBER(set, SIGINT) || SIGISMEMBER(set, SIGTERM) || \ + SIGISMEMBER(set, SIGHUP) || SIGISMEMBER(set, SIGKILL) || \ + SIGISMEMBER(set, SIGQUIT)) + +#define smb_suser(cred) suser_xxx(cred, NULL, 0) +#else +#define SMB_SIGMASK (sigmask(SIGINT)|sigmask(SIGTERM)|sigmask(SIGKILL)| \ + sigmask(SIGHUP)|sigmask(SIGQUIT)) + +#define smb_suser(cred) suser((cred), NULL) +#endif + +/* + * Compatibility wrappers for simple locks + */ +#if __FreeBSD_version < 500000 + +#include <sys/lock.h> + +#define lockdestroy(lock) +#define smb_slock simplelock +#define smb_sl_init(mtx, desc) simple_lock_init(mtx) +#define smb_sl_destroy(mtx) +#define smb_sl_lock(mtx) simple_lock(mtx) +#define smb_sl_unlock(mtx) simple_unlock(mtx) +/* +#define mtx lock +#define mtx_init(mtx, desc, flags) lockinit(mtx, PWAIT, desc, 0, 0) +#define mtx_lock(mtx) lockmgr(mtx, LK_EXCLUSIVE, NULL, curproc) +#define mtx_unlock(mtx) lockmgr(mtx, LK_RELEASE, NULL, curproc) +#define mtx_destroy(mtx) +*/ +#else + +#include <sys/mutex.h> + +#define smb_slock mtx +#define smb_sl_init(mtx, desc) mtx_init(mtx, desc, MTX_DEF) +#define smb_sl_destroy(mtx) mtx_destroy(mtx) +#define smb_sl_lock(mtx) mtx_lock(mtx) +#define smb_sl_unlock(mtx) mtx_unlock(mtx) + +#endif + +#define SMB_STRFREE(p) do { if (p) smb_strfree(p); } while(0) + +/* + * The simple try/catch/finally interface. + * With GCC it is possible to allow more than one try/finally block per + * function, but we'll avoid it to maintain portability. + */ +#define itry { \ + __label__ _finlab, _catchlab; \ + int _tval; \ + +#define icatch(var) \ + goto _finlab; \ + (void)&&_catchlab; \ + _catchlab: \ + var = _tval; + +#define ifinally (void)&&_finlab; \ + _finlab: +#define iendtry } + +#define inocatch \ + goto _finlab; \ + (void)&&_catchlab; \ + _catchlab: \ + +#define ithrow(t) do { \ + if ((_tval = (int)(t)) != 0) \ + goto _catchlab; \ + } while (0) + +#define ierror(t,e) do { \ + if (t) { \ + _tval = e; \ + goto _catchlab; \ + } \ + } while (0) + +typedef u_int16_t smb_unichar; +typedef smb_unichar *smb_uniptr; + +/* + * Crediantials of user/process being processing in the connection procedures + */ +struct smb_cred { + struct proc * scr_p; + struct ucred * scr_cred; +}; + +extern smb_unichar smb_unieol; + +struct mbchain; +struct smb_vc; +struct smb_rq; + +void smb_makescred(struct smb_cred *scred, struct proc *p, struct ucred *cred); +int smb_proc_intr(struct proc *); +char *smb_strdup(const char *s); +void *smb_memdup(const void *umem, int len); +char *smb_strdupin(char *s, int maxlen); +void *smb_memdupin(void *umem, int len); +void smb_strtouni(u_int16_t *dst, const char *src); +void smb_strfree(char *s); +void smb_memfree(void *s); +void *smb_zmalloc(unsigned long size, struct malloc_type *type, int flags); + +int smb_encrypt(const u_char *apwd, u_char *C8, u_char *RN); +int smb_ntencrypt(const u_char *apwd, u_char *C8, u_char *RN); +int smb_maperror(int eclass, int eno); +int smb_put_dmem(struct mbchain *mbp, struct smb_vc *vcp, + const char *src, int len, int caseopt); +int smb_put_dstring(struct mbchain *mbp, struct smb_vc *vcp, + const char *src, int caseopt); +int smb_put_string(struct smb_rq *rqp, const char *src); +int smb_put_asunistring(struct smb_rq *rqp, const char *src); + +#endif /* !_NETSMB_SMB_SUBR_H_ */ |